23 lines
723 B
Python
23 lines
723 B
Python
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 |