langtrace-python-sdk 3.8.10__tar.gz → 3.8.12__tar.gz

{langtrace_python_sdk-3.8.10 → langtrace_python_sdk-3.8.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langtrace-python-sdk
-Version: 3.8.10
+Version: 3.8.12
 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-3.8.10 → langtrace_python_sdk-3.8.12}/src/langtrace_python_sdk/instrumentation/cohere/patch.py

@@ -87,25 +87,25 @@ def rerank(original_method, version, tracer, v2=False):
                     and result.meta.billed_units is not None
                 ):
                     usage = result.meta.billed_units
-                    if usage is not None:
-                        span.set_attribute(
-                            SpanAttributes.LLM_USAGE_PROMPT_TOKENS,
-                            usage.input_tokens or 0,
-                        )
-                        span.set_attribute(
-                            SpanAttributes.LLM_USAGE_COMPLETION_TOKENS,
-                            usage.output_tokens or 0,
-                        )
-
-                        span.set_attribute(
-                            SpanAttributes.LLM_USAGE_TOTAL_TOKENS,
-                            (usage.input_tokens or 0) + (usage.output_tokens or 0),
-                        )
-
-                        span.set_attribute(
-                            "search_units",
-                            usage.search_units or 0,
-                        )
+                    input_tokens = int(usage.input_tokens) if usage.input_tokens else 0
+                    output_tokens = int(usage.output_tokens) if usage.output_tokens else 0
+                    span.set_attribute(
+                        SpanAttributes.LLM_USAGE_PROMPT_TOKENS,
+                        input_tokens,
+                    )
+                    span.set_attribute(
+                        SpanAttributes.LLM_USAGE_COMPLETION_TOKENS,
+                        output_tokens,
+                    )
+                    span.set_attribute(
+                        SpanAttributes.LLM_USAGE_TOTAL_TOKENS,
+                        input_tokens + output_tokens,
+                    )
+                    span.set_attribute(
+                        "search_units",
+                        int(usage.search_units) if usage.search_units else 0,
+                    )
+                        
 
             span.set_status(StatusCode.OK)
             span.end()
@@ -309,25 +309,25 @@ def chat_create(original_method, version, tracer):
                         and result.meta.billed_units is not None
                     ):
                         usage = result.meta.billed_units
-                        if usage is not None:
-                            span.set_attribute(
-                                SpanAttributes.LLM_USAGE_PROMPT_TOKENS,
-                                usage.input_tokens or 0,
-                            )
-                            span.set_attribute(
-                                SpanAttributes.LLM_USAGE_COMPLETION_TOKENS,
-                                usage.output_tokens or 0,
-                            )
-
-                            span.set_attribute(
-                                SpanAttributes.LLM_USAGE_TOTAL_TOKENS,
-                                (usage.input_tokens or 0) + (usage.output_tokens or 0),
-                            )
-
-                            span.set_attribute(
-                                "search_units",
-                                usage.search_units or 0,
-                            )
+                        input_tokens = int(usage.input_tokens) if usage.input_tokens else 0
+                        output_tokens = int(usage.output_tokens) if usage.output_tokens else 0
+                        span.set_attribute(
+                            SpanAttributes.LLM_USAGE_PROMPT_TOKENS,
+                            input_tokens,
+                        )
+                        span.set_attribute(
+                            SpanAttributes.LLM_USAGE_COMPLETION_TOKENS,
+                            output_tokens,
+                        )
+                        span.set_attribute(
+                            SpanAttributes.LLM_USAGE_TOTAL_TOKENS,
+                            input_tokens + output_tokens,
+                        )
+                        span.set_attribute(
+                            "search_units",
+                            int(usage.search_units) if usage.search_units else 0,
+                        )
+                            
                 span.set_status(StatusCode.OK)
                 span.end()
                 return result
@@ -419,10 +419,12 @@ def chat_create_v2(original_method, version, tracer, stream=False):
                     if (hasattr(result.usage, "billed_units") and 
                         result.usage.billed_units is not None):
                         usage = result.usage.billed_units
+                        input_tokens = int(usage.input_tokens) if usage.input_tokens else 0
+                        output_tokens = int(usage.output_tokens) if usage.output_tokens else 0
                         for metric, value in {
-                            "input": usage.input_tokens or 0,
-                            "output": usage.output_tokens or 0,
-                            "total": (usage.input_tokens or 0) + (usage.output_tokens or 0),
+                            "input": input_tokens,
+                            "output": output_tokens,
+                            "total": input_tokens + output_tokens,
                         }.items():
                             span.set_attribute(
                                 f"gen_ai.usage.{metric}_tokens",
@@ -571,26 +573,27 @@ def chat_stream(original_method, version, tracer):
                                 and response.meta.billed_units is not None
                             ):
                                 usage = response.meta.billed_units
