Add gitignor

assistant/llm.py

@@ -0,0 +1,84 @@
+import json
+
+from mistralai.client import Mistral
+from . import config
+
+_client = Mistral(api_key=config.MISTRAL_API_KEY)
+
+_history: list[dict] = []
+
+
+def reset_history() -> None:
+    """Efface l'historique de conversation."""
+    _history.clear()
+
+
+def chat(user_message: str) -> str:
+    """Envoie un message au LLM, gère les appels d'outils MCP et retourne la réponse."""
+    from . import mcp_client
+
+    _history.append({"role": "user", "content": user_message})
+
+    manager = mcp_client.get_manager()
+    tools = manager.get_mistral_tools() if manager.has_tools else None
+
+    while True:
+        messages = [{"role": "system", "content": config.SYSTEM_PROMPT}] + _history
+
+        kwargs: dict = {"model": config.LLM_MODEL, "messages": messages}
+        if tools:
+            kwargs["tools"] = tools
+
+        response = _client.chat.complete(**kwargs)
+        choice = response.choices[0]
+        msg = choice.message
+
+        if msg.tool_calls:
+            # 1. Ajouter le message assistant (avec les appels d'outils) à l'historique
+            _history.append({
+                "role": "assistant",
+                "content": msg.content or "",
+                "tool_calls": [
+                    {
+                        "id": tc.id,
+                        "type": "function",
+                        "function": {
+                            "name": tc.function.name,
+                            "arguments": tc.function.arguments,
+                        },
+                    }
+                    for tc in msg.tool_calls
+                ],
+            })
+
+            # 2. Exécuter chaque outil et ajouter les résultats
+            for tc in msg.tool_calls:
+                tool_name = tc.function.name
+                try:
+                    args = (
+                        json.loads(tc.function.arguments)
+                        if isinstance(tc.function.arguments, str)
+                        else tc.function.arguments
+                    )
+                    print(f"  🔧 [MCP] {tool_name}({_short_args(args)})")
+                    result = manager.call_tool(tool_name, args)
+                except Exception as e:
+                    result = f"Erreur lors de l'appel à {tool_name} : {e}"
+
+                _history.append({
+                    "role": "tool",
+                    "content": result,
+                    "tool_call_id": tc.id,
+                })
+
+            # 3. Reboucler pour obtenir la réponse finale
+
+        else:
+            reply = msg.content or ""
+            _history.append({"role": "assistant", "content": reply})
+            return reply
+
+
+def _short_args(args: dict) -> str:
+    text = json.dumps(args, ensure_ascii=False)
+    return text[:80] + "…" if len(text) > 80 else text