Update agent.py

agent.py

@@ -6,6 +6,7 @@ import yaml
 # Simulated additional tools (implementation depends on external APIs or setup)
 @tool
 class GoogleSearchTool:
+        """Tool for performing Google searches using Custom Search JSON API"""
     def __init__(self):
         self.api_key = os.environ.get("GOOGLE_API_KEY")
         self.cse_id = os.environ.get("GOOGLE_CSE_ID")
@@ -13,19 +14,39 @@ class GoogleSearchTool:
             raise ValueError("GOOGLE_API_KEY and GOOGLE_CSE_ID must be set in environment variables.")
 
     def search(self, query: str) -> str:
-        # Placeholder: In practice, use googleapiclient or similar
+                """Perform a Google search query
+        Args:
+            query: Search query string
+        Returns:
+            Simulated search results (replace with actual API call)
+        """
+        
         return f"Google search results for '{query}' (simulated)."
 
 @tool
 class ImageAnalysisTool:
+    """Tool for analyzing images using computer vision"""
     def analyze(self, image_path: str) -> str:
         # Placeholder: Use Google Vision API or similar in real implementation
+        """Analyze an image file
+        Args:
+            image_path: Path to image file
+        Returns:
+            Simulated image analysis results
+        """
         return f"Analyzed image at '{image_path}' (simulated description)."
 
 @tool
 class LocalFileAudioTool:
+    """Tool for transcribing audio files"""
     def transcribe(self, file_path: str) -> str:
         # Placeholder: Use speech recognition library like SpeechRecognition in real setup
+        """Transcribe audio from file
+        Args:
+            file_path: Path to audio file
+        Returns:
+            Simulated transcription text
+        """    
         return f"Transcribed audio from '{file_path}' (simulated transcription)."
 
 class MagAgent: