agent.py aktualisiert

2026-07-05 13:00:00 +00:00
parent 85290cebd2
commit c7fdc6a912
1 changed files with 17 additions and 10 deletions
@@ -34,20 +34,27 @@ class Agent:
    async def load_tools(self):
        try:
            res = await self.mcp.list_tools()
-
-            self.tools = [
-                {
+    
+            self.tools = []
+            for t in res.tools:
+                schema = t.inputSchema
+    
+                # Bedrock-kompatibles Format erzwingen
+                if isinstance(schema, dict) and "json" not in schema:
+                    bedrock_schema = {"json": schema}
+                else:
+                    bedrock_schema = schema
+    
+                self.tools.append({
                    "toolSpec": {
                        "name": t.name,
                        "description": t.description,
-                        "inputSchema": t.inputSchema
+                        "inputSchema": bedrock_schema
                    }
-                }
-                for t in res.tools
-            ]
-
-            self.logger.info(f"{len(self.tools)} Tools erfolgreich geladen")
-
+                })
+    
+            self.logger.info(f"{len(self.tools)} Tools erfolgreich für Bedrock angepasst und geladen")
+    
        except Exception as e:
            self.logger.error(f"Tool Load Error: {e}")
            self.tools = []