core: refacto

src/agent/stream.py

@@ -0,0 +1,125 @@
+from gradio import ChatMessage
+
+from src.agent.mistral_agent import MistralAgent
+
+agent = MistralAgent()
+
+with open("./prompt.md", encoding="utf-8") as f:
+    SYSTEM_PROMPT = f.read()
+
+async def respond(message, history=None):
+    """
+    Respond to a user message using the Mistral agent.
+    """
+    if history is None:
+        history = []
+
+    history.append(ChatMessage(role="user", content=message))
+    history.append(ChatMessage(role="assistant", content="", metadata={"title": "Thinking", "status": "pending"}))
+    yield history
+
+    messages = [
+        {"role": "system", "content": SYSTEM_PROMPT},
+        {"role": "user", "content": message},
+        {
+            "role": "assistant",
+            "content": "THINKING: Let's tackle this problem, ",
+            "prefix": True,
+        },
+    ]
+    payload = {
+        "agent_id": agent.agent_id,
+        "messages": messages,
+        "stream": True,
+        "max_tokens": None,
+        "tools": agent.tools,
+        "tool_choice": "auto",
+        "presence_penalty": 0,
+        "frequency_penalty": 0,
+        "n": 1
+    }
+    response = await agent.client.agents.stream_async(**payload)
+
+    full = ""
+    thinking = ""
+    tooling = ""
+    final = ""
+
+    current_phase = None  # None | "thinking" | "tooling" | "final"
+
+    history[-1] = ChatMessage(role="assistant", content="", metadata={"title": "Thinking", "status": "pending"})
+
+    async for chunk in response:
+        delta = chunk.data.choices[0].delta
+        content = delta.content or ""
+        full += content
+
+            # Phase finale
+        if "FINAL ANSWER:" in full:
+
+            parts = full.split("FINAL ANSWER:", 1)
+            before_final = parts[0]
+            final = parts[1].strip()
+
+            if "TOOLING:" in before_final:
+                tooling = before_final.split("TOOLING:", 1)[1].strip()
+            else:
+                tooling = ""
+
+            if current_phase != "final":
+                if current_phase == "tooling":
+                    history[-1] = ChatMessage(role="assistant", content=tooling, metadata={"title": "Tooling", "status": "done"})
+                elif current_phase == "thinking":
+                    history[-1] = ChatMessage(role="assistant", content=thinking, metadata={"title": "Thinking", "status": "done"})
+
+                history.append(ChatMessage(role="assistant", content=final))
+                current_phase = "final"
+                yield history
+
+        # Phase outil
+        elif "TOOLING:" in full:
+
+            parts = full.split("TOOLING:", 1)
+            before_tooling = parts[0]
+            tooling = ""
+
+            if "THINKING:" in before_tooling:
+                thinking = before_tooling.split("THINKING:", 1)[1].strip()
+            else:
+                thinking = before_tooling.strip()
+
+            tooling = parts[1].strip()
+
+            if current_phase != "tooling":
+                if current_phase == "thinking":
+                    history[-1] = ChatMessage(role="assistant", content=thinking,
+                                              metadata={"title": "Thinking", "status": "done"})
+                history.append(
+                    ChatMessage(role="assistant", content=tooling, metadata={"title": "Tooling", "status": "pending"}))
+                current_phase = "tooling"
+            else:
+                history[-1] = ChatMessage(role="assistant", content=tooling,
+                                          metadata={"title": "Tooling", "status": "pending"})
+            yield history
+
+        # Phase réflexion
+        elif "THINKING:" in full or current_phase is None:
+
+            if "THINKING:" in full:
+                thinking = full.split("THINKING:", 1)[1].strip()
+            else:
+                thinking = full.strip()
+
+            if current_phase != "thinking":
+                history[-1] = ChatMessage(role="assistant", content=thinking, metadata={"title": "Thinking", "status": "pending"})
+                current_phase = "thinking"
+            else:
+                history[-1] = ChatMessage(role="assistant", content=thinking, metadata={"title": "Thinking", "status": "pending"})
+            yield history
+
+    if current_phase == "thinking":
+        history[-1] = ChatMessage(role="assistant", content=thinking, metadata={"title": "Thinking", "status": "done"})
+    elif current_phase == "tooling":
+        history[-1] = ChatMessage(role="assistant", content=tooling, metadata={"title": "Tooling", "status": "done"})
+
+    yield history

src/ui/sidebar.py

