feat: Add NaliiaBot API
This commit is contained in:
22
src/naliiabotapi/api/v1/endpoints/chat.py
Normal file
22
src/naliiabotapi/api/v1/endpoints/chat.py
Normal file
@@ -0,0 +1,22 @@
|
||||
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}
|
||||
Reference in New Issue
Block a user