bstraehle commited on
Commit
fc4a651
·
verified ·
1 Parent(s): 4c79380

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +6 -6
agent.py CHANGED
@@ -15,12 +15,12 @@ MODEL_ID = "o4-mini"
15
  PHOENIX_CLIENT_HEADERS = "api_key = " + os.environ["DESCRIPTION"]
16
  PHOENIX_COLLECTOR_ENDPOINT = "https://app.phoenix.arize.com"
17
 
18
- tracer_provider = register(
19
- project_name = "default",
20
- endpoint = "https://app.phoenix.arize.com/v1/traces"
21
- )
22
-
23
  def run_agent(question, file_name):
 
 
 
 
 
24
  agent = CodeAgent(
25
  add_base_tools = True,
26
  additional_authorized_imports = ["pandas", "numpy"],
@@ -31,7 +31,7 @@ def run_agent(question, file_name):
31
  WikipediaSearchTool()],
32
  verbosity_level = 1
33
  )
34
-
35
  return agent.run(question)
36
 
37
  #answer = agent.run(question)
 
15
  PHOENIX_CLIENT_HEADERS = "api_key = " + os.environ["DESCRIPTION"]
16
  PHOENIX_COLLECTOR_ENDPOINT = "https://app.phoenix.arize.com"
17
 
 
 
 
 
 
18
  def run_agent(question, file_name):
19
+ tracer_provider = register(
20
+ project_name = "gaia",
21
+ endpoint = "https://app.phoenix.arize.com/v1/traces"
22
+ )
23
+
24
  agent = CodeAgent(
25
  add_base_tools = True,
26
  additional_authorized_imports = ["pandas", "numpy"],
 
31
  WikipediaSearchTool()],
32
  verbosity_level = 1
33
  )
34
+
35
  return agent.run(question)
36
 
37
  #answer = agent.run(question)