langtrace-python-sdk 3.8.17__py3-none-any.whl → 3.8.18__py3-none-any.whl

langtrace_python_sdk/instrumentation/google_genai/patch.py

@@ -1,3 +1,4 @@
+import base64
 from langtrace_python_sdk.utils.llm import (
     get_langtrace_attributes,
     get_llm_request_attributes,
@@ -14,20 +15,65 @@ from langtrace.trace_attributes import SpanAttributes
 
 from typing import Iterator
 
+def capture_input_data(contents):
+    input_data = []
+    
+    if isinstance(contents, list):
+        content_parts = []
+        for content in contents:
+            if hasattr(content, 'parts'):
+                for part in content.parts:
+                    if hasattr(part, 'text') and part.text:
+                        content_parts.append({
+                            "type": "text",
+                            "text": part.text
+                        })
+                    if hasattr(part, 'inline_data') and part.inline_data:
+                        content_parts.append({
+                            "type": "image_url",
+                            "image_url": f"data:{part.inline_data.mime_type};base64,{base64.b64encode(part.inline_data.data).decode('utf-8')}"
+                        })
+            else:
+                if isinstance(content, str):
+                    content_parts.append({
+                        "type": "text",
+                        "text": content
+                    })
+                elif hasattr(content, 'text') and content.text:
+                    content_parts.append({
+                        "type": "text",
+                        "text": content.text
+                    })
+                elif hasattr(content, 'inline_data') and content.inline_data:
+                    content_parts.append({
+                        "type": "image_url",
+                        "image_url": f"data:{content.inline_data.mime_type};base64,{base64.b64encode(content.inline_data.data).decode('utf-8')}"
+                    })
+                else:
+                    content_parts.append({
+                        "type": "text",
+                        "text": content
+                    })
+        input_data.append({
+            "role": "user",
+            "content": content_parts
+        })
+    else:
+        input_data.append({
+            "role": "user",
+            "content": contents
+        })
+    
+    return input_data
+
 
 def patch_google_genai(tracer: Tracer, version: str):
     def traced_method(wrapped, instance, args, kwargs):
-        prompt = [
-            {
-                "role": "user",
-                "content": kwargs["contents"],
-            }
-        ]
         span_attributes = {
             **get_langtrace_attributes(
                 service_provider="google_genai", version=version
             ),
-            **get_llm_request_attributes(kwargs=kwargs, prompts=prompt),
+            **get_llm_request_attributes(kwargs=kwargs, prompts=capture_input_data(kwargs["contents"])),
         }
         with tracer.start_as_current_span(
             name="google.genai.generate_content",

langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "3.8.17"
+__version__ = "3.8.18"

{langtrace_python_sdk-3.8.17.dist-info → langtrace_python_sdk-3.8.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langtrace-python-sdk
-Version: 3.8.17
+Version: 3.8.18
 Summary: Python SDK for LangTrace
 Project-URL: Homepage, https://github.com/Scale3-Labs/langtrace-python-sdk
 Author-email: Scale3 Labs <engineering@scale3labs.com>
@@ -10,7 +10,7 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.9
-Requires-Dist: boto3
+Requires-Dist: boto3==1.38.0
 Requires-Dist: colorama>=0.4.6
 Requires-Dist: fsspec>=2024.6.0
 Requires-Dist: opentelemetry-api>=1.25.0

{langtrace_python_sdk-3.8.17.dist-info → langtrace_python_sdk-3.8.18.dist-info}/RECORD

@@ -121,7 +121,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=JYBCbklWv463lzRUH7pfkHAVoc3YHh-q_2Iv6LZ2HhU,14896
-langtrace_python_sdk/version.py,sha256=csVUBgURmjY_-YGxUIC12qiRsv-dOi-b_wDRDkd1SUg,23
+langtrace_python_sdk/version.py,sha256=whHLu8P7QnGYdLeAzZ-eOHwlW3MJuanQ-a9U15xrJzY,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
@@ -192,7 +192,7 @@ langtrace_python_sdk/instrumentation/gemini/instrumentation.py,sha256=eGWr2dy1f_
 langtrace_python_sdk/instrumentation/gemini/patch.py,sha256=PG5E5v253x2ufQ81-aUUFzDRYq-DDhiqwmNoplHPjsM,6261
 langtrace_python_sdk/instrumentation/google_genai/__init__.py,sha256=nQ2kSsUDpLGZ_kz2QGddDwnawL9juVA7pW4G-BOe5VI,98
 langtrace_python_sdk/instrumentation/google_genai/instrumentation.py,sha256=rSB3nOXXe4gMVPWMGPnIHvWjqyPeRvifxOI62ulJ7uY,1049
-langtrace_python_sdk/instrumentation/google_genai/patch.py,sha256=X0TWY1D4XHweaNu70PlXBDzKEaHIibLpkJiIsp4jF6A,4115
+langtrace_python_sdk/instrumentation/google_genai/patch.py,sha256=WS7VLe5lbbTU0BAiP69vJ5V5plyQ4GeXs1LQDBNqINQ,5995
 langtrace_python_sdk/instrumentation/graphlit/__init__.py,sha256=mkUPwozR-I-D5ZrumJs1gS7sSouY0UN68Ne9LiNEBTs,92
 langtrace_python_sdk/instrumentation/graphlit/instrumentation.py,sha256=V9GRZ6cj2lt20xCVfL55lGU4p0HlZxAYUqwpWogXDtY,2074
 langtrace_python_sdk/instrumentation/graphlit/patch.py,sha256=jVME4xlFTwJRPWOBGUtQiZrGkr1ScR2DSYwcWPd5smk,2921
@@ -312,8 +312,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.8.17.dist-info/METADATA,sha256=feYBeHSHTLTW9hBTvwQxD52bd1Q2eeSpd-8un4fg5Y0,15845
-langtrace_python_sdk-3.8.17.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langtrace_python_sdk-3.8.17.dist-info/entry_points.txt,sha256=1_b9-qvf2fE7uQNZcbUei9vLpFZBbbh9LrtGw95ssAo,70
-langtrace_python_sdk-3.8.17.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-langtrace_python_sdk-3.8.17.dist-info/RECORD,,
+langtrace_python_sdk-3.8.18.dist-info/METADATA,sha256=XHSAwzWsHAh5FLLWZ9896hei2d_hqQeEkomYDIEV0KM,15853
+langtrace_python_sdk-3.8.18.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langtrace_python_sdk-3.8.18.dist-info/entry_points.txt,sha256=1_b9-qvf2fE7uQNZcbUei9vLpFZBbbh9LrtGw95ssAo,70
+langtrace_python_sdk-3.8.18.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+langtrace_python_sdk-3.8.18.dist-info/RECORD,,