Commit ·
5c4f7fa
1
Parent(s): eaf31fa
gemini model
Browse files
agent.py
CHANGED
|
@@ -2,7 +2,7 @@ import os
|
|
| 2 |
from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, WikipediaSearchTool
|
| 3 |
from tools import VisitWebpageTool, DownloadTaskAttachmentTool
|
| 4 |
|
| 5 |
-
MODEL_ID = "gemini/gemini-2.
|
| 6 |
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
|
| 7 |
|
| 8 |
class TheAgent:
|
|
|
|
| 2 |
from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, WikipediaSearchTool
|
| 3 |
from tools import VisitWebpageTool, DownloadTaskAttachmentTool
|
| 4 |
|
| 5 |
+
MODEL_ID = "gemini/gemini-2.0-flash-exp"
|
| 6 |
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
|
| 7 |
|
| 8 |
class TheAgent:
|