langtrace-python-sdk 3.8.3__py3-none-any.whl → 3.8.5__py3-none-any.whl

langtrace_python_sdk/utils/llm.py

@@ -94,6 +94,139 @@ def calculate_price_from_usage(model, usage):
     return 0
 
 
+def convert_mistral_messages_to_serializable(mistral_messages):
+    serializable_messages = []
+    
+    try:
+        for message in mistral_messages:
+            serializable_message = {"role": message.role}
+        
+            # Handle content
+            if hasattr(message, "content"):
+                serializable_message["content"] = message.content
+            
+            # Handle tool_calls
+            if hasattr(message, "tool_calls") and message.tool_calls is not None:
+                serializable_tool_calls = []
+                
+                for tool_call in message.tool_calls:
+                    serializable_tool_call = {}
+                    
+                    # Handle id, type, and index
+                    if hasattr(tool_call, "id"):
+                        serializable_tool_call["id"] = tool_call.id
+                    if hasattr(tool_call, "type"):
+                        serializable_tool_call["type"] = tool_call.type
+                    if hasattr(tool_call, "index"):
+                        serializable_tool_call["index"] = tool_call.index
+                    
+                    # Handle function
+                    if hasattr(tool_call, "function"):
+                        function_call = tool_call.function
+                        serializable_function = {}
+                        
+                        if hasattr(function_call, "name"):
+                            serializable_function["name"] = function_call.name
+                        if hasattr(function_call, "arguments"):
+                            serializable_function["arguments"] = function_call.arguments
+                        
+                        serializable_tool_call["function"] = serializable_function
+                    
+                    serializable_tool_calls.append(serializable_tool_call)
+                
+                serializable_message["tool_calls"] = serializable_tool_calls
+            
+            # Handle tool_call_id for tool messages
+            if hasattr(message, "tool_call_id"):
+                serializable_message["tool_call_id"] = message.tool_call_id
+            
+            serializable_messages.append(serializable_message)
+    except Exception as e:
+        pass
+    
+    return serializable_messages
+
+
+def convert_gemini_messages_to_serializable(formatted_messages, system_message=None):
+    """
+    Converts Gemini-formatted messages back to a JSON serializable format.
+    
+    Args:
+        formatted_messages: The formatted messages from Gemini.
+        system_message (str, optional): System message content.
+        
+    Returns:
+        List[dict]: JSON serializable list of message dictionaries.
+    """
+    serializable_messages = []
+    
+    try:
+        # Add system message if present
+        if system_message:
+            serializable_messages.append({
+                "role": "system",
+                "content": system_message
+            })
+        
+        for message_item in formatted_messages:
+            # Handle the case where the item is a dict with 'role' and 'content' keys
+            if isinstance(message_item, dict) and 'role' in message_item and 'content' in message_item:
+                role = message_item['role']
+                content_value = message_item['content']
+                
+                # Initialize our serializable message
+                serializable_message = {"role": role}
+                
+                # If content is a list of Content objects
+                if isinstance(content_value, list) and len(content_value) > 0:
+                    for content_obj in content_value:
+                        # Process each Content object
+                        if hasattr(content_obj, 'parts') and hasattr(content_obj, 'role'):
+                            parts = content_obj.parts
+                            
+                            # Extract text from parts
+                            text_parts = []
+                            for part in parts:
+                                if hasattr(part, 'text') and part.text:
+                                    text_parts.append(part.text)
+                            
+                            if text_parts:
+                                serializable_message["content"] = " ".join(text_parts)
+                            
+                            # Here you can add additional processing for other part types
+                            # like function_call, function_response, inline_data, etc.
+                            # Similar to the previous implementation
+                
+                # If content is a string or already a primitive type
+                elif isinstance(content_value, (str, int, float, bool)) or content_value is None:
+                    serializable_message["content"] = content_value
+                
+                # Add the processed message to our list
+                serializable_messages.append(serializable_message)
+            
+            # Handle the case where the item is a Content object directly
+            elif hasattr(message_item, 'role') and hasattr(message_item, 'parts'):
+                # This is the case from the previous implementation
+                # Process a Content object directly
+                serializable_message = {"role": message_item.role}
+                
+                parts = message_item.parts
+                text_parts = []
+                
+                for part in parts:
+                    if hasattr(part, 'text') and part.text:
+                        text_parts.append(part.text)
+                
+                if text_parts:
+                    serializable_message["content"] = " ".join(text_parts)
+                
+                serializable_messages.append(serializable_message)
+    except Exception as e:
+        pass
+    
+    return serializable_messages
+
+
 def get_langtrace_attributes(version, service_provider, vendor_type="llm"):
     return {
         SpanAttributes.LANGTRACE_SDK_NAME: LANGTRACE_SDK_NAME,
@@ -120,6 +253,23 @@ def get_llm_request_attributes(kwargs, prompts=None, model=None, operation_name=
         or kwargs.get("top_k", None)
         or kwargs.get("top_n", None)
     )
+    
+    try:
+        prompts = json.dumps(prompts) if prompts else None
+    except Exception as e:
+        if "is not JSON serializable" in str(e):
+            # check model
+            if kwargs.get("model") is not None:
+                if kwargs.get("model").startswith("gemini"):
+                    prompts = json.dumps(convert_gemini_messages_to_serializable(prompts))
+                elif kwargs.get("model").startswith("mistral"):
+                    prompts = json.dumps(convert_mistral_messages_to_serializable(prompts))
+                else:
+                    prompts = "[]"
+            else:
+                prompts = "[]"
+        else:
+            prompts = "[]"
 
     top_p = kwargs.get("p", None) or kwargs.get("top_p", None)
     tools = kwargs.get("tools", None)
@@ -132,7 +282,7 @@ def get_llm_request_attributes(kwargs, prompts=None, model=None, operation_name=
         SpanAttributes.LLM_IS_STREAMING: kwargs.get("stream"),
         SpanAttributes.LLM_REQUEST_TEMPERATURE: kwargs.get("temperature"),
         SpanAttributes.LLM_TOP_K: top_k,
-        SpanAttributes.LLM_PROMPTS: json.dumps(prompts) if prompts else None,
+        SpanAttributes.LLM_PROMPTS: prompts if prompts else None,
         SpanAttributes.LLM_USER: user,
         SpanAttributes.LLM_REQUEST_TOP_P: top_p,
         SpanAttributes.LLM_REQUEST_MAX_TOKENS: kwargs.get("max_tokens"),

langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "3.8.3"
+__version__ = "3.8.5"

{langtrace_python_sdk-3.8.3.dist-info → langtrace_python_sdk-3.8.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langtrace-python-sdk
-Version: 3.8.3
+Version: 3.8.5
 Summary: Python SDK for LangTrace
 Project-URL: Homepage, https://github.com/Scale3-Labs/langtrace-python-sdk
 Author-email: Scale3 Labs <engineering@scale3labs.com>
@@ -41,7 +41,8 @@ Requires-Dist: langchain-openai; extra == 'dev'
 Requires-Dist: litellm==1.48.7; extra == 'dev'
 Requires-Dist: mistralai; extra == 'dev'
 Requires-Dist: ollama; extra == 'dev'
-Requires-Dist: openai==1.60.0; extra == 'dev'
+Requires-Dist: openai-agents>=0.0.3; extra == 'dev'
+Requires-Dist: openai>=1.60.0; extra == 'dev'
 Requires-Dist: phidata; extra == 'dev'
 Requires-Dist: pinecone; extra == 'dev'
 Requires-Dist: python-dotenv; extra == 'dev'

{langtrace_python_sdk-3.8.3.dist-info → langtrace_python_sdk-3.8.5.dist-info}/RECORD

@@ -115,8 +115,8 @@ examples/vertexai_example/main.py,sha256=gndId5X5ksD-ycxnAWMdEqIDbLc3kz5Vt8vm4YP
 examples/weaviate_example/__init__.py,sha256=8JMDBsRSEV10HfTd-YC7xb4txBjD3la56snk-Bbg2Kw,618
 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=L0ktJXsGjvKS0dc6kWGCOUoNsGTERLEmd-1QfzDppq8,13706
-langtrace_python_sdk/version.py,sha256=suroMOhipFiXYE6cgp5XFXSXtW2pHyZibLMFSNkHUec,22
+langtrace_python_sdk/langtrace.py,sha256=T-DsDrwWaL4gAUK1lkTRRpmvoO7F2WtO5hQZdyrVAxE,13791
+langtrace_python_sdk/version.py,sha256=mb3dZLLIE3dKNa7hv1kuERgx4o1UEUlj7DsxZRc2A38,22
 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
@@ -141,10 +141,10 @@ langtrace_python_sdk/constants/instrumentation/weaviate.py,sha256=gtv-JBxvNGClEM
 langtrace_python_sdk/extensions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/extensions/langtrace_exporter.py,sha256=ckd8dMmY6h2oxE04p1JFLwUB5PSJX_Cy4eDFEM6aj4Y,6605
 langtrace_python_sdk/extensions/langtrace_filesystem.py,sha256=34fZutG28EJ66l67OvTGsydAH3ZpXgikdE7hVLqBpG4,7863
-langtrace_python_sdk/instrumentation/__init__.py,sha256=UOtU6nT-ZT-ZvHzOgPNhG_IooMe5jU8lzgdoDCHpeBc,2469
+langtrace_python_sdk/instrumentation/__init__.py,sha256=DC96oELorZedDf5zooQ6HGi-dmieXwUephgeB_LzfaU,2559
 langtrace_python_sdk/instrumentation/agno/__init__.py,sha256=95fn4oA-CHB0mxc6KnVB20KSbXGl_ZZr9n99EEaXzrY,91
-langtrace_python_sdk/instrumentation/agno/instrumentation.py,sha256=CrwHzkzLpdo2m2nBHE69d8MIFMxZuvTsT1LTxqlTWV0,2664
-langtrace_python_sdk/instrumentation/agno/patch.py,sha256=tbMscAxsZuo4MAEUQ9tUPiluyj8RRcksF7sYU-v2knM,13399
+langtrace_python_sdk/instrumentation/agno/instrumentation.py,sha256=XUnfvqpp13IgdF03xGKasq7kGjeaN1sXLIwCf-Nt_Nc,2667
+langtrace_python_sdk/instrumentation/agno/patch.py,sha256=qCUxCkzU9cYu_d8BzLgj_Ss97qib07tRVYpYDiNnNMs,16876
 langtrace_python_sdk/instrumentation/anthropic/__init__.py,sha256=donrurJAGYlxrSRA3BIf76jGeUcAx9Tq8CVpah68S0Y,101
 langtrace_python_sdk/instrumentation/anthropic/instrumentation.py,sha256=ndXdruI0BG7n75rsuEpKjfzePxrZxg40gZ39ONmD_v4,1845
 langtrace_python_sdk/instrumentation/anthropic/patch.py,sha256=ztPN4VZujoxYOKhTbFnup7Ibms9NAzYCPAJY43NUgKw,4935
@@ -225,8 +225,11 @@ langtrace_python_sdk/instrumentation/openai/__init__.py,sha256=VPHRNCQEdkizIVP2d
 langtrace_python_sdk/instrumentation/openai/instrumentation.py,sha256=PZxI0qfoud1VsKGmJu49YDp0Z9z9TzCR8qxR3uznOMA,2810
 langtrace_python_sdk/instrumentation/openai/patch.py,sha256=mRE150sHGL2dq4oOKTYWek1vup7HuRW_sqP6nNOd3N8,27862
 langtrace_python_sdk/instrumentation/openai/types.py,sha256=aVkoa7tmAbYfyOhnyMrDaVjQuwhmRNLMthlNtKMtWX8,4311
+langtrace_python_sdk/instrumentation/openai_agents/__init__.py,sha256=ElRfFIQYXD2-eRyL3fZnjIsDJLTrDolh5cZHPnZv0q8,107
+langtrace_python_sdk/instrumentation/openai_agents/instrumentation.py,sha256=6M4FHXfem7pazrNgsimebrEfMb2FxI8lHrdEMbVf75Y,1860
+langtrace_python_sdk/instrumentation/openai_agents/patch.py,sha256=zXuIjqnPvWNEu8nCDTRxxvTAi8PwAOsBJEtsFn3PXaI,24012
 langtrace_python_sdk/instrumentation/phidata/__init__.py,sha256=q2v6luvqp9gFf1AJX6YrBvuyMC_q6cEnB5syl2HNPlU,97
-langtrace_python_sdk/instrumentation/phidata/instrumentation.py,sha256=E_u0WUtBukmWjKjkgtJgeCS6ab_2xhcamLOH9cERNCM,2694
+langtrace_python_sdk/instrumentation/phidata/instrumentation.py,sha256=S639XMVOGmpIK9jug9NWrUBLqs1G5ywBZiIhVuCkwGk,2697
 langtrace_python_sdk/instrumentation/phidata/patch.py,sha256=-Jf_20wLLRGRM6sY3RreS-ocXjdq5m33-gxNtl_eUdQ,12133
 langtrace_python_sdk/instrumentation/pinecone/__init__.py,sha256=DzXyGh9_MGWveJvXULkFwdkf7PbG2s3bAWtT1Dmz7Ok,99
 langtrace_python_sdk/instrumentation/pinecone/instrumentation.py,sha256=2jhxGmXPrdSsrETRR_e58Rdc_uoKfRunvsCc660v6QA,1831
@@ -246,7 +249,7 @@ langtrace_python_sdk/instrumentation/weaviate/patch.py,sha256=Lqixz32uAvDA2VLU3z
 langtrace_python_sdk/types/__init__.py,sha256=SJSJzkgPjGGTVJXUZ_FyR3p9DJ5kWGx7iAnJfY4ZYHU,4669
 langtrace_python_sdk/utils/__init__.py,sha256=VVDOG-QLd59ZvSHp0avjof0sbxlZ1QQOf0KoOF7ofhQ,3310
 langtrace_python_sdk/utils/langtrace_sampler.py,sha256=BupNndHbU9IL_wGleKetz8FdcveqHMBVz1bfKTTW80w,1753
-langtrace_python_sdk/utils/llm.py,sha256=693egq-ztVLmlwdZ9KkTTzMpg_k397ErA28XFL00_ws,16733
+langtrace_python_sdk/utils/llm.py,sha256=giJU33LvMPaRjPAjUwBCehgHj_ei1HwM7gLJSVWYLnI,23238
 langtrace_python_sdk/utils/misc.py,sha256=LaQr5LOmZMiuwVdjYh7aIu6o2C_Xb1wgpQGNOVmRzfE,1918
 langtrace_python_sdk/utils/prompt_registry.py,sha256=n5dQMVLBw8aJZY8Utvf67bncc25ELf6AH9BYw8_hSzo,2619
 langtrace_python_sdk/utils/sdk_version_checker.py,sha256=F-VVVH7Fmhr5LcY0IIe-34zIi5RQcx26uuxFpPzZesM,1782
@@ -297,8 +300,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.3.dist-info/METADATA,sha256=Etxr0XXT7aO2v4RKQduS0KHOe5qKWY9Ozid_Ciu3efs,15792
-langtrace_python_sdk-3.8.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langtrace_python_sdk-3.8.3.dist-info/entry_points.txt,sha256=1_b9-qvf2fE7uQNZcbUei9vLpFZBbbh9LrtGw95ssAo,70
-langtrace_python_sdk-3.8.3.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-langtrace_python_sdk-3.8.3.dist-info/RECORD,,
+langtrace_python_sdk-3.8.5.dist-info/METADATA,sha256=oM9Ya7m7DDYBDT1VtPe3uguAKkvdJImtz2WGgSxmLbo,15844
+langtrace_python_sdk-3.8.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langtrace_python_sdk-3.8.5.dist-info/entry_points.txt,sha256=1_b9-qvf2fE7uQNZcbUei9vLpFZBbbh9LrtGw95ssAo,70
+langtrace_python_sdk-3.8.5.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+langtrace_python_sdk-3.8.5.dist-info/RECORD,,