Update crew.py
Browse files
crew.py
CHANGED
@@ -13,10 +13,10 @@ PHOENIX_API_KEY = os.environ["PHOENIX_API_KEY"]
|
|
13 |
os.environ["PHOENIX_CLIENT_HEADERS"] = f"api_key={PHOENIX_API_KEY}"
|
14 |
os.environ["PHOENIX_COLLECTOR_ENDPOINT"] = "https://app.phoenix.arize.com"
|
15 |
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
|
21 |
###
|
22 |
from opentelemetry import trace
|
@@ -27,8 +27,8 @@ from openinference.instrumentation.crewai import CrewAIInstrumentor
|
|
27 |
from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
|
28 |
from opentelemetry.sdk.trace.export import ConsoleSpanExporter, SimpleSpanProcessor
|
29 |
|
30 |
-
endpoint = "https://app.phoenix.arize.com/v1/traces"
|
31 |
-
tracer_provider = TracerProvider()
|
32 |
tracer_provider.add_span_processor(SimpleSpanProcessor(OTLPSpanExporter(endpoint)))
|
33 |
|
34 |
CrewAIInstrumentor().instrument(tracer_provider=tracer_provider)
|
|
|
13 |
os.environ["PHOENIX_CLIENT_HEADERS"] = f"api_key={PHOENIX_API_KEY}"
|
14 |
os.environ["PHOENIX_COLLECTOR_ENDPOINT"] = "https://app.phoenix.arize.com"
|
15 |
|
16 |
+
tracer_provider = register(
|
17 |
+
auto_instrument=True,
|
18 |
+
project_name="gaia"
|
19 |
+
)
|
20 |
|
21 |
###
|
22 |
from opentelemetry import trace
|
|
|
27 |
from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
|
28 |
from opentelemetry.sdk.trace.export import ConsoleSpanExporter, SimpleSpanProcessor
|
29 |
|
30 |
+
#endpoint = "https://app.phoenix.arize.com/v1/traces"
|
31 |
+
#tracer_provider = TracerProvider()
|
32 |
tracer_provider.add_span_processor(SimpleSpanProcessor(OTLPSpanExporter(endpoint)))
|
33 |
|
34 |
CrewAIInstrumentor().instrument(tracer_provider=tracer_provider)
|