Fix

basic_agent.py

@@ -8,11 +8,10 @@ from dataclasses import dataclass, field
 import pandas as pd
 import requests
 import rich
-from loguru import logger
 import smolagents
-from smolagents import CodeAgent, DuckDuckGoSearchTool, FinalAnswerTool, HfApiModel, VisitWebpageTool
-
 from get_model import get_model
+from loguru import logger
+from smolagents import CodeAgent, DuckDuckGoSearchTool, FinalAnswerTool, HfApiModel, VisitWebpageTool
 
 print = rich.get_console().print
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -106,6 +105,7 @@ def main():
                 FinalAnswerTool(),
             ]
         )
+        agent.agent.visualize()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None

get_model.py

@@ -29,12 +29,8 @@ def get_model(cat: str = "hf", provider=None, model_id=None):
 
         # assert gemini_api_keys, "No GEMINI_API_KEYS, set env var GEMINI_API_KEYS or put them in .env-gemini and try again."
         if not gemini_api_keys:
-            logger.warning(
-                "cat='gemini' but no GEMINI_API_KEYS found, "
-                " returning HfApiModel()..."
-                " Set env var GEMINI_API_KEYS and/or .env-gemini "
-                " with free space gemini-api-keys if you want to try 'gemini' "
-            )
+            logger.warning("cat='gemini' but no GEMINI_API_KEYS found,  returning HfApiModel()... Set env var GEMINI_API_KEYS and/or .env-gemini  with free space gemini-api-keys if you want to try 'gemini' ")
+            logger.info(" set gemini but return HfApiModel()")
             return HfApiModel()
 
         # setup proxy for gemini and for golay (local)
@@ -90,7 +86,14 @@ def get_model(cat: str = "hf", provider=None, model_id=None):
                 "fallbacks": fallbacks,
             },
         )
+
+        if os.getenv("SILICONFLOW_API_KEY"):
+            logger.info(" set gemini, return LiteLLMRouterModel + fallbacks")
+        else:
+            logger.info(" set gemini, return LiteLLMRouterModel")
+
         return model
 
+    logger.info(" default return default HfApiModel(provider=None, model_id=None)")
     # if cat.lower() in ["hf"]:  default
     return HfApiModel(provider=provider, model_id=model_id)