Spaces:
Paused
Paused
Update agent.py
Browse files
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 |
-
|
| 17 |
-
|
| 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 = "
|
| 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):
|