langtrace-python-sdk 1.2.18__tar.gz → 1.2.20__tar.gz

{langtrace_python_sdk-1.2.18 → langtrace_python_sdk-1.2.20}/PKG-INFO

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

@@ -194,9 +194,20 @@ def chat_completions_create(original_method, version, tracer):
                 span.end()
                 return result
             else:
-                prompt_tokens = calculate_prompt_tokens(
-                    json.dumps(kwargs.get("messages", {})[0]), kwargs.get("model")
-                )
+                # iterate over kwargs.get("messages", {}) and calculate the prompt tokens
+                prompt_tokens = 0
+                for message in kwargs.get("messages", {}):
+                    prompt_tokens += calculate_prompt_tokens(
+                        json.dumps(message), kwargs.get("model")
+                    )
+
+                # iterate over kwargs.get("functions") and calculate the prompt tokens
+                if kwargs.get("functions") is not None:
+                    for function in kwargs.get("functions"):
+                        prompt_tokens += calculate_prompt_tokens(
+                            json.dumps(function), kwargs.get("model")
+                        )
+
                 return handle_streaming_response(
                     result,
                     span,
@@ -219,6 +230,8 @@ def chat_completions_create(original_method, version, tracer):
         completion_tokens = 0
         try:
             for chunk in result:
+                if hasattr(chunk, "model") and chunk.model is not None:
+                    span.set_attribute("llm.model", chunk.model)
                 if hasattr(chunk, "choices") and chunk.choices is not None:
                     token_counts = [
                         (
@@ -287,7 +300,7 @@ def chat_completions_create(original_method, version, tracer):
                         {
                             "message": {
                                 "role": "assistant",
-                                "function_call": "".join(result_content),
+                                "content": "".join(result_content),
                             }
                         }
                     ),

langtrace_python_sdk-1.2.20/src/langtrace_python_sdk/version.py

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

langtrace_python_sdk-1.2.18/src/langtrace_python_sdk/version.py

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