Fixed LiteLLMModel.

agent_config.py

@@ -37,7 +37,7 @@ def create_agent():
         CodeAgent: The configured agent instance.
     """
     # Load API key from environment variable (more secure)
-    token = os.getenv("OPENAI_API_KEY")
+    token = os.getenv("HF_API_KEY")
     if not token:
         raise RuntimeError("Missing Hugging Face API key. Set HF_API_KEY environment variable.")
 
@@ -45,9 +45,9 @@ def create_agent():
     # model = HfApiModel(token=token, model_id="gpt-3.5-turbo", provider="openai")
     
     model = LiteLLMModel(
+            token=token,
             model_id="gemini/gemini-2.0-flash",
-            api_base="https://generativelanguage.googleapis.com/v1beta/models",
-            api_key=os.getenv("GEMINI_API_KEY")
+            api_key=os.getenv("GEMINI_API_KEY"),
         )
 
     # Initialize tools
@@ -61,8 +61,10 @@ def create_agent():
     agent = CodeAgent(
         model=model,
         tools=[wiki_tool, web_search_tool, excel_analysis_tool, visitWebpageTool, python_interpreter_tool, check_answer, reverse_sentence_tool],
+        add_base_tools=True,
         max_steps=8,
-        verbosity_level=2
+        verbosity_level=2,
+        additional_authorized_imports=['pandas','numpy','csv','subprocess']
     )
     return agent
 

requirements.txt

@@ -3,4 +3,5 @@ smolagents
 wikipedia
 pandas
 openpyxl
-huggingface_hub
+huggingface_hub
+smolagents[litellm]