Commit ·
6643df2
1
Parent(s): 5c4f7fa
rollback to maverick
Browse files
agent.py
CHANGED
|
@@ -2,9 +2,6 @@ import os
|
|
| 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:
|
| 9 |
"""Agent that processes questions using LLM and various tools."""
|
| 10 |
|
|
@@ -12,8 +9,8 @@ class TheAgent:
|
|
| 12 |
# Initialize the agent with model and tools
|
| 13 |
self.agent = CodeAgent(
|
| 14 |
model=LiteLLMModel(
|
| 15 |
-
model_id=
|
| 16 |
-
api_key=
|
| 17 |
),
|
| 18 |
tools=[
|
| 19 |
DuckDuckGoSearchTool(),
|
|
|
|
| 2 |
from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, WikipediaSearchTool
|
| 3 |
from tools import VisitWebpageTool, DownloadTaskAttachmentTool
|
| 4 |
|
|
|
|
|
|
|
|
|
|
| 5 |
class TheAgent:
|
| 6 |
"""Agent that processes questions using LLM and various tools."""
|
| 7 |
|
|
|
|
| 9 |
# Initialize the agent with model and tools
|
| 10 |
self.agent = CodeAgent(
|
| 11 |
model=LiteLLMModel(
|
| 12 |
+
model_id="openrouter/meta-llama/llama-4-maverick:free",
|
| 13 |
+
api_key=os.getenv("OPENROUTER_KEY")
|
| 14 |
),
|
| 15 |
tools=[
|
| 16 |
DuckDuckGoSearchTool(),
|