langtrace-python-sdk 3.3.18__py3-none-any.whl → 3.3.20__py3-none-any.whl

langtrace_python_sdk/instrumentation/autogen/patch.py

@@ -56,12 +56,8 @@ def patch_initiate_chat(name, version, tracer: Tracer):
 def patch_generate_reply(name, version, tracer: Tracer):
 
     def traced_method(wrapped, instance, args, kwargs):
-
         llm_config = instance.llm_config
-        kwargs = {
-            **kwargs,
-            **llm_config.get("config_list")[0],
-        }
+        kwargs = parse_kwargs(kwargs, llm_config)
         service_provider = SERVICE_PROVIDERS["AUTOGEN"]
 
         span_attributes = {
@@ -84,20 +80,28 @@ def patch_generate_reply(name, version, tracer: Tracer):
             try:
 
                 result = wrapped(*args, **kwargs)
-                
+
                 # if caching is disabled, return result as langtrace will instrument the rest.
-                if "cache_seed" in llm_config and llm_config.get("cache_seed") is None:
+                if (
+                    llm_config
+                    and "cache_seed" in llm_config
+                    and llm_config.get("cache_seed") is None
+                ):
                     return result
 
                 set_span_attributes(span, attributes)
                 set_event_completion(span, [{"role": "assistant", "content": result}])
-                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 llm_config:
+                    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
 
@@ -130,3 +134,20 @@ def parse_agent(agent):
         "llm_config": str(getattr(agent, "llm_config", None)),
         "human_input_mode": getattr(agent, "human_input_mode", None),
     }
+
+
+def parse_kwargs(kwargs, llm_config):
+    # Handle cases where llm_config is False or None
+    if not llm_config:
+        return kwargs
+
+    if isinstance(llm_config, dict) and "config_list" in llm_config:
+        return {
+            **kwargs,
+            **llm_config.get("config_list")[0],
+        }
+    else:
+        return {
+            **kwargs,
+            **llm_config,
+        }

langtrace_python_sdk/version.py

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

{langtrace_python_sdk-3.3.18.dist-info → langtrace_python_sdk-3.3.20.dist-info}/METADATA

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

@@ -109,7 +109,7 @@ examples/weaviate_example/__init__.py,sha256=8JMDBsRSEV10HfTd-YC7xb4txBjD3la56sn
 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=nvPaJc426Iww3ildrhsSacXtLdzsZIa94_rlK2giyVM,13153
-langtrace_python_sdk/version.py,sha256=9g2UTdPAtK4y3-RMCghlZIFhe-b0mxLKaThdD7MED8M,23
+langtrace_python_sdk/version.py,sha256=81bzJ_fJLZ-62a4Hk7ky5bhaetRoFWuYWwZn42n0fPc,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
@@ -141,7 +141,7 @@ langtrace_python_sdk/instrumentation/anthropic/patch.py,sha256=ztPN4VZujoxYOKhTb
 langtrace_python_sdk/instrumentation/anthropic/types.py,sha256=WdeXe2tkjAisMLK38STKwVe7MJS5SLoETJ_aKhA_-UU,2463
 langtrace_python_sdk/instrumentation/autogen/__init__.py,sha256=unDhpqWQIdHFw24lRsRu1Mm1NCZxZgyBrPRZrAJL3Lo,90
 langtrace_python_sdk/instrumentation/autogen/instrumentation.py,sha256=MVDUCBi6XzLQYmZd6myAounI0HeM8QWX5leuul5Hj0Q,1262
-langtrace_python_sdk/instrumentation/autogen/patch.py,sha256=mp6WxHYVqTXvqZOi6CnZNN0MmzoG5v9LPMU2fjkivsY,4650
+langtrace_python_sdk/instrumentation/autogen/patch.py,sha256=7Sq3C8Q5tT27UkWXd1SZgnLC1pbQf_tpTrxoBIYsDw4,5273
 langtrace_python_sdk/instrumentation/aws_bedrock/__init__.py,sha256=IHqPgR1kdDvcoV1nUb-B21PaJ_qbQB0jc011Udi1ioU,96
 langtrace_python_sdk/instrumentation/aws_bedrock/instrumentation.py,sha256=2l-WiyWYUEoGre92rmylq2jPZ5w4jcxTXmCTuQNC1RU,1911
 langtrace_python_sdk/instrumentation/aws_bedrock/patch.py,sha256=VAroMezSGKT2jQ5tggbdiMRIPr9mtLItGJJgZ-xoGls,6296
@@ -270,8 +270,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.3.18.dist-info/METADATA,sha256=M4r-V_wmy75oIlqyYNY0beRto96gM-DjA7OXdvuCmyg,15676
-langtrace_python_sdk-3.3.18.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langtrace_python_sdk-3.3.18.dist-info/entry_points.txt,sha256=1_b9-qvf2fE7uQNZcbUei9vLpFZBbbh9LrtGw95ssAo,70
-langtrace_python_sdk-3.3.18.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-langtrace_python_sdk-3.3.18.dist-info/RECORD,,
+langtrace_python_sdk-3.3.20.dist-info/METADATA,sha256=cbJScXuK1lpnJnx-ywg1uO_xCgmByqSVGSLwH3w7cNo,15676
+langtrace_python_sdk-3.3.20.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langtrace_python_sdk-3.3.20.dist-info/entry_points.txt,sha256=1_b9-qvf2fE7uQNZcbUei9vLpFZBbbh9LrtGw95ssAo,70
+langtrace_python_sdk-3.3.20.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+langtrace_python_sdk-3.3.20.dist-info/RECORD,,