langtrace-python-sdk 2.1.25__tar.gz → 2.1.27__tar.gz

{langtrace_python_sdk-2.1.25 → langtrace_python_sdk-2.1.27}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 2.1.25
+Version: 2.1.27
 Summary: Python SDK for LangTrace
 Project-URL: Homepage, https://github.com/Scale3-Labs/langtrace-python-sdk
 Author-email: Scale3 Labs <engineering@scale3labs.com>
@@ -282,11 +282,14 @@ Langtrace automatically captures traces from the following vendors:
 | Cohere       | LLM             | :white_check_mark: | :white_check_mark: |
 | Groq         | LLM             | :x:                | :white_check_mark: |
 | Langchain    | Framework       | :x:                | :white_check_mark: |
+| Langgraph    | Framework       | :x:                | :white_check_mark: |
 | LlamaIndex   | Framework       | :white_check_mark: | :white_check_mark: |
 | DSPy         | Framework       | :x:                | :white_check_mark: |
 | CrewAI       | Framework       | :x:                | :white_check_mark: |
+| Ollama       | Framework       | :x:                | :white_check_mark: |
 | Pinecone     | Vector Database | :white_check_mark: | :white_check_mark: |
 | ChromaDB     | Vector Database | :white_check_mark: | :white_check_mark: |
+| Weaviate     | Vector Database | :white_check_mark: | :white_check_mark: |
 | QDrant       | Vector Database | :x:                | :white_check_mark: |
 
 ---
@@ -302,7 +305,19 @@ Langtrace automatically captures traces from the following vendors:
 
 We welcome contributions to this project. To get started, fork this repository and start developing. To get involved, join our [Discord](https://discord.langtrace.ai) workspace.
 
-If you want to run any of the examples go to `run_example.py` file, you will find `ENABLED_EXAMPLES`. choose the example you want to run and just toggle the flag to `True` and run the file using `python src/run_example.py`
+- If you want to run any of the examples go to `run_example.py` file, you will find `ENABLED_EXAMPLES`. choose the example you want to run and just toggle the flag to `True` and run the file using `python src/run_example.py`
+
+- If you want to run tests, make sure to install dev & test dependencies:
+
+  ```python
+  pip install '.[test]' && pip install '.[dev]'
+  ```
+
+  then run `pytest` using:
+
+  ```python
+  pytest -v
+  ```
 
 ---
 

{langtrace_python_sdk-2.1.25 → langtrace_python_sdk-2.1.27}/README.md

@@ -240,11 +240,14 @@ Langtrace automatically captures traces from the following vendors:
 | Cohere       | LLM             | :white_check_mark: | :white_check_mark: |
 | Groq         | LLM             | :x:                | :white_check_mark: |
 | Langchain    | Framework       | :x:                | :white_check_mark: |
+| Langgraph    | Framework       | :x:                | :white_check_mark: |
 | LlamaIndex   | Framework       | :white_check_mark: | :white_check_mark: |
 | DSPy         | Framework       | :x:                | :white_check_mark: |
 | CrewAI       | Framework       | :x:                | :white_check_mark: |
+| Ollama       | Framework       | :x:                | :white_check_mark: |
 | Pinecone     | Vector Database | :white_check_mark: | :white_check_mark: |
 | ChromaDB     | Vector Database | :white_check_mark: | :white_check_mark: |
+| Weaviate     | Vector Database | :white_check_mark: | :white_check_mark: |
 | QDrant       | Vector Database | :x:                | :white_check_mark: |
 
 ---
@@ -260,7 +263,19 @@ Langtrace automatically captures traces from the following vendors:
 
 We welcome contributions to this project. To get started, fork this repository and start developing. To get involved, join our [Discord](https://discord.langtrace.ai) workspace.
 
-If you want to run any of the examples go to `run_example.py` file, you will find `ENABLED_EXAMPLES`. choose the example you want to run and just toggle the flag to `True` and run the file using `python src/run_example.py`
+- If you want to run any of the examples go to `run_example.py` file, you will find `ENABLED_EXAMPLES`. choose the example you want to run and just toggle the flag to `True` and run the file using `python src/run_example.py`
+
+- If you want to run tests, make sure to install dev & test dependencies:
+
+  ```python
+  pip install '.[test]' && pip install '.[dev]'
+  ```
+
+  then run `pytest` using:
+
+  ```python
+  pytest -v
+  ```
 
 ---
 

{langtrace_python_sdk-2.1.25 → langtrace_python_sdk-2.1.27}/src/langtrace_python_sdk/instrumentation/openai/patch.py

@@ -28,7 +28,11 @@ from langtrace_python_sdk.constants.instrumentation.common import (
     SERVICE_PROVIDERS,
 )
 from langtrace_python_sdk.constants.instrumentation.openai import APIS
-from langtrace_python_sdk.utils.llm import calculate_prompt_tokens, estimate_tokens
+from langtrace_python_sdk.utils.llm import (
+    calculate_prompt_tokens,
+    estimate_tokens,
+    get_tool_calls,
+)
 from openai._types import NOT_GIVEN
 
 
@@ -430,9 +434,10 @@ def chat_completions_create(original_method, version, tracer):
         # handle tool calls in the kwargs
         llm_prompts = []
         for item in kwargs.get("messages", []):
-            if hasattr(item, "tool_calls") and item.tool_calls is not None:
+            tools = get_tool_calls(item)
+            if tools is not None:
                 tool_calls = []
-                for tool_call in item.tool_calls:
+                for tool_call in tools:
                     tool_call_dict = {
                         "id": tool_call.id if hasattr(tool_call, "id") else "",
                         "type": tool_call.type if hasattr(tool_call, "type") else "",
@@ -611,9 +616,10 @@ def async_chat_completions_create(original_method, version, tracer):
         # handle tool calls in the kwargs
         llm_prompts = []
         for item in kwargs.get("messages", []):
-            if hasattr(item, "tool_calls") and item.tool_calls is not None:
+            tools = get_tool_calls(item)
+            if tools is not None:
                 tool_calls = []
-                for tool_call in item.tool_calls:
+                for tool_call in tools:
                     tool_call_dict = {
                         "id": tool_call.id if hasattr(tool_call, "id") else "",
                         "type": tool_call.type if hasattr(tool_call, "type") else "",

{langtrace_python_sdk-2.1.25 → langtrace_python_sdk-2.1.27}/src/langtrace_python_sdk/utils/llm.py

@@ -65,3 +65,15 @@ def set_span_attributes(span, name, value):
         if value != "":
             span.set_attribute(name, value)
     return
+
+
+def get_tool_calls(item):
+    if isinstance(item, dict):
+        if "tool_calls" in item and item["tool_calls"] is not None:
+            return item["tool_calls"]
+        return None
+
+    else:
+        if hasattr(item, "tool_calls") and item.tool_calls is not None:
+            return item.tool_calls
+        return None

langtrace_python_sdk-2.1.27/src/langtrace_python_sdk/version.py

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

langtrace_python_sdk-2.1.25/src/langtrace_python_sdk/version.py

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