langtrace-python-sdk 3.3.12__tar.gz → 3.3.14__tar.gz

{langtrace_python_sdk-3.3.12 → langtrace_python_sdk-3.3.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 3.3.12
+Version: 3.3.14
 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.3.12 → langtrace_python_sdk-3.3.14}/src/langtrace_python_sdk/instrumentation/dspy/instrumentation.py

@@ -27,12 +27,12 @@ class DspyInstrumentation(BaseInstrumentor):
     The DspyInstrumentor class represents the DSPy instrumentation"""
 
     def instrumentation_dependencies(self) -> Collection[str]:
-        return ["dspy-ai >= 2.0.0"]
+        return ["dspy >= 2.0.0"]
 
     def _instrument(self, **kwargs):
         tracer_provider = kwargs.get("tracer_provider")
         tracer = get_tracer(__name__, "", tracer_provider)
-        version = v("dspy-ai")
+        version = v("dspy")
         _W(
             "dspy.teleprompt.bootstrap",
             "BootstrapFewShot.compile",

{langtrace_python_sdk-3.3.12 → langtrace_python_sdk-3.3.14}/src/langtrace_python_sdk/instrumentation/gemini/patch.py

@@ -137,7 +137,7 @@ def set_response_attributes(
     if hasattr(result, "text"):
         set_event_completion(span, [{"role": "assistant", "content": result.text}])
 
-    if hasattr(result, "usage_metadata"):
+    if hasattr(result, "usage_metadata") and result.usage_metadata is not None:
         usage = result.usage_metadata
         input_tokens = usage.prompt_token_count
         output_tokens = usage.candidates_token_count
@@ -152,7 +152,7 @@ def build_streaming_response(span, response):
         item_to_yield = item
         complete_response += str(item.text)
         yield item_to_yield
-        if hasattr(item, "usage_metadata"):
+        if hasattr(item, "usage_metadata") and item.usage_metadata is not None:
             usage = item.usage_metadata
             input_tokens = usage.prompt_token_count
             output_tokens = usage.candidates_token_count
@@ -171,7 +171,7 @@ async def abuild_streaming_response(span, response):
         item_to_yield = item
         complete_response += str(item.text)
         yield item_to_yield
-        if hasattr(item, "usage_metadata"):
+        if hasattr(item, "usage_metadata") and item.usage_metadata is not None:
             usage = item.usage_metadata
             input_tokens = usage.prompt_token_count
             output_tokens = usage.candidates_token_count

{langtrace_python_sdk-3.3.12 → langtrace_python_sdk-3.3.14}/src/langtrace_python_sdk/instrumentation/langchain_core/patch.py

@@ -98,7 +98,7 @@ def generic_patch(
                 result = wrapped(*args, **kwargs)
                 if trace_output:
                     span.set_attribute("langchain.outputs", to_json_string(result))
-                    if hasattr(result, "usage_metadata"):
+                    if hasattr(result, "usage_metadata") and result.usage_metadata is not None:
                         span.set_attribute(
                             SpanAttributes.LLM_USAGE_PROMPT_TOKENS,
                             result.usage_metadata["input_tokens"],

{langtrace_python_sdk-3.3.12 → langtrace_python_sdk-3.3.14}/src/langtrace_python_sdk/instrumentation/vertexai/patch.py

@@ -77,7 +77,7 @@ def set_response_attributes(span: Span, result):
     if hasattr(result, "text"):
         set_event_completion(span, [{"role": "assistant", "content": result.text}])
 
-    if hasattr(result, "usage_metadata"):
+    if hasattr(result, "usage_metadata") and result.usage_metadata is not None:
         usage = result.usage_metadata
         input_tokens = usage.prompt_token_count
         output_tokens = usage.candidates_token_count

{langtrace_python_sdk-3.3.12 → langtrace_python_sdk-3.3.14}/src/langtrace_python_sdk/langtrace.py

@@ -275,7 +275,7 @@ def init(
         "weaviate-client": WeaviateInstrumentation(),
         "sqlalchemy": SQLAlchemyInstrumentor(),
         "ollama": OllamaInstrumentor(),
-        "dspy-ai": DspyInstrumentation(),
+        "dspy": DspyInstrumentation(),
         "crewai": CrewAIInstrumentation(),
         "vertexai": VertexAIInstrumentation(),
         "google-cloud-aiplatform": VertexAIInstrumentation(),

{langtrace_python_sdk-3.3.12 → langtrace_python_sdk-3.3.14}/src/langtrace_python_sdk/utils/llm.py

@@ -421,7 +421,7 @@ class StreamWrapper:
                 self.completion_tokens = chunk.usage.completion_tokens
 
         # VertexAI
-        if hasattr(chunk, "usage_metadata"):
+        if hasattr(chunk, "usage_metadata") and chunk.usage_metadata is not None:
             self.completion_tokens = chunk.usage_metadata.candidates_token_count
             self.prompt_tokens = chunk.usage_metadata.prompt_token_count
 

langtrace_python_sdk-3.3.14/src/langtrace_python_sdk/version.py

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

langtrace_python_sdk-3.3.12/src/langtrace_python_sdk/version.py

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