agent spec

mini_agents.py

@@ -58,7 +58,8 @@ AUTHORIZED_IMPORTS = [
 
 audio_model = InferenceClientModel(
     model_id=MODEL_CHOICES["audio"][0],
-    token=os.getenv("HUGGINGFACE_API_KEY")
+    token=os.getenv("HUGGINGFACE_API_KEY"),
+    max_tokens=12000
 )
 
 audio_agent = CodeAgent(
@@ -73,7 +74,8 @@ audio_agent = CodeAgent(
 
 vlm_model = InferenceClientModel(
     model_id=MODEL_CHOICES["vlm"][0],
-    token=os.getenv("HUGGINGFACE_API_KEY")
+    token=os.getenv("HUGGINGFACE_API_KEY"),
+    max_tokens=12000
 )
 
 vlm_agent = CodeAgent(
@@ -88,7 +90,8 @@ vlm_agent = CodeAgent(
 
 arithmetic_model = InferenceClientModel(
     model_id=MODEL_CHOICES["arithmetic"][0],
-    token=os.getenv("HUGGINGFACE_API_KEY")
+    token=os.getenv("HUGGINGFACE_API_KEY"),
+    max_tokens=6000
 )
 
 arithmetic_agent = CodeAgent(
@@ -102,7 +105,8 @@ arithmetic_agent = CodeAgent(
 
 pandas_model = InferenceClientModel(
     model_id=MODEL_CHOICES["pandas"][0],
-    token=os.getenv("HUGGINGFACE_API_KEY")
+    token=os.getenv("HUGGINGFACE_API_KEY"),
+    max_tokens=6000
 )
 
 pandas_agent = CodeAgent(
@@ -141,7 +145,8 @@ pandas_agent = CodeAgent(
 
 master_model = InferenceClientModel(
     model_id=MODEL_CHOICES["master"][0],
-    token=os.getenv("HUGGINGFACE_API_KEY")
+    token=os.getenv("HUGGINGFACE_API_KEY"),
+    max_tokens=24000
 )
 
 class MasterAgentWrapper: