Kackle commited on
Commit
f6f8440
·
verified ·
1 Parent(s): 6b18d3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
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 = genai.GenerativeModel('gemini-2.0-flash-exp')
 
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(