Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -76,6 +76,13 @@ class WikipediaSearchTool(Tool):
|
|
| 76 |
|
| 77 |
return f"Wikipedia search unavailable for '{query}'. Use your knowledge to answer."
|
| 78 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 79 |
# --- Basic Agent Definition ---
|
| 80 |
# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
|
| 81 |
class SlpMultiAgent:
|
|
@@ -95,7 +102,8 @@ class SlpMultiAgent:
|
|
| 95 |
api_key = os.getenv('GOOGLE_API_KEY')
|
| 96 |
genai.configure(api_key=api_key)
|
| 97 |
|
| 98 |
-
model
|
|
|
|
| 99 |
|
| 100 |
# Create only essential agents with reduced complexity
|
| 101 |
research_agent = CodeAgent(
|
|
|
|
| 76 |
|
| 77 |
return f"Wikipedia search unavailable for '{query}'. Use your knowledge to answer."
|
| 78 |
|
| 79 |
+
# --- Gemini Model Adapter to patch .generate() for smolagents compatibility ---
|
| 80 |
+
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 ---
|
| 87 |
# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
|
| 88 |
class SlpMultiAgent:
|
|
|
|
| 102 |
api_key = os.getenv('GOOGLE_API_KEY')
|
| 103 |
genai.configure(api_key=api_key)
|
| 104 |
|
| 105 |
+
# Patch: wrap Gemini model for smolagents compatibility
|
| 106 |
+
model = GeminiModelAdapter(genai.GenerativeModel('gemini-2.0-flash-exp'))
|
| 107 |
|
| 108 |
# Create only essential agents with reduced complexity
|
| 109 |
research_agent = CodeAgent(
|