Update multiagents.py

multiagents.py

@@ -1,4 +1,3 @@
-# a multi agent proposal to solve HF agent course final assignment
 import os
 import dotenv
 import openai
@@ -36,21 +35,148 @@ class OpenAIAgent:
             if hasattr(tool, "name") and hasattr(tool, "run"):
                 name = tool.name
                 description = tool.__doc__ or ""
-                # TODO: Build params schema — if smolagents already exposes argument spec, use that
-                params = {
-                    "type": "object",
-                    "properties": {
-                        "url": {"type": "string", "description": "URL to fetch"}
-                    },
-                    "required": ["url"]
-                }
-                functions.append({"name": name, "description": description, "parameters": params})
+                
+                # Create proper schema based on tool name
+                if name == "search_web":
+                    params = {
+                        "type": "object",
+                        "properties": {
+                            "query": {"type": "string", "description": "Search query"}
+                        },
+                        "required": ["query"]
+                    }
+                elif name == "fetch_webpage":
+                    params = {
+                        "type": "object",
+                        "properties": {
+                            "url": {"type": "string", "description": "URL to fetch"}
+                        },
+                        "required": ["url"]
+                    }
+                elif name == "get_youtube_transcript":
+                    params = {
+                        "type": "object",
+                        "properties": {
+                            "url": {"type": "string", "description": "YouTube URL"}
+                        },
+                        "required": ["url"]
+                    }
+                elif name == "get_youtube_title_description":
+                    params = {
+                        "type": "object",
+                        "properties": {
+                            "url": {"type": "string", "description": "YouTube URL"}
+                        },
+                        "required": ["url"]
+                    }
+                elif name == "get_text_transcript_from_audio_file":
+                    params = {
+                        "type": "object",
+                        "properties": {
+                            "file_path": {"type": "string", "description": "Path to audio file"}
+                        },
+                        "required": ["file_path"]
+                    }
+                elif name == "analyze_image":
+                    params = {
+                        "type": "object",
+                        "properties": {
+                            "image_path": {"type": "string", "description": "Path to image file"}
+                        },
+                        "required": ["image_path"]
+                    }
+                else:
+                    # Default schema for unknown tools
+                    params = {
+                        "type": "object",
+                        "properties": {
+                            "input": {"type": "string", "description": "Input for the tool"}
+                        },
+                        "required": ["input"]
+                    }
+                
+                functions.append({
+                    "type": "function",
+                    "function": {
+                        "name": name,
+                        "description": description,
+                        "parameters": params
+                    }
+                })
+            
             # Handle normal Python functions
             elif hasattr(tool, "__name__"):
                 name = tool.__name__
                 description = tool.__doc__ or ""
-                # ... same param mapping as before ...
-                functions.append({...})
+                
+                # Create proper schema based on function name
+                if name == "search_web":
+                    params = {
+                        "type": "object",
+                        "properties": {
+                            "query": {"type": "string", "description": "Search query"}
+                        },
+                        "required": ["query"]
+                    }
+                elif name == "fetch_webpage":
+                    params = {
+                        "type": "object",
+                        "properties": {
+                            "url": {"type": "string", "description": "URL to fetch"}
+                        },
+                        "required": ["url"]
+                    }
+                elif name == "get_youtube_transcript":
+                    params = {
+                        "type": "object",
+                        "properties": {
+                            "url": {"type": "string", "description": "YouTube URL"}
+                        },
+                        "required": ["url"]
+                    }
+                elif name == "get_youtube_title_description":
+                    params = {
+                        "type": "object",
+                        "properties": {
+                            "url": {"type": "string", "description": "YouTube URL"}
+                        },
+                        "required": ["url"]
+                    }
+                elif name == "get_text_transcript_from_audio_file":
+                    params = {
+                        "type": "object",
+                        "properties": {
+                            "file_path": {"type": "string", "description": "Path to audio file"}
+                        },
+                        "required": ["file_path"]
+                    }
+                elif name == "analyze_image":
+                    params = {
+                        "type": "object",
+                        "properties": {
+                            "image_path": {"type": "string", "description": "Path to image file"}
+                        },
+                        "required": ["image_path"]
+                    }
+                else:
+                    # Default schema for unknown functions
+                    params = {
+                        "type": "object",
+                        "properties": {
+                            "input": {"type": "string", "description": "Input for the function"}
+                        },
+                        "required": ["input"]
+                    }
+                
+                functions.append({
+                    "type": "function",
+                    "function": {
+                        "name": name,
+                        "description": description,
+                        "parameters": params
+                    }
+                })
+        
         return functions
     
     def _execute_tool(self, tool_name: str, arguments: Dict[str, Any]):
@@ -145,10 +271,8 @@ class ManagerAgent(OpenAIAgent):
         agent_info = "\n".join([f"- {agent.name}: {agent.description}" for agent in self.managed_agents])
         
         system_prompt = f"""You are {self.name}. {self.description}
-
 Available agents you can delegate to:
 {agent_info}
-
 When you need to delegate a task, clearly state which agent should handle it and what specific task they should perform.
 You should coordinate the work and synthesize the results from different agents to provide a comprehensive answer.
 """