ABVM commited on
Commit
f280b77
·
verified ·
1 Parent(s): 9e3c1a2

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +5 -19
agent.py CHANGED
@@ -4,19 +4,17 @@ from smolagents import VisitWebpageTool, WebSearchTool, WikipediaSearchTool, Pyt
4
 
5
  class GaiaAgent:
6
  def __init__(self):
 
 
 
7
  self.model = InferenceClientModel(
8
  max_tokens = 2048,
9
  temperature = 0.5,
10
  model_id = "Qwen/Qwen2.5-Coder-32B-Instruct",
11
- custom_role_conversions=None,
12
- #provider = "huggingface", #'https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
13
  #provider="hf-inference"
14
  )
15
- #or
16
- #model=InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct", provider="together")
17
 
18
- #with open("prompts.yaml", 'r') as stream:
19
- # prompt_templates = yaml.safe_load(stream)
20
 
21
  #Creating the agent
22
  self.agent = CodeAgent(
@@ -42,18 +40,6 @@ class GaiaAgent:
42
  except Exception as e:
43
  return f"ERROR: {e}"
44
 
45
- #manager_agent = CodeAgent(
46
- # model=InferenceClientModel("deepseek-ai/DeepSeek-R1", provider="together", max_tokens=8096),
47
- # managed_agent=[web_agent],
48
- # additional_authorized_imports=[
49
- # "datetime",
50
- # "pandas",
51
- # "numpy",
52
- # "json"
53
- # ],
54
- #planning_intervals=5,
55
- #verbosity_level=2,
56
- #max_steps=15,
57
- #)
58
 
59
 
 
4
 
5
  class GaiaAgent:
6
  def __init__(self):
7
+ import os
8
+ token = os.getenv("hf_token")
9
+ print("DEBUG: hf_token is", token[:6]+ "...")
10
  self.model = InferenceClientModel(
11
  max_tokens = 2048,
12
  temperature = 0.5,
13
  model_id = "Qwen/Qwen2.5-Coder-32B-Instruct",
14
+ custom_role_conversions=None,
 
15
  #provider="hf-inference"
16
  )
 
 
17
 
 
 
18
 
19
  #Creating the agent
20
  self.agent = CodeAgent(
 
40
  except Exception as e:
41
  return f"ERROR: {e}"
42
 
43
+
 
 
 
 
 
 
 
 
 
 
 
 
44
 
45