handle cache case in autogen

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

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