Update app.py
Browse files
app.py
CHANGED
|
@@ -22,12 +22,12 @@ If you are asked for a comma separated list, apply the above rules depending on
|
|
| 22 |
|
| 23 |
class BasicAgent:
|
| 24 |
def __init__(self):
|
| 25 |
-
token = os.getenv("
|
| 26 |
-
print(f"[BasicAgent]
|
| 27 |
self.agent = CodeAgent(
|
| 28 |
tools=[DuckDuckGoSearchTool(), WikipediaSearchTool()],
|
| 29 |
-
model=
|
| 30 |
-
model_id="
|
| 31 |
token=token,
|
| 32 |
),
|
| 33 |
max_steps=12,
|
|
|
|
| 22 |
|
| 23 |
class BasicAgent:
|
| 24 |
def __init__(self):
|
| 25 |
+
token = os.getenv("GEMINI_API_KEY")
|
| 26 |
+
print(f"[BasicAgent] GEMINI_API_KEY loaded: {bool(api_key)}")
|
| 27 |
self.agent = CodeAgent(
|
| 28 |
tools=[DuckDuckGoSearchTool(), WikipediaSearchTool()],
|
| 29 |
+
model=LiteLLMModel(
|
| 30 |
+
model_id="gemini/gemini-2.0-flash",
|
| 31 |
token=token,
|
| 32 |
),
|
| 33 |
max_steps=12,
|