Update main.py

main.py

@@ -188,6 +188,15 @@ class LegalRequest(BaseModel):
     feature: str
     locale: str
     payload: LegalPayload
+class ChatPayload(BaseModel):
+    message: str
+    historique: Optional[List[Dict[str, Any]]] = []
+
+class ChatRequest(BaseModel):
+    requestId: str
+    feature: str
+    locale: str
+    payload: ChatPayload
 
 @app.get("/")
 def read_root():
@@ -200,21 +209,33 @@ def read_root():
 # ROUTE 1 : REQUÊTE TEXTE (Llama 3.3 70B + RAG Externe)
 # =========================================================
 @app.post("/api/chat")
-async def chat_text(request: TextRequest):
+async def chat_text(request: ChatRequest):
     if not groq_client:
         raise HTTPException(status_code=500, detail="Le moteur d'IA Groq n'est pas configuré.")
     
     try:
+        user_message = request.payload.message
+        
         completion = groq_client.chat.completions.create(
             model="llama-3.3-70b-versatile",
             messages=[
                 {"role": "system", "content": FULL_SYSTEM_PROMPT},
-                {"role": "user", "content": request.text}
+                {"role": "user", "content": user_message}
             ],
             temperature=0.3,
             max_tokens=1024
         )
-        return {"ai_response": completion.choices[0].message.content}
+        
+        # Récupération de la réponse brute de l'IA
+        ai_reply = completion.choices[0].message.content
+        
+        # Retour au format strict demandé pour le frontend
+        return {
+            "responseId": f"res_{request.requestId}",
+            "reply": ai_reply,
+            "status": "success"
+        }
+        
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Erreur d'inférence Groq BSTP : {str(e)}")