bstraehle commited on
Commit
c191089
·
verified ·
1 Parent(s): d8e1981

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +9 -9
agent.py CHANGED
@@ -13,18 +13,18 @@ from smolagents import (
13
 
14
  MODEL_ID = "o4-mini"
15
 
16
- PHOENIX_CLIENT_HEADERS = "api_key = " + os.environ["DESCRIPTION"]
17
  PHOENIX_COLLECTOR_ENDPOINT = "https://app.phoenix.arize.com"
18
 
19
- def run_agent(question, file_name):
20
- tracer_provider = register(
21
- project_name = "gaia",
22
- endpoint = "https://app.phoenix.arize.com/v1/traces"
23
- )
24
 
25
- OpenAIInstrumentor().instrument(tracer_provider = tracer_provider)
26
- SmolagentsInstrumentor().instrument(tracer_provider = tracer_provider)
27
-
 
28
  agent = CodeAgent(
29
  add_base_tools = True,
30
  additional_authorized_imports = ["pandas", "numpy"],
 
13
 
14
  MODEL_ID = "o4-mini"
15
 
16
+ PHOENIX_CLIENT_HEADERS = "api_key = " + os.environ["PHOENIX_API_KEY"]
17
  PHOENIX_COLLECTOR_ENDPOINT = "https://app.phoenix.arize.com"
18
 
19
+ tracer_provider = register(
20
+ project_name = "gaia",
21
+ endpoint = "https://app.phoenix.arize.com/v1/traces"
22
+ )
 
23
 
24
+ #OpenAIInstrumentor().instrument(tracer_provider = tracer_provider)
25
+ SmolagentsInstrumentor().instrument(tracer_provider = tracer_provider)
26
+
27
+ def run_agent(question, file_name):
28
  agent = CodeAgent(
29
  add_base_tools = True,
30
  additional_authorized_imports = ["pandas", "numpy"],