@@ -6,4 +6,10 @@
 class EmailMessageSchema(BaseModel):
 subject: str
 contents: str
-invalid_request: bool | None = Field(default=False)
+invalid_request: bool | None = Field(default=False)
+
+
+
+
+class SupervisorMessageSchema(BaseModel):
+content: str

@@ -1,9 +1,10 @@
 from typing import List
-from fastapi import APIRouter, Depends
+from fastapi import APIRouter, Depends, HTTPException
 from sqlmodel import Session, select
 
 from api.db import get_session
-from api.ai.schemas import EmailMessageSchema
+from api.ai.agents import get_supervisor
+from api.ai.schemas import EmailMessageSchema, SupervisorMessageSchema
 from api.ai.services import generate_email_message
 from .models import ChatMessagePayload, ChatMessage, ChatMessageListItem
 router = APIRouter()
@@ -29,7 +30,11 @@ def chat_list_messages(session: Session = Depends(get_session)):
 # curl -X POST -d '{"message": "Give me a summary of why it is good to go outside"}' -H "Content-Type: application/json" http://localhost:8080/api/chats/
 
 # curl -X POST -d '{"message": "Give me a summary of why it is good to go outside"}' -H "Content-Type: application/json" https://docker-api-fastapi-python-app-3jid9.ondigitalocean.app/api/chats/
-@router.post("/", response_model=EmailMessageSchema)
+
+
+# curl -X POST -d '{"message": "Research why it is good to go outside and email me the results"}' -H "Content-Type: application/json" http://localhost:8080/api/chats/
+
+@router.post("/", response_model=SupervisorMessageSchema)
 def chat_create_message(
 payload:ChatMessagePayload,
 session: Session = Depends(get_session)
@@ -38,7 +43,18 @@ def chat_create_message(
 obj = ChatMessage.model_validate(data)
 session.add(obj)
 session.commit()
-# session.refresh(obj) # ensure id/primary key added to the obj instance
-# ready to store in the database
-response = generate_email_message(payload.message)
-return response
+supe = get_supervisor()
+msg_data = {
+"messages": [
+{"role": "user",
+"content": f"{payload.message}"
+},
+]
+}
+result = supe.invoke(msg_data)
+if not result:
+raise HTTPException(status_code=400, detail="Error with supervisor")
+messages = result.get("messages")
+if not messages:
+raise HTTPException(status_code=400, detail="Error with supervisor")
+return messages[-1]