langtrace-python-sdk 3.8.10__py3-none-any.whl → 3.8.11__py3-none-any.whl

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/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/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/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/version.py

@@ -1 +1 @@
-__version__ = "3.8.10"
+__version__ = "3.8.11"

{langtrace_python_sdk-3.8.10.dist-info → langtrace_python_sdk-3.8.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langtrace-python-sdk
-Version: 3.8.10
+Version: 3.8.11
 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.dist-info → langtrace_python_sdk-3.8.11.dist-info}/RECORD

@@ -120,8 +120,8 @@ examples/vertexai_example/main.py,sha256=gndId5X5ksD-ycxnAWMdEqIDbLc3kz5Vt8vm4YP
 examples/weaviate_example/__init__.py,sha256=8JMDBsRSEV10HfTd-YC7xb4txBjD3la56snk-Bbg2Kw,618
 examples/weaviate_example/query_text.py,sha256=wPHQTc_58kPoKTZMygVjTj-2ZcdrIuaausJfMxNQnQc,127162
 langtrace_python_sdk/__init__.py,sha256=VZM6i71NR7pBQK6XvJWRelknuTYUhqwqE7PlicKa5Wg,1166
-langtrace_python_sdk/langtrace.py,sha256=P7aHGunNP8YQqd2glsVLXhSe4QDeAarfG2eCV9hhAm4,13944
-langtrace_python_sdk/version.py,sha256=Sz-5SfPXsiZbJO0dp0pVWeIm92HyINLVnoY9zyB3FQI,23
+langtrace_python_sdk/langtrace.py,sha256=mTCx5nyBbXNINr0oVWvMZ9iU47MWhRg2DBel2gwlnLg,13979
+langtrace_python_sdk/version.py,sha256=SOTowMrQm37hhBSrXviQdSFjTgt0PPZQX6q3WhiUwNY,23
 langtrace_python_sdk/constants/__init__.py,sha256=3CNYkWMdd1DrkGqzLUgNZXjdAlM6UFMlf_F-odAToyc,146
 langtrace_python_sdk/constants/exporter/langtrace_exporter.py,sha256=EVCrouYCpY98f0KSaKr4PzNxPULTZZO6dSA_crEOyJU,106
 langtrace_python_sdk/constants/instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -174,7 +174,7 @@ langtrace_python_sdk/instrumentation/cleanlab/instrumentation.py,sha256=CtWdtDUC
 langtrace_python_sdk/instrumentation/cleanlab/patch.py,sha256=aI0QfG9UH5P7OvmzeIdeMhRd28oV4JMuXIkPv8s869s,3138
 langtrace_python_sdk/instrumentation/cohere/__init__.py,sha256=sGUSLdTUyYf36Tm6L5jQflhzCqvmWrhnBOMYHjvp6Hs,95
 langtrace_python_sdk/instrumentation/cohere/instrumentation.py,sha256=1wxMhWMfsvKprdV52BIfCZhZS1beRYBW9rUzUDDkyCk,2854
-langtrace_python_sdk/instrumentation/cohere/patch.py,sha256=q29gJnik8bnJ7fnwaJ8PArHTQbJkWhhwBfDdQOTRBng,25422
+langtrace_python_sdk/instrumentation/cohere/patch.py,sha256=qeXm4ycnXfDO7lxWrpA2GaUIvGTkLcSMiq9-xMtlP6U,25782
 langtrace_python_sdk/instrumentation/crewai/__init__.py,sha256=_UBKfvQv7l0g2_wnmA5F6CdSAFH0atNOVPd49zsN3aM,88
 langtrace_python_sdk/instrumentation/crewai/instrumentation.py,sha256=5Umzq8zjEnMEtjZZiMB4DQOPkxZ-1vts7RKC6JWpn24,2969
 langtrace_python_sdk/instrumentation/crewai/patch.py,sha256=VoyOtGKYzaOIu7UnVNTemZeB3LrCIodrrYwmXLdxRw8,9133
@@ -235,7 +235,7 @@ langtrace_python_sdk/instrumentation/ollama/instrumentation.py,sha256=jdsvkqUJAA
 langtrace_python_sdk/instrumentation/ollama/patch.py,sha256=w99r9wCCVDdJnZQEezYE2lW_iNFEtrldt9vq3ISAsag,5375
 langtrace_python_sdk/instrumentation/openai/__init__.py,sha256=VPHRNCQEdkizIVP2d0Uw_a7t8XOTSTprEIB8oboJFbs,95
 langtrace_python_sdk/instrumentation/openai/instrumentation.py,sha256=kvbRr5DrcP23ka4mLZpwr7W0ZxeTGHilkOlc_h5V5s8,3226
