agent.py aktualisiert

2026-07-04 21:36:54 +00:00
parent ef8dae04b9
commit b8e75f16e7
1 changed files with 107 additions and 1 deletions
@@ -32,3 +32,109 @@ class Agent:
                )
            }
        ]
+    async def load_tools(self):
+        """
+        MCP Tools holen und in OpenRouter Format konvertieren
+        """
+
+        response = await self.mcp.list_tools()
+
+        self.tools = []
+
+        for t in response.tools:
+            self.tools.append({
+                "type": "function",
+                "function": {
+                    "name": t.name,
+                    "description": t.description,
+                    "parameters": t.inputSchema,
+                }
+            })
+
+        print(f"[JARVIS] {len(self.tools)} Tools geladen")
+    def _extract_json(self, text: str):
+        """
+        Repariert kaputte LLM JSON Outputs
+        """
+
+        try:
+            return json.loads(text)
+        except:
+            pass
+
+        # JSON aus Text extrahieren
+        match = re.search(r"\{.*\}", text, re.DOTALL)
+
+        if match:
+            try:
+                return json.loads(match.group(0))
+            except:
+                pass
+
+        return None
+    async def _execute_tool(self, tool_call):
+        """
+        Führt MCP Tool aus
+        """
+
+        name = tool_call["tool"]
+        args = tool_call.get("args", {})
+
+        result = await self.mcp.call_tool(name, args)
+
+        return result
+    async def run(self, user_input: str):
+        """
+        Vollständiger Tool-Calling Loop
+        """
+
+        self.messages.append({
+            "role": "user",
+            "content": user_input
+        })
+
+        MAX_STEPS = 8
+
+        for _ in range(MAX_STEPS):
+
+            response = await client.chat.completions.create(
+                model=OPENROUTER_MODEL,
+                messages=self.messages,
+                tools=self.tools
+            )
+
+            msg = response.choices[0].message.content
+
+            parsed = self._extract_json(msg)
+
+            # ❌ Falls kein gültiges JSON → direkte Antwort
+            if not parsed:
+                return msg
+
+            # 🧠 Tool Call
+            if "tool" in parsed:
+
+                tool_result = await self._execute_tool(parsed)
+
+                self.messages.append({
+                    "role": "assistant",
+                    "content": msg
+                })
+
+                self.messages.append({
+                    "role": "tool",
+                    "content": str(tool_result)
+                })
+
+                continue
+
+            # ✅ Final Response
+            if "response" in parsed:
+                self.messages.append({
+                    "role": "assistant",
+                    "content": parsed["response"]
+                })
+
+                return parsed["response"]
+
+        return "⚠️ Tool Loop Limit erreicht"