Upload 42 files

src/apis/interfaces/chat_interface.py

@@ -7,7 +7,6 @@ class ChatBody(BaseModel):
     history: Optional[list] = Field(None, title="Chat history")
     language: Optional[str] = Field("en", title="Language")
     topic: Optional[str] = Field("education", title="Topic")
-    filter: Optional[dict] = Field(None, title="Filter")
 
     model_config = {
         "json_schema_extra": {
@@ -22,7 +21,31 @@ class ChatBody(BaseModel):
                 ],
                 "language": "Vietnamese",
                 "topic": "education",
-                "filter": {"session_number": 1, "lesson_id": "L01"},
+            }
+        }
+    }
+
+
+class PrimaryChatBody(ChatBody):
+    pass
+
+
+class TutorChatBody(ChatBody):
+    filter: Optional[dict] = Field(None, title="Filter")
+    model_config = {
+        "json_schema_extra": {
+            "example": {
+                "query": "Vai trò của tri thức lịch sử là gì",
+                "history": [
+                    {"content": "Môn này là gì", "type": "human"},
+                    {
+                        "content": "Lịch sử về Châu Á",
+                        "type": "ai",
+                    },
+                ],
+                "language": "Vietnamese",
+                "topic": "education",
+                "filter": {"lesson_id": "L01"},
             }
         }
     }

src/apis/routers/chat_router.py

@@ -1,7 +1,12 @@
 from fastapi import APIRouter, status, Depends
 from fastapi.responses import JSONResponse
 from typing import Annotated
-from src.apis.interfaces.chat_interface import ChatBody, HighlightExplainBody
+from src.apis.interfaces.chat_interface import (
+    PrimaryChatBody,
+    ChatBody,
+    HighlightExplainBody,
+    TutorChatBody,
+)
 from src.apis.interfaces.entrance_eval_interface import TestResultsBody
 from src.agents.primary_chatbot.flow import rag_accuracy, rag_speed
 from src.agents.highlight_explain_agent.flow import highlight_workflow
@@ -10,26 +15,25 @@ from src.agents.entrance_eval_agent.flow import entrance_eval_agent
 router = APIRouter(prefix="/ai", tags=["AI"])
 
 
-@router.post("/rag_accuracy")
-async def primary_chat_accuracy(body: ChatBody):
+@router.post("/primary_chat")
+async def primary_chat(body: PrimaryChatBody):
     print("topic", body.topic)
-    response = await rag_accuracy.ainvoke(
+    response = await rag_speed.ainvoke(
         {
             "user_query": body.query,
             "messages_history": body.history,
             "language": body.language,
             "topic": body.topic,
-            "filter": body.filter,
+            "filter": None,
         }
     )
     final_response = response["llm_response"]
     return JSONResponse(status_code=status.HTTP_200_OK, content=final_response)
 
 
-@router.post("/rag_speed")
-async def primary_chat_speed(body: ChatBody):
-    print("filter", body.filter)
-    response = await rag_speed.ainvoke(
+@router.post("/tutor_chat")
+async def tutor_chat(body: TutorChatBody):
+    response = await rag_accuracy.ainvoke(
         {
             "user_query": body.query,
             "messages_history": body.history,