LibertyBird commited on
Commit
6643df2
·
1 Parent(s): 5c4f7fa

rollback to maverick

Browse files
Files changed (1) hide show
  1. agent.py +2 -5
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=MODEL_ID,
16
- api_key=GOOGLE_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(),