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}