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

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +7 -7
agent.py CHANGED
@@ -15,14 +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
- 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"],
 
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
+ SmolagentsInstrumentor().instrument(tracer_provider = tracer_provider)
25
+
26
  agent = CodeAgent(
27
  add_base_tools = True,
28
  additional_authorized_imports = ["pandas", "numpy"],