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)
|