-langtrace_python_sdk/instrumentation/openai/patch.py,sha256=I_Gzp8ULbOQzlG9PltrembOiBvfp0FQHDRwR5q5tmZg,31358
+langtrace_python_sdk/instrumentation/openai/patch.py,sha256=o3OB8UfZr6kZo2uBfbXsTRyuv1vWBAsRlcB3DRcjN6s,31450
 langtrace_python_sdk/instrumentation/openai/types.py,sha256=aVkoa7tmAbYfyOhnyMrDaVjQuwhmRNLMthlNtKMtWX8,4311
 langtrace_python_sdk/instrumentation/openai_agents/__init__.py,sha256=ElRfFIQYXD2-eRyL3fZnjIsDJLTrDolh5cZHPnZv0q8,107
 langtrace_python_sdk/instrumentation/openai_agents/instrumentation.py,sha256=6M4FHXfem7pazrNgsimebrEfMb2FxI8lHrdEMbVf75Y,1860
@@ -261,7 +261,7 @@ langtrace_python_sdk/instrumentation/weaviate/patch.py,sha256=igqYUR_bEjs1XtyyZ8
 langtrace_python_sdk/types/__init__.py,sha256=SJSJzkgPjGGTVJXUZ_FyR3p9DJ5kWGx7iAnJfY4ZYHU,4669
 langtrace_python_sdk/utils/__init__.py,sha256=VVDOG-QLd59ZvSHp0avjof0sbxlZ1QQOf0KoOF7ofhQ,3310
 langtrace_python_sdk/utils/langtrace_sampler.py,sha256=BupNndHbU9IL_wGleKetz8FdcveqHMBVz1bfKTTW80w,1753
-langtrace_python_sdk/utils/llm.py,sha256=UOPgt8OcOyjBiBEtxWGYBVLg9EfVCBLyusgcVwLr8Us,23425
+langtrace_python_sdk/utils/llm.py,sha256=yhokm_xqisPLDwopKxqp9OHnmksTKtGanBmXvu3KaHQ,23445
 langtrace_python_sdk/utils/misc.py,sha256=LaQr5LOmZMiuwVdjYh7aIu6o2C_Xb1wgpQGNOVmRzfE,1918
 langtrace_python_sdk/utils/prompt_registry.py,sha256=n5dQMVLBw8aJZY8Utvf67bncc25ELf6AH9BYw8_hSzo,2619
 langtrace_python_sdk/utils/sdk_version_checker.py,sha256=F-VVVH7Fmhr5LcY0IIe-34zIi5RQcx26uuxFpPzZesM,1782
@@ -312,8 +312,8 @@ tests/pinecone/cassettes/test_query.yaml,sha256=b5v9G3ssUy00oG63PlFUR3JErF2Js-5A
 tests/pinecone/cassettes/test_upsert.yaml,sha256=neWmQ1v3d03V8WoLl8FoFeeCYImb8pxlJBWnFd_lITU,38607
 tests/qdrant/conftest.py,sha256=9n0uHxxIjWk9fbYc4bx-uP8lSAgLBVx-cV9UjnsyCHM,381
 tests/qdrant/test_qdrant.py,sha256=pzjAjVY2kmsmGfrI2Gs2xrolfuaNHz7l1fqGQCjp5_o,3353
-langtrace_python_sdk-3.8.10.dist-info/METADATA,sha256=-4u_t1c_jWQZ_yk1oIedRtGD9L5gCxcUbJBwfwTbfh0,15845
-langtrace_python_sdk-3.8.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langtrace_python_sdk-3.8.10.dist-info/entry_points.txt,sha256=1_b9-qvf2fE7uQNZcbUei9vLpFZBbbh9LrtGw95ssAo,70
-langtrace_python_sdk-3.8.10.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-langtrace_python_sdk-3.8.10.dist-info/RECORD,,
+langtrace_python_sdk-3.8.11.dist-info/METADATA,sha256=uthy1_AtXV7QgTGRDn_gOcXyZokYgF5_FU858iZeEbw,15845
+langtrace_python_sdk-3.8.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langtrace_python_sdk-3.8.11.dist-info/entry_points.txt,sha256=1_b9-qvf2fE7uQNZcbUei9vLpFZBbbh9LrtGw95ssAo,70
+langtrace_python_sdk-3.8.11.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+langtrace_python_sdk-3.8.11.dist-info/RECORD,,