chef_agent

services/agent_server/chef_service.py

@@ -0,0 +1,23 @@
+from langchain_core.messages import HumanMessage
+from langchain_core.runnables import RunnableConfig
+
+from lc.chef_agent import chef_agent
+
+
+class ChefService:
+
+    @staticmethod
+    def chat(
+        thread_id: str,
+        message: str,
+    ):
+        config: RunnableConfig = {"configurable": {"thread_id": thread_id}}
+        result = chef_agent.agent.invoke({"messages": [HumanMessage(content=message)]}, config)
+
+        return result
+
+    @staticmethod
+    def chat_stream(thread_id: str, message: str):
+        config: RunnableConfig = {"configurable": {"thread_id": thread_id}}
+        result = chef_agent.agent.stream({"messages": [("user", message)]}, config=config, stream_mode="messages")
+        return result