Update agent.py

agent.py

@@ -5,49 +5,52 @@ import yaml
 
 # Simulated additional tools (implementation depends on external APIs or setup)
 @tool
-def GoogleSearchTool(query: str) -> str:
-    """Tool for performing Google searches using Custom Search JSON API
-        
-    Args:
-        query (str): Search query string to look up
+class GoogleSearchTool:
+    """Tool for performing Google searches using Custom Search JSON API"""
     
-    Returns:
-        str: Formatted search results (simulated or real)
-    """
     def __init__(self):
         self.api_key = os.environ.get("GOOGLE_API_KEY")
         self.cse_id = os.environ.get("GOOGLE_CSE_ID")
         if not self.api_key or not self.cse_id:
             raise ValueError("GOOGLE_API_KEY and GOOGLE_CSE_ID must be set in environment variables.")
 
+    @tool  # Decorate the method that implements the tool logic
+    def search(self, query: str) -> str:
+        """Perform a Google search query
+        Args:
+            query (str): Search query string
+        Returns:
+            str: Formatted search results
+        """
         return f"Google search results for '{query}' (simulated)."
 
 @tool
-def ImageAnalysisTool(image_path: str) -> str:
-    """Tool for analyzing images using computer vision
-        
-    Args:
-        image_path: Path to image file
-    Returns:
-        Simulated image analysis results
-    """
+class ImageAnalysisTool:
+    """Tool for analyzing images using computer vision"""
+    
+    @tool
     def analyze(self, image_path: str) -> str:
-        # Placeholder: Use Google Vision API or similar in real implementation
-        return f"Analyzed image at '{image_path}' (simulated description)."
+        """Analyze an image file
+        Args:
+            image_path (str): Path to image file
+        Returns:
+            str: Analysis results
+        """
+        return f"Analyzed image at '{image_path}' (simulated)."
 
 @tool
-def LocalFileAudioTool(file_path: str) -> str:
-    """Tool for transcribing audio files
-
-    Args:
-        file_path: Path to audio file
-    Returns:
-        Simulated transcription text
-    """    
-
+class LocalFileAudioTool:
+    """Tool for transcribing audio files"""
+    
+    @tool
     def transcribe(self, file_path: str) -> str:
-        # Placeholder: Use speech recognition library like SpeechRecognition in real setup
-        return f"Transcribed audio from '{file_path}' (simulated transcription)."
+        """Transcribe audio from file
+        Args:
+            file_path (str): Path to audio file
+        Returns:
+            str: Transcription text
+        """
+        return f"Transcribed audio from '{file_path}' (simulated)."
 
 class MagAgent:
     def __init__(self):