langtrace-python-sdk 3.3.7__py3-none-any.whl → 3.3.9__py3-none-any.whl

langtrace_python_sdk/instrumentation/mistral/patch.py

@@ -46,7 +46,7 @@ from langtrace_python_sdk.instrumentation.openai.patch import extract_content
 
 
 def chat_complete(original_method, version, tracer, is_async=False, is_streaming=False):
-        
+
     def traced_method(wrapped, instance, args, kwargs):
         service_provider = SERVICE_PROVIDERS["MISTRAL"]
         llm_prompts = []
@@ -65,7 +65,6 @@ def chat_complete(original_method, version, tracer, is_async=False, is_streaming
 
         attributes = LLMSpanAttributes(**span_attributes)
 
-
         span = tracer.start_span(
             name=get_span_name(APIS[api]["METHOD"]),
             kind=SpanKind.CLIENT,
@@ -87,7 +86,6 @@ def chat_complete(original_method, version, tracer, is_async=False, is_streaming
                 span.set_status(StatusCode.OK)
                 span.end()
                 return result
-                
 
         except Exception as error:
             span.record_exception(error)
@@ -186,7 +184,6 @@ def _set_response_attributes(span, kwargs, result):
             for choice in result.choices
         ]
         set_event_completion(span, responses)
-
     # Get the usage
     if hasattr(result, "usage") and result.usage is not None:
-        set_usage_attributes(span, result.usage)
+        set_usage_attributes(span, dict(result.usage))

langtrace_python_sdk/utils/with_root_span.py

@@ -14,6 +14,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
+import base64
 import asyncio
 import os
 from deprecated import deprecated
@@ -145,7 +146,9 @@ class SendUserFeedback:
     _langtrace_api_key: str
 
     def __init__(self):
-        self._langtrace_host = os.environ.get("LANGTRACE_API_HOST", LANGTRACE_REMOTE_URL)
+        self._langtrace_host = os.environ.get(
+            "LANGTRACE_API_HOST", LANGTRACE_REMOTE_URL
+        )
         # When the host is set to /api/trace, remove the /api/trace
         if self._langtrace_host.endswith("/api/trace"):
             self._langtrace_host = self._langtrace_host.replace("/api/trace", "")
@@ -162,14 +165,13 @@ class SendUserFeedback:
                 print(Fore.RESET)
                 return
 
-            # convert spanId and traceId to hexadecimals
-            span_hex_number = hex(int(data["spanId"], 10))[2:]  # Convert to hex and remove the '0x' prefix
-            formatted_span_hex_number = span_hex_number.zfill(16)  # Pad with zeros to 16 characters
-            data["spanId"] = f"0x{formatted_span_hex_number}"
+            # convert spanId and traceId to Base64
+            span_bytes = int(data["spanId"], 10).to_bytes(8, "big")
+            data["spanId"] = base64.b64encode(span_bytes).decode("ascii")
 
-            trace_hex_number = hex(int(data["traceId"], 10))[2:]  # Convert to hex and remove the '0x' prefix
-            formatted_trace_hex_number = trace_hex_number.zfill(32)  # Pad with zeros to 32 characters
-            data["traceId"] = f"0x{formatted_trace_hex_number}"
+            # Convert traceId to base64
+            trace_bytes = int(data["traceId"], 10).to_bytes(16, "big")
+            data["traceId"] = base64.b64encode(trace_bytes).decode("ascii")
 
             evaluation = self.get_evaluation(data["spanId"])
             headers = {"x-api-key": self._langtrace_api_key}

langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "3.3.7"
+__version__ = "3.3.9"

{langtrace_python_sdk-3.3.7.dist-info → langtrace_python_sdk-3.3.9.dist-info}/METADATA

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

@@ -105,7 +105,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=TtRWuUiWUB0S7JiQpUsF9lZsiyqPG3m9mMDX-QlDgAw,12601
-langtrace_python_sdk/version.py,sha256=3F4-qyOqFzf6p1RSj1DFFxsNoAZY7v8Lh4g109vr39E,22
+langtrace_python_sdk/version.py,sha256=yEHnZnJtVwSuoRBWEbpTXI-KzATzAJQjuf0BcvE1EkQ,22
 langtrace_python_sdk/constants/__init__.py,sha256=3CNYkWMdd1DrkGqzLUgNZXjdAlM6UFMlf_F-odAToyc,146
 langtrace_python_sdk/constants/exporter/langtrace_exporter.py,sha256=d-3Qn5C_NTy1NkmdavZvy-6vePwTC5curN6QMy2haHc,50
 langtrace_python_sdk/constants/instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -189,7 +189,7 @@ langtrace_python_sdk/instrumentation/milvus/instrumentation.py,sha256=bpf6uRIJ44
 langtrace_python_sdk/instrumentation/milvus/patch.py,sha256=0yY5aQz0x7hpQZ8U-0qfyk_KIizW60s0e2CZrXe3FNk,4729
 langtrace_python_sdk/instrumentation/mistral/__init__.py,sha256=mkGALBQvq0jSfwDl6TU09SFwnVs6O4zkUi-yVmd3SNg,90
 langtrace_python_sdk/instrumentation/mistral/instrumentation.py,sha256=qtCkHCSOaiicUChbmTID4lcK1rbeW8oRSbpda2ogbgM,2328
-langtrace_python_sdk/instrumentation/mistral/patch.py,sha256=1peU0vqt9BGYn2PFNyKAMKNRVMEujXNyZGzgttPJrTQ,6580
+langtrace_python_sdk/instrumentation/mistral/patch.py,sha256=5EzqMeIEcMUH8P-l9-ijpRNWSc6mYwe9Gz0VY4PLrS0,6559
 langtrace_python_sdk/instrumentation/ollama/__init__.py,sha256=g2zJsXnDHinXPzTc-WxDeTtHmr9gmAj3K6l_00kP8c8,82
 langtrace_python_sdk/instrumentation/ollama/instrumentation.py,sha256=jdsvkqUJAAUNLVPtAkn_rG26HXetVQXWtjn4a6eWZro,2029
 langtrace_python_sdk/instrumentation/ollama/patch.py,sha256=7ETx0tQic5h_kH1f-IeptFwgNTBb4hSkTkWsB18Avm0,5375
@@ -221,7 +221,7 @@ langtrace_python_sdk/utils/prompt_registry.py,sha256=n5dQMVLBw8aJZY8Utvf67bncc25
 langtrace_python_sdk/utils/sdk_version_checker.py,sha256=F-VVVH7Fmhr5LcY0IIe-34zIi5RQcx26uuxFpPzZesM,1782
 langtrace_python_sdk/utils/silently_fail.py,sha256=wzmvRDZppaRZgVP8C1xpq2GlWXYCwubhaeWvEbQP1SI,1196
 langtrace_python_sdk/utils/types.py,sha256=l-N6o7cnWUyrD6dBvW7W3Pf5CkPo5QaoT__k1XLbrQg,383
-langtrace_python_sdk/utils/with_root_span.py,sha256=2iWu8XD1NOFqSFgDZDJiMHZ1JB4HzmYPLr_F3Ugul2k,8480
+langtrace_python_sdk/utils/with_root_span.py,sha256=EW0WQfxESntf23dJt0E0rFdHV-7BRniW1Ce6nrCoEPY,8291
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/conftest.py,sha256=0Jo6iCZTXbdvyJVhG9UpYGkLabL75378oauCzmt-Sa8,603
 tests/utils.py,sha256=8ZBYvxBH6PynipT1sqenfyjTGLhEV7SORQH1NJjnpsM,2500
@@ -264,8 +264,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.7.dist-info/METADATA,sha256=mPY5ZCzq5zv5tozpM1La21OLTU6I2phpk9zevTexhns,15642
-langtrace_python_sdk-3.3.7.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-langtrace_python_sdk-3.3.7.dist-info/entry_points.txt,sha256=1_b9-qvf2fE7uQNZcbUei9vLpFZBbbh9LrtGw95ssAo,70
-langtrace_python_sdk-3.3.7.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-langtrace_python_sdk-3.3.7.dist-info/RECORD,,
+langtrace_python_sdk-3.3.9.dist-info/METADATA,sha256=ZFliHANcgt7Nt3sraMuYfw7_C4vheqrKZEz7P-yL30o,15642
+langtrace_python_sdk-3.3.9.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+langtrace_python_sdk-3.3.9.dist-info/entry_points.txt,sha256=1_b9-qvf2fE7uQNZcbUei9vLpFZBbbh9LrtGw95ssAo,70
+langtrace_python_sdk-3.3.9.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+langtrace_python_sdk-3.3.9.dist-info/RECORD,,