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

add thread id as inp

Browse Source
This commit is contained in:
goulustis
2025-10-29 18:53:39 +08:00
parent 3d5fb89283
commit 87ce104931
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lang_agent/pipeline.py
View File

@@ -121,7 +121,7 @@ class Pipeline:
return f"ws://{self.config.host}:{self.config.port}" return f"ws://{self.config.host}:{self.config.port}"
def chat(self, inp:str, as_stream:bool=False, as_raw:bool=False)->str: def chat(self, inp:str, as_stream:bool=False, as_raw:bool=False, thread_id:int = None)->str:
u = """ u = """
你叫小盏，是一个点餐助手，你的回复要简洁明了，不需要给用户提供选择。对话过程中不要出现提示用户下一步的操作，用可爱的语气进行交流，根据用户的语言使用对应的语言回答 你叫小盏，是一个点餐助手，你的回复要简洁明了，不需要给用户提供选择。对话过程中不要出现提示用户下一步的操作，用可爱的语气进行交流，根据用户的语言使用对应的语言回答
@@ -143,8 +143,9 @@ class Pipeline:
返回数据库中 status=1 并且时间是最新的数据。 返回数据库中 status=1 并且时间是最新的数据。
""" """
thread_id = thread_id if thread_id is not None else 3
inp = {"messages":[SystemMessage(u), inp = {"messages":[SystemMessage(u),
HumanMessage(inp)]}, {"configurable": {"thread_id": 3}} HumanMessage(inp)]}, {"configurable": {"thread_id": thread_id}}
out = self.invoke(*inp, as_stream=as_stream, as_raw=as_raw) out = self.invoke(*inp, as_stream=as_stream, as_raw=as_raw)