bstraehle commited on
Commit
0bf6228
·
verified ·
1 Parent(s): b0cfc6d

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -10
agent.py CHANGED
@@ -1,7 +1,6 @@
1
  import os
2
 
3
  from openai import OpenAI
4
- from openinference.instrumentation.openai import OpenAIInstrumentor
5
  from openinference.instrumentation.smolagents import SmolagentsInstrumentor
6
  from phoenix.otel import register
7
  from smolagents import (
@@ -13,21 +12,15 @@ from smolagents import (
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
- ###
20
- os.environ["OTEL_EXPORTER_OTLP_HEADERS"] = PHOENIX_CLIENT_HEADERS;
21
- os.environ["PHOENIX_CLIENT_HEADERS"] = PHOENIX_CLIENT_HEADERS;
22
  os.environ["PHOENIX_COLLECTOR_ENDPOINT"] = "https://app.phoenix.arize.com";
23
- ###
24
 
25
  tracer_provider = register(
26
- project_name = "default",
27
  endpoint = "https://app.phoenix.arize.com/v1/traces"
28
  )
29
 
30
- #OpenAIInstrumentor().instrument(tracer_provider = tracer_provider)
31
  SmolagentsInstrumentor().instrument(tracer_provider = tracer_provider)
32
 
33
  def run_agent(question, file_name):
 
1
  import os
2
 
3
  from openai import OpenAI
 
4
  from openinference.instrumentation.smolagents import SmolagentsInstrumentor
5
  from phoenix.otel import register
6
  from smolagents import (
 
12
 
13
  MODEL_ID = "o4-mini"
14
 
15
+ os.environ["OTEL_EXPORTER_OTLP_HEADERS"] = "api_key = " + os.environ["PHOENIX_API_KEY"];
16
+ os.environ["PHOENIX_CLIENT_HEADERS"] = "api_key = " + os.environ["PHOENIX_API_KEY"];
 
 
 
 
17
  os.environ["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
  SmolagentsInstrumentor().instrument(tracer_provider = tracer_provider)
25
 
26
  def run_agent(question, file_name):