Update agent.py

agent.py

@@ -4,7 +4,6 @@ import os
 import re
 from typing import Optional
 from token_bucket import Limiter
-from youtube_transcript_api import YouTubeTranscriptApi 
 import yaml
 from PIL import Image
 import requests
@@ -94,19 +93,6 @@ def SpeechToTextTool(audio_path: str) -> str:
     return result.get("text", "")
 
 
-@tool
-def youtube_transcript(url: str) -> str:
-    """
-    Get transcript of YouTube video.
-    Args:
-        url: YouTube video url in ""
-    """    
-    video_id = url.partition("https://www.youtube.com/watch?v=")[2]
-
-    transcript = YouTubeTranscriptApi.get_transcript(video_id)
-    transcript_text = " ".join([item["text"] for item in transcript])
-    return {"youtube_transcript": transcript_text}
-    
 #@tool
 #class LocalFileAudioTool:
 #    """Tool for transcribing audio files"""
@@ -127,7 +113,7 @@ class MagAgent:
         self.rate_limiter = rate_limiter
         print("Initializing MagAgent with search tools...")
         model = LiteLLMModel(
-            model_id="gemini/gemini-2.0-flash",
+            model_id="gemini/gemini-2.0-flash-lite",
             api_key= os.environ.get("GEMINI_KEY"),
             max_tokens=8192
         )
@@ -170,14 +156,15 @@ class MagAgent:
             )
 
             # Ensure response is a string, fixing the integer error
-            response = str(response) if response is not None else "No answer found."
+            response = str(response) 
+            if response is None: 
+                print(f"No answer found.")
        
-            if not response or "No Wikipedia page found" in response:
-                # Fallback response if search fails
-                response = "Unable to retrieve exact data. Please refine the question or check external sources."
             print(f"MagAgent response: {response[:50]}...")
             return response
+        
         except Exception as e:
             error_msg = f"Error processing question: {str(e)}. Check API key or network connectivity."
             print(error_msg)
-            return error_msg
+            return error_msg
+