langtrace-python-sdk 3.3.19__tar.gz → 3.3.21__tar.gz

{langtrace_python_sdk-3.3.19 → langtrace_python_sdk-3.3.21}/PKG-INFO

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

@@ -92,13 +92,16 @@ def patch_generate_reply(name, version, tracer: Tracer):
                 set_span_attributes(span, attributes)
                 set_event_completion(span, [{"role": "assistant", "content": result}])
                 if llm_config:
-                    total_cost, response_model = list(instance.get_total_usage().keys())
-                    set_span_attribute(
-                        span, SpanAttributes.LLM_RESPONSE_MODEL, response_model
-                    )
-                    set_usage_attributes(
-                        span, instance.get_total_usage().get(response_model)
-                    )
+                    if instance.get_total_usage() is not None:
+                        total_cost, response_model = list(
+                            instance.get_total_usage().keys()
+                        )
+                        set_span_attribute(
+                            span, SpanAttributes.LLM_RESPONSE_MODEL, response_model
+                        )
+                        set_usage_attributes(
+                            span, instance.get_total_usage().get(response_model)
+                        )
 
                 return result
 

{langtrace_python_sdk-3.3.19 → langtrace_python_sdk-3.3.21}/src/langtrace_python_sdk/instrumentation/weaviate/patch.py

@@ -49,7 +49,9 @@ METADATA_ATTRIBUTES = [
 def extract_inputs(args, kwargs):
     extracted_params = {}
     kwargs_without_properties = {
-        k: v for k, v in kwargs.items() if k not in ["properties", "fusion_type", "filters"]
+        k: v
+        for k, v in kwargs.items()
+        if k not in ["properties", "fusion_type", "filters"]
     }
     extracted_params.update(extract_input_params(args, kwargs_without_properties))
     if kwargs.get("filters", None):
@@ -94,7 +96,7 @@ def aggregate_responses(result):
         # For single object responses
         all_responses = get_response_object_attributes(result)
 
-    return json.dumps(all_responses)
+    return json.dumps(all_responses, default=str)
 
 
 def get_response_object_attributes(response_object):

langtrace_python_sdk-3.3.21/src/langtrace_python_sdk/version.py

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

langtrace_python_sdk-3.3.19/src/langtrace_python_sdk/version.py

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