chef_agent

services/agent_server/chef_service.py

@@ -1,3 +1,5 @@
+from typing import Iterator
+
 from langchain_core.messages import HumanMessage
 from langchain_core.runnables import RunnableConfig
 
@@ -17,7 +19,11 @@ class ChefService:
         return result
 
     @staticmethod
-    def chat_stream(thread_id: str, message: str):
+    def chat_stream(thread_id: str, message: str) -> Iterator[str]:
         config: RunnableConfig = {"configurable": {"thread_id": thread_id}}
         result = chef_agent.agent.stream({"messages": [("user", message)]}, config=config, stream_mode="messages")
-        return result
+
+        for message_chunk, metadata in result:
+            content = message_chunk.content
+            if content:
+                yield content