jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

revert: Traces Export Headers For Arize & Phoenix Integration (#7825)

Browse source 
Co-authored-by: Eric Hare <ericrhare@gmail.com>
Co-authored-by: Lucas Oliveira <62335616+lucaseduoli@users.noreply.github.com>
This commit is contained in:
Ali Saleh 2025-04-29 04:34:31 +05:00 • committed by GitHub
commit 93e8b3e8c5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 15 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

28
src/backend/base/langflow/services/tracing/arize_phoenix.py
View file

@ -102,8 +102,11 @@ class ArizePhoenixTracer(BaseTracer):
arize_collector_endpoint = os.getenv("ARIZE_COLLECTOR_ENDPOINT", "https://otlp.arize.com") arize_collector_endpoint = os.getenv("ARIZE_COLLECTOR_ENDPOINT", "https://otlp.arize.com")
enable_arize_tracing = bool(arize_api_key and arize_space_id) enable_arize_tracing = bool(arize_api_key and arize_space_id)
arize_endpoint = f"{arize_collector_endpoint}/v1" arize_endpoint = f"{arize_collector_endpoint}/v1"
if enable_arize_tracing: arize_headers = {
os.environ["OTEL_EXPORTER_OTLP_TRACES_HEADERS"] = f"api_key={arize_api_key},space_id={arize_space_id}" "api_key": arize_api_key,
"space_id": arize_space_id,
"authorization": f"Bearer {arize_api_key}",
}
# Phoenix Config # Phoenix Config
phoenix_api_key = os.getenv("PHOENIX_API_KEY", None) phoenix_api_key = os.getenv("PHOENIX_API_KEY", None)
@ -119,28 +122,27 @@ class ArizePhoenixTracer(BaseTracer):
return False return False
try: try:
from openinference.instrumentation import TracerProvider from phoenix.otel import (
from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import ( PROJECT_NAME,
OTLPSpanExporter as GRPCSpanExporter, BatchSpanProcessor,
GRPCSpanExporter,
HTTPSpanExporter,
Resource,
SimpleSpanProcessor,
TracerProvider,
) )
from opentelemetry.exporter.otlp.proto.http.trace_exporter import (
OTLPSpanExporter as HTTPSpanExporter,
)
from opentelemetry.sdk.resources import Resource
from opentelemetry.sdk.trace.export import BatchSpanProcessor, SimpleSpanProcessor
from phoenix.otel import PROJECT_NAME
name_without_space = self.flow_name.replace(" ", "-") name_without_space = self.flow_name.replace(" ", "-")
project_name = self.project_name if name_without_space == "None" else name_without_space project_name = self.project_name if name_without_space == "None" else name_without_space
attributes = {PROJECT_NAME: project_name, "model_id": project_name} attributes = {PROJECT_NAME: project_name, "model_id": project_name}
resource = Resource.create(attributes=attributes) resource = Resource.create(attributes=attributes)
tracer_provider = TracerProvider(resource=resource) tracer_provider = TracerProvider(resource=resource, verbose=False)
span_processor = BatchSpanProcessor if arize_phoenix_batch else SimpleSpanProcessor span_processor = BatchSpanProcessor if arize_phoenix_batch else SimpleSpanProcessor
if enable_arize_tracing: if enable_arize_tracing:
tracer_provider.add_span_processor( tracer_provider.add_span_processor(
span_processor=span_processor( span_processor=span_processor(
span_exporter=GRPCSpanExporter(endpoint=arize_endpoint), span_exporter=GRPCSpanExporter(endpoint=arize_endpoint, headers=arize_headers),
) )
) )