RozzaCreat commited on
Commit
b3fb03d
·
verified ·
1 Parent(s): e283efe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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("HF_TOKEN")
26
- print(f"[BasicAgent] HF_TOKEN loaded: {bool(token)}, length: {len(token) if token else 0}")
27
  self.agent = CodeAgent(
28
  tools=[DuckDuckGoSearchTool(), WikipediaSearchTool()],
29
- model=InferenceClientModel(
30
- model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
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,