add analysis model_id arg

agents/single_agent.py

@@ -3,21 +3,22 @@ Web agent for finding and extracting song lyrics from online sources.
 """
 
 from loguru import logger
-from smolagents import CodeAgent, VisitWebpageTool, FinalAnswerTool, DuckDuckGoSearchTool
+from smolagents import CodeAgent, VisitWebpageTool, FinalAnswerTool
 
 from config import load_prompt_templates
 from tools.analysis_tools import AnalyzeLyricsTool
 from tools.formatting_tools import FormatAnalysisResultsTool
-from tools.search_tools import ThrottledDuckDuckGoSearchTool, BraveSearchTool
+from tools.search_tools import BraveSearchTool
 from tools.image_generation_tools import GenerateImageTool
 
 
-def create_single_agent(model):
+def create_single_agent(model, analysis_tool_model_id=None):
     """ 
     Create an agent specialized in web browsing and lyrics extraction.
     
     Args:
         model: The LLM model to use with this agent
+        model_id: The model ID string to use for lyrics analysis (optional)
         
     Returns:
         A configured CodeAgent for web searches
@@ -31,7 +32,7 @@ def create_single_agent(model):
             BraveSearchTool(),
             # ThrottledDuckDuckGoSearchTool(min_delay=7.0, max_delay=15.0),
             VisitWebpageTool(),
-            AnalyzeLyricsTool(),
+            AnalyzeLyricsTool(model_id=analysis_tool_model_id),
             FormatAnalysisResultsTool(),
             GenerateImageTool(),
             FinalAnswerTool(),

app.py

@@ -46,7 +46,7 @@ def main():
         model = LiteLLMModel(model_id=model_id)
     
     # Create the manager agent which will create and manage the other agents
-    single_agent = create_single_agent(model)
+    single_agent = create_single_agent(model, analysis_tool_model_id=model_id)
     
     # Start the Gradio UI server
     logger.info("Initializing Gradio UI and launching server")

tools/analysis_tools.py

@@ -22,6 +22,10 @@ class AnalyzeLyricsTool(Tool):
         }
     output_type = "string"
     
+    def __init__(self, model_id=None):
+        super().__init__()
+        self.model_id = model_id or "openrouter/google/gemini-2.0-flash-lite-preview-02-05:free"
+    
     def forward(self, song_title: str, artist: str, lyrics: str) -> str:
         """
         Performs a deep analysis of the given lyrics and artist metadata.
@@ -69,8 +73,7 @@ class AnalyzeLyricsTool(Tool):
         {lyrics}
         """
 
-        # model_to_use = "gemini/gemini-2.0-flash"
-        model_to_use = "openrouter/google/gemini-2.0-flash-lite-preview-02-05:free"
+        model_to_use = self.model_id
         logger.info("Using {} for lyrics analysis", model_to_use)
 
         # Use the function with retry mechanism