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

pass in device_id

Browse Source
This commit is contained in:
goulustis
2025-12-30 22:44:21 +08:00
parent a8c2dd7723
commit 69713b3977
1 changed files with 17 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
lang_agent/pipeline.py
View File

@@ -153,8 +153,15 @@ class Pipeline:
# NOTE: this prompt will be overwritten by 'configs/route_sys_prompts/chat_prompt.txt' for route graph # NOTE: this prompt will be overwritten by 'configs/route_sys_prompts/chat_prompt.txt' for route graph
u = DEFAULT_PROMPT u = DEFAULT_PROMPT
device_id = "0"
spl_ls = thread_id.split("_")
assert len(spl_ls) <= 2, "something wrong!"
if len(spl_ls) == 2:
thread_id, device_id = spl_ls
inp = {"messages":[SystemMessage(u), inp = {"messages":[SystemMessage(u),
HumanMessage(inp)]}, {"configurable": {"thread_id": thread_id}} HumanMessage(inp)]}, {"configurable": {"thread_id": thread_id,
"device_id":device_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)
@@ -197,8 +204,16 @@ class Pipeline:
# NOTE: this prompt will be overwritten by 'configs/route_sys_prompts/chat_prompt.txt' for route graph # NOTE: this prompt will be overwritten by 'configs/route_sys_prompts/chat_prompt.txt' for route graph
u = DEFAULT_PROMPT u = DEFAULT_PROMPT
device_id = "0"
spl_ls = thread_id.split("_")
assert len(spl_ls) <= 2, "something wrong!"
if len(spl_ls) == 2:
thread_id, device_id = spl_ls
print(f"\033[32m====================DEVICE ID: {device_id}=============================\033[0m")
inp_data = {"messages":[SystemMessage(u), inp_data = {"messages":[SystemMessage(u),
HumanMessage(inp)]}, {"configurable": {"thread_id": thread_id}} HumanMessage(inp)]}, {"configurable": {"thread_id": thread_id,
"device_id":device_id}}
if as_stream: if as_stream:
# Return async generator for streaming # Return async generator for streaming