22 lines
607 B
Python
22 lines
607 B
Python
from fastapi import APIRouter, Depends
|
|
from ...dependencies import get_agent
|
|
from langchain_core.messages import HumanMessage
|
|
|
|
router = APIRouter()
|
|
|
|
@router.post("/chat")
|
|
def chat(messages: dict, agent = Depends(get_agent)):
|
|
"""
|
|
Simulate a chat response based on the input message.
|
|
|
|
Args:
|
|
message (str): The input message from the user.
|
|
|
|
Returns:
|
|
dict: A dictionary containing the response message.
|
|
"""
|
|
|
|
messages = [HumanMessage(content=messages["messages"])]
|
|
response_message = agent.invoke({"messages": messages})
|
|
|
|
return {"response": response_message} |