Release

src/langtrace_python_sdk/extensions/langtrace_exporter.py

@@ -90,10 +90,8 @@ def export(self, spans: typing.Sequence[ReadableSpan]) -> SpanExportResult:
                 data=json.dumps(data),
                 headers={"Content-Type": "application/json", "x-api-key": self.api_key},
             )
-            print(f"Traces sent To {LANGTRACE_REMOTE_URL}")
             return SpanExportResult.SUCCESS
         except Exception as e:
-            print("Error sending data to remote URL", e)
             return SpanExportResult.FAILURE
 
     def shutdown(self) -> None:

src/langtrace_python_sdk/instrumentation/anthropic/instrumentation.py

@@ -11,6 +11,10 @@
 
 from langtrace_python_sdk.instrumentation.anthropic.patch import messages_create
 
+import logging
+
+logging.basicConfig(level=logging.FATAL)
+
 
 class AnthropicInstrumentation(BaseInstrumentor):
     """

src/langtrace_python_sdk/instrumentation/chroma/instrumentation.py

@@ -11,6 +11,9 @@
 
 from langtrace_python_sdk.constants.instrumentation.chroma import APIS
 from langtrace_python_sdk.instrumentation.chroma.patch import collection_patch
+import logging
+
+logging.basicConfig(level=logging.FATAL)
 
 
 class ChromaInstrumentation(BaseInstrumentor):

src/langtrace_python_sdk/instrumentation/langchain/instrumentation.py

@@ -12,6 +12,10 @@
 
 from langtrace_python_sdk.instrumentation.langchain.patch import generic_patch
 
+import logging
+
+logging.basicConfig(level=logging.FATAL)
+
 
 def patch_module_classes(
     module_name, tracer, version, task, trace_output=True, trace_input=True

src/langtrace_python_sdk/instrumentation/llamaindex/instrumentation.py

@@ -12,6 +12,10 @@
 
 from langtrace_python_sdk.instrumentation.llamaindex.patch import generic_patch
 
+import logging
+
+logging.basicConfig(level=logging.FATAL)
+
 
 class LlamaindexInstrumentation(BaseInstrumentor):
     """

src/langtrace_python_sdk/instrumentation/openai/instrumentation.py

@@ -1,7 +1,6 @@
 import importlib.metadata
 from typing import Collection
 
-import openai
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 from opentelemetry.trace import get_tracer
 from wrapt import wrap_function_wrapper
@@ -12,6 +11,10 @@
     images_generate,
 )
 
+import logging
+
+logging.basicConfig(level=logging.FATAL)
+
 
 class OpenAIInstrumentation(BaseInstrumentor):
 

src/langtrace_python_sdk/instrumentation/openai/patch.py

@@ -133,7 +133,7 @@ def traced_method(wrapped, instance, args, kwargs):
         try:
             # Attempt to call the original method
             result = wrapped(*args, **kwargs)
-            if kwargs.get("stream") is False:
+            if kwargs.get("stream") is False or kwargs.get("stream") is None:
                 span.set_attribute("llm.model", result.model)
                 if hasattr(result, "choices") and result.choices is not None:
                     responses = [
@@ -338,8 +338,6 @@ def traced_method(wrapped, instance, args, kwargs):
             APIS["EMBEDDINGS_CREATE"]["METHOD"], kind=SpanKind.CLIENT
         ) as span:
 
-            print("Inside embeddings_create", trace.get_current_span())
-
             for field, value in attributes.model_dump(by_alias=True).items():
                 if value is not None:
                     span.set_attribute(field, value)

src/langtrace_python_sdk/instrumentation/pinecone/instrumentation.py

@@ -14,6 +14,10 @@
 from langtrace_python_sdk.constants.instrumentation.pinecone import APIS
 from langtrace_python_sdk.instrumentation.pinecone.patch import generic_patch
 
+import logging
+
+logging.basicConfig(level=logging.FATAL)
+
 
 class PineconeInstrumentation(BaseInstrumentor):
     """