Kackle commited on
Commit
d74213e
·
verified ·
1 Parent(s): 788978f

last attempt with smol then back to llama

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -81,6 +81,7 @@ class GeminiModelAdapter:
81
  def __init__(self, model):
82
  self.model = model
83
  def generate(self, *args, **kwargs):
 
84
  return self.model.generate_content(*args, **kwargs)
85
 
86
  # --- Basic Agent Definition ---
@@ -135,7 +136,13 @@ class SlpMultiAgent:
135
  tools=[KnowledgeBaseTool(), WikipediaSearchTool(), DuckDuckGoSearchTool()], # Pass all retrieval tools
136
  managed_agents=[research_agent, solver_agent], # Only 2 agents
137
  name="ManagerAgent",
138
- description="Efficient manager for quick problem solving. Uses Wikipedia and DuckDuckGo for factual questions.",
 
 
 
 
 
 
139
  additional_authorized_imports=["re", "math"],
140
  planning_interval=1, # Faster planning
141
  verbosity_level=0, # Reduce verbosity
 
81
  def __init__(self, model):
82
  self.model = model
83
  def generate(self, *args, **kwargs):
84
+ kwargs.pop('stop_sequences', None) # Remove unsupported argument for Gemini
85
  return self.model.generate_content(*args, **kwargs)
86
 
87
  # --- Basic Agent Definition ---
 
136
  tools=[KnowledgeBaseTool(), WikipediaSearchTool(), DuckDuckGoSearchTool()], # Pass all retrieval tools
137
  managed_agents=[research_agent, solver_agent], # Only 2 agents
138
  name="ManagerAgent",
139
+ description=(
140
+ "Efficient manager for quick problem solving. "
141
+ "Uses Wikipedia and DuckDuckGo for factual questions. "
142
+ "IMPORTANT: Only output a single code block in this format: "
143
+ "<code>\nfinal_answer(\"your direct, simple answer\")\n</code> "
144
+ "Do not include any other code, comments, or explanations."
145
+ ),
146
  additional_authorized_imports=["re", "math"],
147
  planning_interval=1, # Faster planning
148
  verbosity_level=0, # Reduce verbosity