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

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +7 -5
agent.py CHANGED
@@ -15,12 +15,14 @@ MODEL_ID = "o4-mini"
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"],
 
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 = "gaia",
20
+ endpoint = "https://app.phoenix.arize.com/v1/traces"
21
+ )
22
+
23
+ SmolagentsInstrumentor().instrument(tracer_provider = tracer_provider)
24
 
25
+ def run_agent(question, file_name):
26
  agent = CodeAgent(
27
  add_base_tools = True,
28
  additional_authorized_imports = ["pandas", "numpy"],