clean up

Arize-ai · Jan 25, 2024 · f50d0b5 · f50d0b5
1 parent dcdc3f3
commit f50d0b5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 11 deletions.
diff --git a/...on/instrumentation/openinference-instrumentation-langchain/examples/openai_chat_stream.py b/...on/instrumentation/openinference-instrumentation-langchain/examples/openai_chat_stream.py
@@ -1,5 +1,3 @@
-import os
-
 from langchain_openai import ChatOpenAI
 from openinference.instrumentation.langchain import LangChainInstrumentor
 from opentelemetry import trace as trace_api
@@ -11,15 +9,10 @@
 resource = Resource(attributes={})
 tracer_provider = trace_sdk.TracerProvider(resource=resource)
 trace_api.set_tracer_provider(tracer_provider=tracer_provider)
-
-# Logs to the Phoenix Collector if running locally
-if os.environ.get("PHOENIX_COLLECTOR_ENDPOINT"):
-    endpoint = os.environ["PHOENIX_COLLECTOR_ENDPOINT"] + "/v1/traces"
-    span_otlp_exporter = OTLPSpanExporter(endpoint=endpoint)
-    tracer_provider.add_span_processor(SimpleSpanProcessor(span_exporter=span_otlp_exporter))
-else:
-    span_console_exporter = ConsoleSpanExporter()
-    tracer_provider.add_span_processor(SimpleSpanProcessor(span_exporter=span_console_exporter))
+span_otlp_exporter = OTLPSpanExporter(endpoint="http://127.0.0.1:6006/v1/traces")
+tracer_provider.add_span_processor(SimpleSpanProcessor(span_exporter=span_otlp_exporter))
+span_console_exporter = ConsoleSpanExporter()
+tracer_provider.add_span_processor(SimpleSpanProcessor(span_exporter=span_console_exporter))
 
 LangChainInstrumentor().instrument()
 

diff --git a/python/instrumentation/openinference-instrumentation-langchain/examples/requirements.txt b/python/instrumentation/openinference-instrumentation-langchain/examples/requirements.txt
@@ -0,0 +1,4 @@
+langchain_openai
+opentelemetry-sdk
+opentelemetry-exporter-otlp
+openinference-instrumentation-langchain