langtrace-python-sdk 2.2.22__tar.gz → 2.2.24__tar.gz

{langtrace_python_sdk-2.2.22 → langtrace_python_sdk-2.2.24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 2.2.22
+Version: 2.2.24
 Summary: Python SDK for LangTrace
 Project-URL: Homepage, https://github.com/Scale3-Labs/langtrace-python-sdk
 Author-email: Scale3 Labs <engineering@scale3labs.com>

{langtrace_python_sdk-2.2.22 → langtrace_python_sdk-2.2.24}/src/langtrace_python_sdk/extensions/langtrace_exporter.py

@@ -1,6 +1,7 @@
 import json
 import os
 import typing
+import sys
 
 import requests
 from opentelemetry.sdk.trace.export import ReadableSpan, SpanExporter, SpanExportResult
@@ -49,11 +50,13 @@ class LangTraceExporter(SpanExporter):
 
     api_key: str
     api_host: str
+    disable_logging: bool
 
     def __init__(
         self,
         api_host,
         api_key: str = None,
+        disable_logging: bool = False,
     ) -> None:
         self.api_key = api_key or os.environ.get("LANGTRACE_API_KEY")
         self.api_host = (
@@ -61,6 +64,7 @@ class LangTraceExporter(SpanExporter):
             if api_host == LANGTRACE_REMOTE_URL
             else api_host
         )
+        self.disable_logging = disable_logging
 
     def export(self, spans: typing.Sequence[ReadableSpan]) -> SpanExportResult:
         """
@@ -72,7 +76,7 @@ class LangTraceExporter(SpanExporter):
         Returns:
             The result of the export SUCCESS or FAILURE
         """
-        if not self.api_key:
+        if not self.api_key and not self.disable_logging:
             print(Fore.RED)
             print(
                 "Missing Langtrace API key, proceed to https://langtrace.ai to create one"
@@ -107,14 +111,15 @@ class LangTraceExporter(SpanExporter):
 
             if not response.ok:
                 raise RequestException(response.text)
-
-            print(
-                Fore.GREEN + f"Exported {len(spans)} spans successfully." + Fore.RESET
-            )
+            if not self.disable_logging:
+                print(
+                    Fore.GREEN + f"Exported {len(spans)} spans successfully." + Fore.RESET
+                )
             return SpanExportResult.SUCCESS
         except RequestException as err:
-            print(Fore.RED + "Failed to export spans.")
-            print(Fore.RED + f"Error: {err}" + Fore.RESET)
+            if not self.disable_logging:
+                print(Fore.RED + "Failed to export spans.")
+                print(Fore.RED + f"Error: {err}" + Fore.RESET)
             return SpanExportResult.FAILURE
 
     def shutdown(self) -> None:

{langtrace_python_sdk-2.2.22 → langtrace_python_sdk-2.2.24}/src/langtrace_python_sdk/instrumentation/cohere/patch.py

@@ -23,7 +23,6 @@ from langtrace_python_sdk.utils.llm import (
     get_llm_url,
     get_span_name,
     set_event_completion,
-    set_event_completion_chunk,
     set_usage_attributes,
 )
 from langtrace.trace_attributes import Event, LLMSpanAttributes

{langtrace_python_sdk-2.2.22 → langtrace_python_sdk-2.2.24}/src/langtrace_python_sdk/instrumentation/gemini/patch.py

@@ -13,7 +13,6 @@ from langtrace_python_sdk.utils.llm import (
     get_span_name,
     is_streaming,
     set_event_completion,
-    set_event_completion_chunk,
     set_span_attributes,
     set_usage_attributes,
 )
@@ -156,7 +155,6 @@ def build_streaming_response(span, response):
         item_to_yield = item
         complete_response += str(item.text)
         yield item_to_yield
-        set_event_completion_chunk(span, item.text)
         if hasattr(item, "usage_metadata"):
             usage = item.usage_metadata
             input_tokens = usage.prompt_token_count
@@ -176,7 +174,6 @@ async def abuild_streaming_response(span, response):
         item_to_yield = item
         complete_response += str(item.text)
         yield item_to_yield
-        set_event_completion_chunk(span, item.text)
         if hasattr(item, "usage_metadata"):
             usage = item.usage_metadata
             input_tokens = usage.prompt_token_count

{langtrace_python_sdk-2.2.22 → langtrace_python_sdk-2.2.24}/src/langtrace_python_sdk/instrumentation/groq/patch.py

@@ -31,7 +31,6 @@ from langtrace_python_sdk.utils.llm import (
     get_langtrace_attributes,
     get_span_name,
     set_event_completion,
-    set_event_completion_chunk,
     set_usage_attributes,
 )
 from langtrace_python_sdk.constants.instrumentation.common import (
@@ -245,14 +244,6 @@ def chat_completions_create(original_method, version, tracer):
                 else:
                     content = []
 
-                set_event_completion_chunk(
-                    span,
-                    (
-                        "".join(content)
-                        if len(content) > 0 and content[0] is not None
-                        else ""
-                    ),
-                )
                 result_content.append(content[0] if len(content) > 0 else "")
                 yield chunk
         finally:

{langtrace_python_sdk-2.2.22 → langtrace_python_sdk-2.2.24}/src/langtrace_python_sdk/instrumentation/ollama/patch.py

@@ -8,7 +8,6 @@ from langtrace_python_sdk.utils.llm import (
     get_llm_url,
     get_span_name,
     set_event_completion,
-    set_event_completion_chunk,
 )
 from langtrace_python_sdk.utils.silently_fail import silently_fail
 from langtrace_python_sdk.constants.instrumentation.common import SERVICE_PROVIDERS

{langtrace_python_sdk-2.2.22 → langtrace_python_sdk-2.2.24}/src/langtrace_python_sdk/langtrace.py

@@ -72,7 +72,10 @@ def init(
     disable_instrumentations: Optional[DisableInstrumentations] = None,
     disable_tracing_for_functions: Optional[InstrumentationMethods] = None,
     service_name: Optional[str] = None,
+    disable_logging = False
 ):
+    if disable_logging:
+        sys.stdout = open(os.devnull, "w")
 
     host = (
         os.environ.get("LANGTRACE_API_HOST", None) or api_host or LANGTRACE_REMOTE_URL
@@ -90,7 +93,7 @@ def init(
     provider = TracerProvider(resource=resource, sampler=sampler)
 
     remote_write_exporter = (
-        LangTraceExporter(api_key=api_key, api_host=host)
+        LangTraceExporter(api_key=api_key, api_host=host, disable_logging=disable_logging)
         if custom_remote_exporter is None
         else custom_remote_exporter
     )
@@ -146,6 +149,8 @@ def init(
         print(Fore.BLUE + "Exporting spans to Langtrace cloud.." + Fore.RESET)
         provider.add_span_processor(batch_processor_remote)
 
+    sys.stdout = sys.__stdout__
+
 
 def init_instrumentations(
     disable_instrumentations: DisableInstrumentations, all_instrumentations: dict

langtrace_python_sdk-2.2.24/src/langtrace_python_sdk/version.py

@@ -0,0 +1 @@
+__version__ = "2.2.24"

langtrace_python_sdk-2.2.22/src/langtrace_python_sdk/version.py

@@ -1 +0,0 @@
-__version__ = "2.2.22"