Merge branch 'development' of github.com:Scale3-Labs/langtrace-python-sdk into ali/bedrock-refactor

Author: alizenhom

src/langtrace_python_sdk/instrumentation/autogen/patch.py

@@ -92,13 +92,16 @@ def traced_method(wrapped, instance, args, kwargs):
                 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
 

src/langtrace_python_sdk/instrumentation/weaviate/patch.py

@@ -49,7 +49,9 @@
 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):

src/langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "3.3.19"
+__version__ = "3.3.21"