Update src/modules/jasmine_agent.py

src/modules/jasmine_agent.py

@@ -1,7 +1,7 @@
 """
-MODULE JASMINE AGENT - V50 (AIP LOGIC KERNEL ADAPTED)
-=====================================================
-Mise à jour : Intègre le raisonnement explicite (Chain of Thought) dans l'architecture existante.
+MODULE JASMINE AGENT - V FINAL (AIP LOGIC KERNEL + GROQ)
+========================================================
+Responsabilité : Traduire le langage naturel en actions ontologiques précises.
 """
 import google.generativeai as genai
 from groq import Groq
@@ -46,44 +46,43 @@ PREFIX rdfs: <http://www.w3.org/2000/01/rdf-schema#>
 PREFIX xsd: <http://www.w3.org/2001/XMLSchema#>
 """
 
-def build_base_system_prompt(self):
+    def build_base_system_prompt(self):
         return f"""
-ROLE: CERVEAU NUMÉRIQUE (AIP LOGIC KERNEL).
-MISSION: Résoudre les requêtes opérationnelles en ancrant chaque décision dans la vérité de l'Ontologie.
-
---- RÈGLES DE NAVIGATION (ENTITÉS vs NOMS) ---
-TRÈS IMPORTANT : Les prédicats comme 'a_emprunteur' ou 'est_garanti_par' lient des IDs techniques (ex: vortex:CLI-2026-0001). 
-Pour chercher par NOM (ex: 'Aichatou'), tu dois impérativement faire une jointure :
-1. Lier l'ID : ?pret vortex:a_emprunteur ?client .
-2. Lier le Nom : ?client vortex:nom ?nom . (ou rdfs:label)
-3. Filtrer : FILTER(CONTAINS(LCASE(STR(?nom)), 'aichatou'))
-
---- RÈGLES DE CALCUL & FILTRES (OAG STRICT) ---
-Pour toute question impliquant des chiffres (sommes, moyennes, seuils > 100k) :
-1. UTILISE l'outil 'execute_sparql'.
-2. NE JAMAIS utiliser 'search_semantic' pour un filtre numérique.
-3. PRIORITÉ GOLDEN RECORD : Utilise 'vortex:montant_net' pour les calculs de somme afin d'éviter les doublons Master/Update.
-
---- RÈGLES CRITIQUES (SPARQL) ---
-1. Namespace UNIQUE : <http://vortex.ai/ontology#> (prefix: vortex).
-2. Préfixes OBLIGATOIRES :
+ROLE: Tu es le Cerveau Numérique (AIP Logic) de VORTEX.
+MISSION: Répondre aux questions en interrogeant le Knowledge Graph via SPARQL.
+
+--- TON TERRITOIRE (ONTOLOGIE) ---
+1. Prêts : vortex:Pret (liés via 'a_emprunteur' à Client, 'est_garanti_par' à Garant).
+2. Clients : vortex:Client (propriété 'nom' ou 'rdfs:label' pour le nom complet).
+3. Montants : 
+   - 'vortex:montant' : Donnée brute (peut contenir des doublons).
+   - 'vortex:montant_net' : GOLDEN RECORD (Valeur unique et fiable pour les calculs).
+
+--- RÈGLES DE NAVIGATION (CRITIQUES) ---
+Pour chercher un dossier par NOM (ex: "Koulnan" ou "Aichatou") :
+NE JAMAIS faire : ?pret vortex:a_emprunteur 'Koulnan'. (Car a_emprunteur pointe vers un ID technique).
+TU DOIS FAIRE UNE JOINTURE :
+   ?pret vortex:a_emprunteur ?client_id .
+   ?client_id rdfs:label ?nom_client .
+   FILTER(CONTAINS(LCASE(STR(?nom_client)), 'koulnan'))
+
+--- RÈGLES DE CALCUL ---
+Pour une SOMME, un MONTANT TOTAL ou un SEUIL (> 100000) :
+1. Utilise TOUJOURS 'vortex:montant_net' (le Golden Record).
+2. Utilise TOUJOURS l'outil 'execute_sparql'.
+3. NE JAMAIS utiliser 'search_semantic' pour des filtres numériques.
+
+--- RÈGLES TECHNIQUES ---
+- Namespace UNIQUE : <http://vortex.ai/ontology#> (prefix: vortex).
+- Préfixes OBLIGATOIRES :
 {self._get_prefixes()}
 
