Spaces:
Sleeping
Sleeping
last attempt with smol then back to llama
Browse files
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=
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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
|