@@ -1,130 +1,6 @@
 import gradio as gr
-import json
-from gradio import ChatMessage
 
-from src.agent.inference import MistralAgent
-
-agent = MistralAgent()
-
-with open("./prompt.md", encoding="utf-8") as f:
-    SYSTEM_PROMPT = f.read()
-
-async def respond(message, history=None):
-    """
-    Respond to a user message using the Mistral agent.
-    """
-    if history is None:
-        history = []
-
-    history.append(ChatMessage(role="user", content=message))
-    history.append(ChatMessage(role="assistant", content="", metadata={"title": "Thinking", "status": "pending"}))
-    yield history
-
-    messages = [
-        {"role": "system", "content": SYSTEM_PROMPT},
-        {"role": "user", "content": message},
-        {
-            "role": "assistant",
-            "content": "THINKING: Let's tackle this problem, ",
-            "prefix": True,
-        },
-    ]
-    payload = {
-        "agent_id": agent.agent_id,
-        "messages": messages,
-        "stream": True,
-        "max_tokens": None,
-        "tools": agent.tools,
-        "tool_choice": "auto",
-        "presence_penalty": 0,
-        "frequency_penalty": 0,
-        "n": 1
-    }
-    response = await agent.client.agents.stream_async(**payload)
-
-    full = ""
-    thinking = ""
-    tooling = ""
-    final = ""
-
-    current_phase = None  # None | "thinking" | "tooling" | "final"
-
-    history[-1] = ChatMessage(role="assistant", content="", metadata={"title": "Thinking", "status": "pending"})
-
-    async for chunk in response:
-        delta = chunk.data.choices[0].delta
-        content = delta.content or ""
-        full += content
-
-            # Phase finale
-        if "FINAL ANSWER:" in full:
-
-            parts = full.split("FINAL ANSWER:", 1)
-            before_final = parts[0]
-            final = parts[1].strip()
-
-            if "TOOLING:" in before_final:
-                tooling = before_final.split("TOOLING:", 1)[1].strip()
-            else:
-                tooling = ""
-
-            if current_phase != "final":
-                if current_phase == "tooling":
-                    history[-1] = ChatMessage(role="assistant", content=tooling, metadata={"title": "Tooling", "status": "done"})
-                elif current_phase == "thinking":
-                    history[-1] = ChatMessage(role="assistant", content=thinking, metadata={"title": "Thinking", "status": "done"})
-
-                history.append(ChatMessage(role="assistant", content=final))
-                current_phase = "final"
-                yield history
-
-        # Phase outil
-        elif "TOOLING:" in full:
-
-            parts = full.split("TOOLING:", 1)
-            before_tooling = parts[0]
-            tooling = ""
-
-            if "THINKING:" in before_tooling:
-                thinking = before_tooling.split("THINKING:", 1)[1].strip()
-            else:
-                thinking = before_tooling.strip()
-
-            tooling = parts[1].strip()
-
-            if current_phase != "tooling":
-                if current_phase == "thinking":
-                    history[-1] = ChatMessage(role="assistant", content=thinking,
-                                              metadata={"title": "Thinking", "status": "done"})
-                history.append(
-                    ChatMessage(role="assistant", content=tooling, metadata={"title": "Tooling", "status": "pending"}))
-                current_phase = "tooling"
-            else:
-                history[-1] = ChatMessage(role="assistant", content=tooling,
-                                          metadata={"title": "Tooling", "status": "pending"})
-            yield history
-
-        # Phase réflexion
-        elif "THINKING:" in full or current_phase is None:
-
-            if "THINKING:" in full:
-                thinking = full.split("THINKING:", 1)[1].strip()
-            else:
-                thinking = full.strip()
-
-            if current_phase != "thinking":
-                history[-1] = ChatMessage(role="assistant", content=thinking, metadata={"title": "Thinking", "status": "pending"})
-                current_phase = "thinking"
-            else:
-                history[-1] = ChatMessage(role="assistant", content=thinking, metadata={"title": "Thinking", "status": "pending"})
-            yield history
-
-    if current_phase == "thinking":
-        history[-1] = ChatMessage(role="assistant", content=thinking, metadata={"title": "Thinking", "status": "done"})
-    elif current_phase == "tooling":
-        history[-1] = ChatMessage(role="assistant", content=tooling, metadata={"title": "Tooling", "status": "done"})
-
-    yield history
+from src.agent.stream import respond
 
 
 def sidebar_ui(state, width=700, visible=True):