Fix model initiation

community_tools.py

@@ -1,9 +1,6 @@
 from langchain_google_community import GooglePlacesTool
 from langchain_community.agent_toolkits.load_tools import load_tools
-from smolagents.tools import Tool 
-
-import os
-os.environ["GPLACES_API_KEY"] = "AIzaSyBpYITz-7liGXiXF-bpkjMFsmDoyRp0YXc"
+from smolagents.tools import Tool
 
 google_map_tool = Tool.from_langchain(GooglePlacesTool())
 

mini_agents.py

@@ -19,8 +19,8 @@ MODEL_CHOICES = {
 }
 
 audio_model = InferenceClientModel(
-    model=MODEL_CHOICES["audio"][0],
-    api_key=os.getenv("HUGGINGFACE_API_KEY"),
+    model_id=MODEL_CHOICES["audio"][0],
+    token=os.getenv("HUGGINGFACE_API_KEY"),
     api_url="https://api.openai.com/v1/audio/transcriptions"
 )
 
@@ -33,8 +33,8 @@ audio_agent = CodeAgent(
 )
 
 vlm_model = InferenceClientModel(
-    model=MODEL_CHOICES["vlm"][0],
-    api_key=os.getenv("HUGGINGFACE_API_KEY"),
+    model_id=MODEL_CHOICES["vlm"][0],
+    token=os.getenv("HUGGINGFACE_API_KEY"),
     api_url="https://api.openai.com/v1/images/generations"
 )
 
@@ -47,8 +47,8 @@ vlm_agent = CodeAgent(
 )
 
 arithmetic_model = InferenceClientModel(
-    model=MODEL_CHOICES["arithmetic"][0],
-    api_key=os.getenv("HUGGINGFACE_API_KEY"),
+    model_id=MODEL_CHOICES["arithmetic"][0],
+    token=os.getenv("HUGGINGFACE_API_KEY"),
     api_url="https://api.openai.com/v1/chat/completions"
 )
 
@@ -61,8 +61,8 @@ arithmetic_agent = CodeAgent(
 )
 
 pandas_model = InferenceClientModel(
-    model=MODEL_CHOICES["pandas"][0],
-    api_key=os.getenv("HUGGINGFACE_API_KEY"),
+    model_id=MODEL_CHOICES["pandas"][0],
+    token=os.getenv("HUGGINGFACE_API_KEY"),
     api_url="https://api.openai.com/v1/chat/completions"
 )
 
@@ -74,8 +74,14 @@ pandas_agent = CodeAgent(
     description="This agent is responsible for converting data to a dataframe, performing pandas operations on such dataframe and converting the dataframe back to a json or a csv file."
 )
 
+code_model = InferenceClientModel(
+    model_id=MODEL_CHOICES["code"][0],
+    token=os.getenv("HUGGINGFACE_API_KEY"),
+    api_url="https://api.openai.com/v1/chat/completions"
+)
+
 multimodal_manager = CodeAgent(
-    model=MODEL_CHOICES["code"][0],
+    model=code_model,
     managed_agents=[audio_agent, vlm_agent],
     tools=[sort_list],
     max_steps=8,
@@ -85,7 +91,7 @@ multimodal_manager = CodeAgent(
 )
 
 operation_manager = CodeAgent(
-    model=MODEL_CHOICES["code"][0],
+    model=code_model,
     managed_agents=[arithmetic_agent, pandas_agent],
     tools=[sort_list],
     max_steps=8,
@@ -94,8 +100,14 @@ operation_manager = CodeAgent(
     description="This agent is responsible for managing the arithmetic and pandas agents."
 )
 
+master_model = InferenceClientModel(
+    model_id=MODEL_CHOICES["master"][0],
+    token=os.getenv("HUGGINGFACE_API_KEY"),
+    api_url="https://api.openai.com/v1/chat/completions"
+)
+
 master_agent = CodeAgent(
-    model=MODEL_CHOICES["master"][0],
+    model=master_model,
     managed_agents=[multimodal_manager, operation_manager],
     tools=[sort_list, *community_tools],
     max_steps=16,