-                                if usage is not None:
-                                    span.set_attribute(
-                                        SpanAttributes.LLM_USAGE_PROMPT_TOKENS,
-                                        usage.input_tokens or 0,
-                                    )
-                                    span.set_attribute(
-                                        SpanAttributes.LLM_USAGE_COMPLETION_TOKENS,
-                                        usage.output_tokens or 0,
-                                    )
-
+                                input_tokens = int(usage.input_tokens) if usage.input_tokens else 0
+                                output_tokens = int(usage.output_tokens) if usage.output_tokens else 0
+                                span.set_attribute(
+                                    SpanAttributes.LLM_USAGE_PROMPT_TOKENS,
+                                    input_tokens,
+                                )
+                                span.set_attribute(
+                                    SpanAttributes.LLM_USAGE_COMPLETION_TOKENS,
+                                    output_tokens,
+                                )
+                                span.set_attribute(
+                                    SpanAttributes.LLM_USAGE_TOTAL_TOKENS,
+                                    input_tokens + output_tokens,
+                                )
+
+                                if usage.search_units is not None:
                                     span.set_attribute(
-                                        SpanAttributes.LLM_USAGE_TOTAL_TOKENS,
-                                        (usage.input_tokens or 0)
-                                        + (usage.output_tokens or 0),
+                                        "search_units",
+                                        int(usage.search_units) if usage.search_units else 0,
                                     )
-                                    if usage.search_units is not None:
-                                        span.set_attribute(
-                                            "search_units",
-                                            usage.search_units or 0,
-                                        )
+                                    
 
                     yield event
             finally:

{langtrace_python_sdk-3.8.10 → langtrace_python_sdk-3.8.12}/src/langtrace_python_sdk/instrumentation/neo4j/patch.py

@@ -37,7 +37,11 @@ from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME
 
 def driver_patch(operation_name, version, tracer):
     def traced_method(wrapped, instance, args, kwargs):
-        
+        try:
+            query = args[0].text if hasattr(args[0], "text") else args[0]
+            query_text = json.dumps(query)
+        except (AttributeError, TypeError):
+            query_text = args[0]
         api = APIS[operation_name]
         service_provider = SERVICE_PROVIDERS.get("NEO4J", "neo4j")
         extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
@@ -49,7 +53,7 @@ def driver_patch(operation_name, version, tracer):
             "langtrace.version": v(LANGTRACE_SDK_NAME),
             "db.system": "neo4j",
             "db.operation": api["OPERATION"],
-            "db.query": json.dumps(args[0]) if args and len(args) > 0 else "",
+            "db.query": query_text,
             **(extra_attributes if extra_attributes is not None else {}),
         }
         

{langtrace_python_sdk-3.8.10 → langtrace_python_sdk-3.8.12}/src/langtrace_python_sdk/instrumentation/openai/patch.py

@@ -114,6 +114,9 @@ def openai_responses_create(version: str, tracer: Tracer) -> Callable:
                     return StreamWrapper(response, span)
                 else:
                     _set_openai_agentic_response_attributes(span, response)
+                    
+                span.set_status(StatusCode.OK)
+                span.end()
                 return response
             except Exception as err:
                 span.record_exception(err)
@@ -738,7 +741,7 @@ def _set_openai_agentic_response_attributes(span: Span, response) -> None:
             "input_tokens": response.usage.input_tokens,
             "output_tokens": response.usage.output_tokens,
             "total_tokens": response.usage.total_tokens,
-            "cached_tokens": response.usage.input_tokens_details["cached_tokens"],
+            "cached_tokens": response.usage.input_tokens_details.cached_tokens,
         },
     )
 

{langtrace_python_sdk-3.8.10 → langtrace_python_sdk-3.8.12}/src/langtrace_python_sdk/langtrace.py

@@ -299,7 +299,8 @@ def init(
     init_instrumentations(config.disable_instrumentations, all_instrumentations)
     add_span_processor(provider, config, exporter)
 
-    sys.stdout = sys.__stdout__
+    if config.disable_logging:
+        sys.stdout = sys.__stdout__
     init_sentry(config, host)
 
 

{langtrace_python_sdk-3.8.10 → langtrace_python_sdk-3.8.12}/src/langtrace_python_sdk/utils/llm.py

@@ -347,19 +347,19 @@ def set_usage_attributes(span, usage):
     set_span_attribute(
         span,
         SpanAttributes.LLM_USAGE_PROMPT_TOKENS,
-        input_tokens,
+        int(input_tokens),
     )
 
     set_span_attribute(
         span,
         SpanAttributes.LLM_USAGE_COMPLETION_TOKENS,
-        output_tokens,
+        int(output_tokens),
     )
 
     set_span_attribute(
         span,
         SpanAttributes.LLM_USAGE_TOTAL_TOKENS,
-        input_tokens + output_tokens,
+        int(input_tokens) + int(output_tokens),
     )
 
     if "search_units" in usage:

langtrace_python_sdk-3.8.12/src/langtrace_python_sdk/version.py

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

langtrace_python_sdk-3.8.10/src/langtrace_python_sdk/version.py

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