langtrace-python-sdk 1.2.19__py3-none-any.whl → 1.2.21__py3-none-any.whl

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,
@@ -270,30 +281,19 @@ def chat_completions_create(original_method, version, tracer):
                     }
                 ),
             )
-            if function_call is False:
-                span.set_attribute(
-                    "llm.responses",
-                    json.dumps(
+            span.set_attribute(
+                "llm.responses",
+                json.dumps(
+                    [
                         {
                             "message": {
                                 "role": "assistant",
                                 "content": "".join(result_content),
                             }
                         }
-                    ),
-                )
-            else:
-                span.set_attribute(
-                    "llm.responses",
-                    json.dumps(
-                        {
-                            "message": {
-                                "role": "assistant",
-                                "function_call": "".join(result_content),
-                            }
-                        }
-                    ),
-                )
+                    ]
+                ),
+            )
             span.set_status(StatusCode.OK)
             span.end()
 

langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "1.2.19"
+__version__ = "1.2.21"

{langtrace_python_sdk-1.2.19.dist-info → langtrace_python_sdk-1.2.21.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 1.2.19
+Version: 1.2.21
 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.19.dist-info → langtrace_python_sdk-1.2.21.dist-info}/RECORD

@@ -18,7 +18,7 @@ examples/pinecone_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 examples/pinecone_example/basic.py,sha256=hdV6-5Fmol9zeyFzDtdabD62vkqUJ4lCHG2YcVNpIpI,933
 langtrace_python_sdk/__init__.py,sha256=SlHg447-nQBbw8exRNJP_OyHUZ39Sldb7aaQ35hIRm8,262
 langtrace_python_sdk/langtrace.py,sha256=sWC7WqbOdOUB1sZ_hzLZVp0_Q8fBTRJRMUvOPyO6f20,3170
-langtrace_python_sdk/version.py,sha256=vBqPpUxy9sFZkDmAAvQCrIqAA_ye7foz7jV5SkuJi0I,23
+langtrace_python_sdk/version.py,sha256=3uF8dV6igE_qWszFViblw_rtJOgCwoi-rVuWcQrSR0A,23
 langtrace_python_sdk/constants/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/constants/exporter/langtrace_exporter.py,sha256=OfgvphRBPENwsAizqzka9lFu9aDw-A7ErYJ0atl3bpI,56
 langtrace_python_sdk/constants/instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -50,14 +50,14 @@ langtrace_python_sdk/instrumentation/llamaindex/instrumentation.py,sha256=wVSvxS
 langtrace_python_sdk/instrumentation/llamaindex/patch.py,sha256=NqBsPM4eE3yMu1aIev8p8yH5Thsv72JL68ABEZNe2ug,2043
 langtrace_python_sdk/instrumentation/openai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/openai/instrumentation.py,sha256=stlFopPggB8mgnn4QdVh41rmA8mlzH5x-C00XHfApCE,1408
-langtrace_python_sdk/instrumentation/openai/patch.py,sha256=H0c12pl7UZDyn2Go0a2ftcXcJNaGWlW4FzT4qHBMtUY,15228
+langtrace_python_sdk/instrumentation/openai/patch.py,sha256=iP4RFL5ZHqu3BlNNq5DmmstLMaqaXmk4VKRbGXts1EI,15341
 langtrace_python_sdk/instrumentation/pinecone/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/pinecone/instrumentation.py,sha256=o0EUd5jvHaDKOUTj4NjnL5UfDHDHxyXkWGlTW4oeRDk,1784
 langtrace_python_sdk/instrumentation/pinecone/patch.py,sha256=5lF7hQmg2-U2EWtOC0w8_peRaNMysBomb0fjiNoS6eQ,2200
 langtrace_python_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/utils/llm.py,sha256=4z2e-md_ELXCEuOIRVWracR6qH2pmsOxCqpkuF9_3Nw,1589
 langtrace_python_sdk/utils/with_root_span.py,sha256=N7ONrcF0myZbHBy5gpQffDbX-Kf63Crsz9szG0i3m08,1889
-langtrace_python_sdk-1.2.19.dist-info/METADATA,sha256=rm-JsAkSY5nhdnd1Wuuv6liUzDVnKdSm_yzeX4SS5Ao,7695
-langtrace_python_sdk-1.2.19.dist-info/WHEEL,sha256=as-1oFTWSeWBgyzh0O_qF439xqBe6AbBgt4MfYe5zwY,87
-langtrace_python_sdk-1.2.19.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-langtrace_python_sdk-1.2.19.dist-info/RECORD,,
+langtrace_python_sdk-1.2.21.dist-info/METADATA,sha256=9yDWY-fJ-Sgnc6-M4apfEza7U9X-j0TWn6OQhfWmOZA,7695
+langtrace_python_sdk-1.2.21.dist-info/WHEEL,sha256=as-1oFTWSeWBgyzh0O_qF439xqBe6AbBgt4MfYe5zwY,87
+langtrace_python_sdk-1.2.21.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+langtrace_python_sdk-1.2.21.dist-info/RECORD,,