forgot the routes and schema

App/Chat/ChatRoutes.py

@@ -6,7 +6,8 @@ from App.Transcription.Model import Transcriptions
 from App.Transcription.Schemas import *
 from App import bot
 from .utils.Summarize import Summarizer
-
+from .utils.RAG import GenerativeAIAssistant
+from .Schema import ChatRequest
 import aiohttp
 import os
 
@@ -26,8 +27,9 @@ async def fetch_and_forward_request(request_body):
 
 
 @chat_router.post("/Chat")
-async def generate_response(request_body: str):
-    pass
+async def generate_response(req:ChatRequest):
+    assistant=GenerativeAIAssistant()
+    return assistant.generate_message(req.messages,req.taskId)
 
 
 @chat_router.get("/summarize/{task_id}")

App/Chat/Schema.py

@@ -0,0 +1,7 @@
+from typing import Optional
+from pydantic import BaseModel
+
+
+class ChatRequest(BaseModel):
+    messages: list[str]
+    taskId: str