rename
This commit is contained in:
@@ -13,7 +13,7 @@ from langgraph.prebuilt import create_react_agent
|
|||||||
from langgraph.checkpoint.memory import MemorySaver
|
from langgraph.checkpoint.memory import MemorySaver
|
||||||
|
|
||||||
from lang_agent.config import InstantiateConfig
|
from lang_agent.config import InstantiateConfig
|
||||||
from lang_agent.tool_manager import ToolManager, ToolManagerConfig
|
from lang_agent.client_tool_manager import ToolManager, ToolManagerConfig
|
||||||
|
|
||||||
@tyro.conf.configure(tyro.conf.SuppressFixed)
|
@tyro.conf.configure(tyro.conf.SuppressFixed)
|
||||||
@dataclass
|
@dataclass
|
||||||
@@ -74,8 +74,6 @@ class Pipeline:
|
|||||||
tools = []
|
tools = []
|
||||||
self.agent = create_react_agent(self.llm, tools, checkpointer=memory)
|
self.agent = create_react_agent(self.llm, tools, checkpointer=memory)
|
||||||
|
|
||||||
# def respond(self, msg:str | List[SystemMessage, HumanMessage]):
|
|
||||||
# return self.agent.invoke(msg)
|
|
||||||
|
|
||||||
def invoke(self, *nargs, **kwargs):
|
def invoke(self, *nargs, **kwargs):
|
||||||
return self.agent.invoke(*nargs, **kwargs)
|
return self.agent.invoke(*nargs, **kwargs)
|
||||||
|
|||||||
Reference in New Issue
Block a user