---- RÈGLES DE RECHERCHE (TEXTE) ---
-Si l'utilisateur cherche un NOM ou un TEXTE :
-NE JAMAIS utiliser l'égalité stricte (=). Utilise TOUJOURS :
-FILTER(CONTAINS(LCASE(STR(?variable)), 'koulnan'))
-
---- PROCESSUS DE PENSÉE (CHAIN OF THOUGHT) ---
-Structure ton raisonnement dans 'thought_trace' :
-1. ANALYSE : Intention (Calcul, Navigation 360°, Recherche).
-2. EXTRACTION : Liste les entités (ex: Client:Aichatou).
-3. NAVIGATION : Identifie les sauts de jointure nécessaires (Prêt -> Client -> Nom).
-4. OUTIL : Choix de l'outil déterministe.
-
-CONTRAINTES :
-1. Réponds UNIQUEMENT en JSON.
-2. Si une donnée manque, renvoie "tool": "none".
+CONTRAINTES DE SORTIE (JSON STRICT) :
+{{
+  "thought_trace": "1. ANALYSE: ... 2. EXTRACTION: ... 3. NAVIGATION: ... 4. OUTIL: ...",
+  "tool": "execute_sparql" | "search_semantic" | "none",
+  "args": {{ "query": "..." }}
+}}
 """
 
     def _format_messages_for_groq(self, system_prompt, chat_history, user_message):
@@ -124,13 +123,6 @@ CONTRAINTES :
 --- 🎯 PLAN TACTIQUE ---
 STRATÉGIE: {plan['strategy']}
 CONTEXTE: {plan['reason']}
-
-⚠️ SORTIE JSON OBLIGATOIRE :
-{{
-  "thought_trace": "1. ANALYSE: ... 2. EXTRACTION: ... 3. NAVIGATION: ...",
-  "tool": "{plan['tool']}",
-  "args": {{ "query": "..." }}
-}}
 """
         last_error = None
         for model_name in self.MODEL_CASCADE:
@@ -140,35 +132,36 @@ CONTEXTE: {plan['reason']}
                     if not self.google_key: continue
                     msgs = self._format_messages_for_gemini(full_system_prompt, chat_history, user_message)
                     model = genai.GenerativeModel(model_name)
-                    res = model.generate_content(msgs)
+                    # Force JSON mode for Gemini
+                    res = model.generate_content(msgs, generation_config={"response_mime_type": "application/json"})
                     response_text = res.text
                 else:
                     if not self.groq_client: continue
                     msgs = self._format_messages_for_groq(full_system_prompt, chat_history, user_message)
                     completion = self.groq_client.chat.completions.create(
-                        model=model_name, messages=msgs, temperature=0.0
+                        model=model_name, messages=msgs, temperature=0.0, response_format={"type": "json_object"}
                     )
                     response_text = completion.choices[0].message.content
 
                 clean_text = response_text.strip()
-                match = re.search(r'(\{.*"tool":.*\})', clean_text, re.DOTALL)
+                # Nettoyage au cas où le modèle ajoute du markdown
+                clean_text = re.sub(r'```json\s*|\s*```', '', clean_text)
                 
-                action = None
-                thought_trace = "Raisonnement non disponible."
-
-                if match:
-                    try:
-                        json_str = match.group(1)
-                        action = json.loads(json_str)
-                        thought_trace = action.get("thought_trace", planning_log)
-                        
-                        if action.get("tool") == "execute_sparql":
-                            raw_query = action["args"].get("query", "")
-                            if "PREFIX vortex:" not in raw_query:
-                                action["args"]["query"] = self._get_prefixes() + "\n" + raw_query
-                        
-                        return "", action, thought_trace 
-                    except: pass
+                try:
+                    action = json.loads(clean_text)
+                    
+                    # Sécurité Préfixes SPARQL
+                    if action.get("tool") == "execute_sparql":
+                        raw_query = action["args"].get("query", "")
+                        if "PREFIX vortex:" not in raw_query:
+                            action["args"]["query"] = self._get_prefixes() + "\n" + raw_query
+                    
+                    # Récupération du Chain of Thought
+                    thought_trace = action.get("thought_trace", planning_log)
+                    
+                    return "", action, thought_trace 
+                except json.JSONDecodeError:
+                    pass
                 
                 return clean_text, None, planning_log