Tangledup-ai/lang-agent
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

change to invoke

Browse Source
This commit is contained in:
goulustis
2025-10-15 14:31:08 +08:00
parent 09b11edc56
commit 3bb3954499
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lang_agent/pipeline.py
View File

@@ -73,8 +73,11 @@ class Pipeline:
tools = self.tool_manager.get_tools() tools = self.tool_manager.get_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]): # def respond(self, msg:str | List[SystemMessage, HumanMessage]):
return self.agent.invoke(msg) # return self.agent.invoke(msg)
def invoke(self, *nargs, **kwargs):
return self.agent.invoke(*nargs, **kwargs)
async def handle_connection(self, websocket:ServerConnection): async def handle_connection(self, websocket:ServerConnection):
try: try:
@@ -84,7 +87,7 @@ class Pipeline:
await websocket.send(message) await websocket.send(message)
else: else:
# TODO: handle this better, will have system/user prompt send here # TODO: handle this better, will have system/user prompt send here
response = self.respond(message) response = self.invoke(message)
await websocket.send(response) await websocket.send(response)
except websockets.ConnectionClosed: except websockets.ConnectionClosed:
pass pass