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

streamable_tags bug fix

Browse Source
This commit is contained in:
goulustis
2026-01-06 14:03:33 +08:00
parent 95b61c7695
commit aa8971659b
1 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
lang_agent/graphs/dual_path.py
View File

@@ -21,7 +21,7 @@ from langgraph.graph import StateGraph, START, END
SYS_PROMPT = "you are a helpful helper who will have a fun conversation with the user" SYS_PROMPT = "you are a helpful helper who will have a fun conversation with the user"
TOOL_SYS_PROMPT = "base on the user's speech, identify their emotions and change the light color to its appropriate colors. If it sounds neutral, do nothing" TOOL_SYS_PROMPT = "base on the user's speech, identify their emotions and change the light color to its appropriate colors. Always use the tool"
@dataclass @dataclass
@@ -37,7 +37,12 @@ def turn_lights(col:Literal["red", "green", "yellow", "blue"]):
""" """
Turn on the color of the lights Turn on the color of the lights
""" """
print(f"TURNED ON LIGHT: {col} !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!") # print(f"TURNED ON LIGHT: {col} !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!")
import time
for _ in range(10):
print(f"TURNED ON LIGHT: {col} !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!")
time.sleep(0.3)
class Dual(GraphBase): class Dual(GraphBase):
@@ -46,7 +51,7 @@ class Dual(GraphBase):
self._build_modules() self._build_modules()
self.workflow = self._build_graph() self.workflow = self._build_graph()
self.streamable_tags = ["dual_chat_llm"] self.streamable_tags = [["dual_chat_llm"]]
def _build_modules(self): def _build_modules(self):
self.chat_llm = init_chat_model(model=self.config.llm_name, self.chat_llm = init_chat_model(model=self.config.llm_name,
@@ -66,8 +71,8 @@ class Dual(GraphBase):
self.memory = MemorySaver() self.memory = MemorySaver()
self.tool_manager: ToolManager = self.config.tool_manager_config.setup() self.tool_manager: ToolManager = self.config.tool_manager_config.setup()
self.chat_agent = create_agent(self.chat_llm, [], checkpointer=self.memory) self.chat_agent = create_agent(self.chat_llm, [], checkpointer=self.memory)
# self.tool_agent = create_agent(self.tool_llm, self.tool_manager.get_langchain_tools()) self.tool_agent = create_agent(self.tool_llm, self.tool_manager.get_langchain_tools())
self.tool_agent = create_agent(self.tool_llm, [turn_lights]) # self.tool_agent = create_agent(self.tool_llm, [turn_lights])
self.streamable_tags = [["dual_chat_llm"]] self.streamable_tags = [["dual_chat_llm"]]
@@ -105,5 +110,7 @@ if __name__ == "__main__":
HumanMessage("I feel very very sad")] HumanMessage("I feel very very sad")]
}, {"configurable": {"thread_id": "3"}} }, {"configurable": {"thread_id": "3"}}
out = dual.invoke(*nargs) # out = dual.invoke(*nargs)
print(out) # print(out)
for chunk in dual.invoke(*nargs, as_stream=True):
continue