langtrace-python-sdk 2.1.26__tar.gz → 2.1.27__tar.gz

{langtrace_python_sdk-2.1.26 → langtrace_python_sdk-2.1.27}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 2.1.26
+Version: 2.1.27
 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-2.1.26 → langtrace_python_sdk-2.1.27}/src/langtrace_python_sdk/instrumentation/openai/patch.py

@@ -28,7 +28,11 @@ from langtrace_python_sdk.constants.instrumentation.common import (
     SERVICE_PROVIDERS,
 )
 from langtrace_python_sdk.constants.instrumentation.openai import APIS
-from langtrace_python_sdk.utils.llm import calculate_prompt_tokens, estimate_tokens
+from langtrace_python_sdk.utils.llm import (
+    calculate_prompt_tokens,
+    estimate_tokens,
+    get_tool_calls,
+)
 from openai._types import NOT_GIVEN
 
 
@@ -430,9 +434,10 @@ def chat_completions_create(original_method, version, tracer):
         # handle tool calls in the kwargs
         llm_prompts = []
         for item in kwargs.get("messages", []):
-            if hasattr(item, "tool_calls") and item.tool_calls is not None:
+            tools = get_tool_calls(item)
+            if tools is not None:
                 tool_calls = []
-                for tool_call in item.tool_calls:
+                for tool_call in tools:
                     tool_call_dict = {
                         "id": tool_call.id if hasattr(tool_call, "id") else "",
                         "type": tool_call.type if hasattr(tool_call, "type") else "",
@@ -611,9 +616,10 @@ def async_chat_completions_create(original_method, version, tracer):
         # handle tool calls in the kwargs
         llm_prompts = []
         for item in kwargs.get("messages", []):
-            if hasattr(item, "tool_calls") and item.tool_calls is not None:
+            tools = get_tool_calls(item)
+            if tools is not None:
                 tool_calls = []
-                for tool_call in item.tool_calls:
+                for tool_call in tools:
                     tool_call_dict = {
                         "id": tool_call.id if hasattr(tool_call, "id") else "",
                         "type": tool_call.type if hasattr(tool_call, "type") else "",

{langtrace_python_sdk-2.1.26 → langtrace_python_sdk-2.1.27}/src/langtrace_python_sdk/utils/llm.py

@@ -65,3 +65,15 @@ def set_span_attributes(span, name, value):
         if value != "":
             span.set_attribute(name, value)
     return
+
+
+def get_tool_calls(item):
+    if isinstance(item, dict):
+        if "tool_calls" in item and item["tool_calls"] is not None:
+            return item["tool_calls"]
+        return None
+
+    else:
+        if hasattr(item, "tool_calls") and item.tool_calls is not None:
+            return item.tool_calls
+        return None

langtrace_python_sdk-2.1.27/src/langtrace_python_sdk/version.py

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

langtrace_python_sdk-2.1.26/src/langtrace_python_sdk/version.py

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