langtrace-python-sdk 2.0.13__py3-none-any.whl → 2.1.1__py3-none-any.whl

langtrace_python_sdk/utils/types.py

@@ -0,0 +1,19 @@
+from typing import TypedDict
+
+
+class LangTraceEvaluation:
+    # Define the class based on your needs
+    pass
+
+
+class LangTraceApiError(Exception):
+    def __init__(self, message: str, status_code: int):
+        super().__init__(message)
+        self.status_code = status_code
+
+
+class EvaluationAPIData(TypedDict):
+    userId: str
+    userScore: int
+    traceId: str
+    spanId: str

langtrace_python_sdk/utils/with_root_span.py

@@ -16,13 +16,24 @@ limitations under the License.
 
 import asyncio
 from functools import wraps
+from typing import Optional
 
+from langtrace_python_sdk.constants.exporter.langtrace_exporter import (
+    LANGTRACE_REMOTE_URL,
+)
+from langtrace_python_sdk.utils.types import (
+    EvaluationAPIData,
+    LangTraceApiError,
+    LangTraceEvaluation,
+)
 from opentelemetry import baggage, context, trace
 from opentelemetry.trace import SpanKind
 
 from langtrace_python_sdk.constants.instrumentation.common import (
     LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY,
 )
+import os
+import requests
 
 
 def with_langtrace_root_span(
@@ -35,17 +46,37 @@ def with_langtrace_root_span(
         def sync_wrapper(*args, **kwargs):
             tracer = trace.get_tracer(__name__)
             operation_name = name if name else func.__name__
+            span_id = None
+            trace_id = None
 
-            with tracer.start_as_current_span(operation_name, kind=kind):
+            with tracer.start_as_current_span(operation_name, kind=kind) as span:
+                span_id = str(span.get_span_context().span_id)
+                trace_id = str(span.get_span_context().trace_id)
 
-                return func(*args, **kwargs)
+                if (
+                    "span_id" in func.__code__.co_varnames
+                    and "trace_id" in func.__code__.co_varnames
+                ):
+                    return func(*args, span_id, trace_id, **kwargs)
+                else:
+                    return func(*args, **kwargs)
 
         @wraps(func)
         async def async_wrapper(*args, **kwargs):
+            span_id = None
+            trace_id = None
             tracer = trace.get_tracer(__name__)
             operation_name = name if name else func.__name__
-            with tracer.start_as_current_span(operation_name, kind=kind):
-                return await func(*args, **kwargs)
+            with tracer.start_as_current_span(operation_name, kind=kind) as span:
+                span_id = span.get_span_context().span_id
+                trace_id = span.get_span_context().trace_id
+                if (
+                    "span_id" in func.__code__.co_varnames
+                    and "trace_id" in func.__code__.co_varnames
+                ):
+                    return await func(*args, span_id, trace_id, **kwargs)
+                else:
+                    return await func(*args, **kwargs)
 
         if asyncio.iscoroutinefunction(func):
             return async_wrapper
@@ -79,3 +110,67 @@ def with_additional_attributes(attributes={}):
             return sync_wrapper
 
     return decorator
+
+
+def send_user_feedback(data: EvaluationAPIData) -> None:
+    try:
+        evaluation = _get_evaluation(data["spanId"])
+        headers = {"x-api-key": os.environ["LANGTRACE_API_KEY"]}
+        if evaluation is not None:
+            # Make a PUT request to update the evaluation
+            response = requests.put(
+                f"{LANGTRACE_REMOTE_URL}/api/evaluation",
+                json=data,
+                params={"spanId": data["spanId"]},
+                headers=headers,
+                timeout=None,
+            )
+            response.raise_for_status()
+            print("SENDING FEEDBACK PUT", response.status_code, response.text)
+
+        else:
+            # Make a POST request to create a new evaluation
+            response = requests.post(
+                f"{LANGTRACE_REMOTE_URL}/api/evaluation",
+                json=data,
+                params={"spanId": data["spanId"]},
+                headers=headers,
+                timeout=None,
+            )
+            print("Response", response.status_code, response.text)
+            response.raise_for_status()
+
+    except requests.RequestException as err:
+        # Handle specific HTTP errors or general request exceptions
+        error_msg = str(err)
+        if err.response:
+            try:
+                # Try to extract server-provided error message
+                error_msg = err.response.json().get("error", error_msg)
+            except ValueError:
+                # Fallback if response is not JSON
+                error_msg = err.response.text
+        raise Exception(f"API error: {error_msg}") from err
+
+
+def _get_evaluation(span_id: str) -> Optional[LangTraceEvaluation]:
+    try:
+        response = requests.get(
+            f"{LANGTRACE_REMOTE_URL}/api/evaluation",
+            params={"spanId": span_id},
+            headers={"x-api-key": os.environ["LANGTRACE_API_KEY"]},
+            timeout=None,
+        )
+        evaluations = response.json()["evaluations"]
+        response.raise_for_status()
+        return None if not evaluations else evaluations[0]
+
+    except requests.RequestException as err:
+        if err.response is not None:
+            message = (
+                err.response.json().get("message", "")
+                if err.response.json().get("message", "")
+                else err.response.text if err.response.text else str(err)
+            )
+            raise LangTraceApiError(message, err.response.status_code)
+        raise LangTraceApiError(str(err), 500)

langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "2.0.13"
+__version__ = "2.1.1"

{langtrace_python_sdk-2.0.13.dist-info → langtrace_python_sdk-2.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 2.0.13
+Version: 2.1.1
 Summary: Python SDK for LangTrace
 Project-URL: Homepage, https://github.com/Scale3-Labs/langtrace-python-sdk
 Author-email: Scale3 Labs <engineering@scale3labs.com>
@@ -17,7 +17,8 @@ Requires-Dist: opentelemetry-instrumentation
 Requires-Dist: opentelemetry-sdk
 Requires-Dist: pinecone-client
 Requires-Dist: tiktoken
-Requires-Dist: trace-attributes==3.0.5
+Requires-Dist: trace-attributes>=4.0.2
+Requires-Dist: weaviate-client
 Provides-Extra: dev
 Requires-Dist: anthropic; extra == 'dev'
 Requires-Dist: chromadb; extra == 'dev'
@@ -25,10 +26,10 @@ Requires-Dist: cohere; extra == 'dev'
 Requires-Dist: langchain; extra == 'dev'
 Requires-Dist: langchain-community; extra == 'dev'
 Requires-Dist: langchain-openai; extra == 'dev'
-Requires-Dist: llama-index; extra == 'dev'
 Requires-Dist: openai; extra == 'dev'
 Requires-Dist: python-dotenv; extra == 'dev'
 Requires-Dist: qdrant-client; extra == 'dev'
+Requires-Dist: weaviate-client; extra == 'dev'
 Provides-Extra: test
 Requires-Dist: pytest; extra == 'test'
 Requires-Dist: pytest-asyncio; extra == 'test'
@@ -65,24 +66,25 @@ To use the managed SaaS version of Langtrace, follow the steps below:
 
 Get started by adding simply three lines to your code!
 
-``` python
+```python
 pip install langtrace-python-sdk
 ```
 
-``` python
+```python
 from langtrace_python_sdk import langtrace # Must precede any llm module imports
 langtrace.init(api_key=<your_api_key>)
 ```
 
 OR
 
-``` python
+```python
 from langtrace_python_sdk import langtrace # Must precede any llm module imports
 langtrace.init() # LANGTRACE_API_KEY as an ENVIRONMENT variable
 ```
 
 ## FastAPI Quick Start
-Initialize FastAPI project and add this inside the ```main.py``` file
+
+Initialize FastAPI project and add this inside the `main.py` file
 
 ```python
 from fastapi import FastAPI
@@ -104,7 +106,9 @@ def root():
 ```
 
 ## Django Quick Start
-Initialize django project and add this inside the ```__init.py__``` file
+
+Initialize django project and add this inside the `__init.py__` file
+
 ```python
 from langtrace_python_sdk import langtrace
 from openai import OpenAI
@@ -122,7 +126,9 @@ client.chat.completions.create(
 ```
 
 ## Flask Quick Start
-Initialize flask project and this inside ```app.py``` file
+
+Initialize flask project and this inside `app.py` file
+
 ```python
 from flask import Flask
 from langtrace_python_sdk import langtrace
@@ -147,11 +153,11 @@ def main():
 
 Get started by adding simply two lines to your code and see traces being logged to the console!
 
-``` python
+```python
 pip install langtrace-python-sdk
 ```
 
-``` python
+```python
 from langtrace_python_sdk import langtrace # Must precede any llm module imports
 langtrace.init(write_spans_to_console=True)
 ```
@@ -160,15 +166,26 @@ langtrace.init(write_spans_to_console=True)
 
 Get started by adding simply three lines to your code and see traces being exported to your remote location!
 
-``` python
+```python
 pip install langtrace-python-sdk
 ```
 
-``` python
+```python
 from langtrace_python_sdk import langtrace # Must precede any llm module imports
 langtrace.init(custom_remote_exporter=<your_exporter>, batch=<True or False>)
 ```
 
+### Configure Langtrace
+
+| Parameter                  | Type                                | Default Value                 | Description                                                                    |
+| -------------------------- | ----------------------------------- | ----------------------------- | ------------------------------------------------------------------------------ |
+| `api_key`                  | `str`                               | `LANGTRACE_API_KEY` or `None` | The API key for authentication.                                                |
+| `batch`                    | `bool`                              | `True`                        | Whether to batch spans before sending them.                                    |
+| `write_spans_to_console`   | `bool`                              | `False`                       | Whether to write spans to the console.                                         |
+| `custom_remote_exporter`   | `Optional[Exporter]`                | `None`                        | Custom remote exporter. If `None`, a default `LangTraceExporter` will be used. |
+| `api_host`                 | `Optional[str]`                     | `https://langtrace.ai/`       | The API host for the remote exporter.                                          |
+| `disable_instrumentations` | `Optional[DisableInstrumentations]` | `None`                        | You can pass an object to disable instrumentation for specific vendors ex: `{'only': ['openai']}` or `{'all_except': ['openai']}`
+
 ### Additional Customization
 
 - `@with_langtrace_root_span` - this decorator is designed to organize and relate different spans, in a hierarchical manner. When you're performing multiple operations that you want to monitor together as a unit, this function helps by establishing a "parent" (`LangtraceRootSpan` or whatever is passed to `name`) span. Then, any calls to the LLM APIs made within the given function (fn) will be considered "children" of this parent span. This setup is especially useful for tracking the performance or behavior of a group of operations collectively, rather than individually.
@@ -186,7 +203,6 @@ def example():
     return response
 ```
 
-
 - `with_additional_attributes` - this function is designed to enhance the traces by adding custom attributes to the current context. These custom attributes provide extra details about the operations being performed, making it easier to analyze and understand their behavior.
 
 ```python
@@ -223,7 +239,6 @@ def chat_completion():
 
 - `get_prompt_from_registry` - this function is designed to fetch the desired prompt from the `Prompt Registry`. You can pass two options for filtering `prompt_version` & `variables`.
 
-
 ```python
 from langtrace_python_sdk import get_prompt_from_registry
 

{langtrace_python_sdk-2.0.13.dist-info → langtrace_python_sdk-2.1.1.dist-info}/RECORD

@@ -32,23 +32,25 @@ examples/openai_example/tool_calling_nonstreaming.py,sha256=Yc848IooZRXNynHL6z0k
 examples/openai_example/tool_calling_streaming.py,sha256=mV1RbyAoVhumGRPpqPWQ6PMhnJyeifrlELd2-K1qJ_w,7015
 examples/perplexity_example/basic.py,sha256=bp7n27gaugJkaFVyt8pjaEfi66lYcqP6eFFjPewUShY,668
 examples/pinecone_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/pinecone_example/basic.py,sha256=uwWu_UFuDtthgs-x8gOPnsUGOeBDCVPEoiz9uwSEjfg,1007
+examples/pinecone_example/basic.py,sha256=ymSguqRssQ4fbzJ2MMV-z9FZwlTvUyApt2AykiJQiF4,1137
 examples/qdrant_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 examples/qdrant_example/basic.py,sha256=DCMjHSuBZKkhEjCkwy5d5La9WMyW0lCWqtcZWiFCEm4,1425
-langtrace_python_sdk/__init__.py,sha256=_r2S-lLZFTFFvWjf0lWBX4nzkJN5R7FHX0Dnmr76CEk,848
-langtrace_python_sdk/langtrace.py,sha256=o0xJInE_dwNBkAfIQRqMKPz6glCAcGdRJxEG8kasq8I,6839
-langtrace_python_sdk/version.py,sha256=qRJzfkgUc9bpxpBssrO665ZtIyUM0fsaZDIbahg3TLU,23
+examples/weaviate_example/query_text.py,sha256=s3f0hYYi-xdbh_LlTBDHegQiVi-4bW_eACb2EnEj8J4,64533
+langtrace_python_sdk/__init__.py,sha256=R4dv73QARUw4WRfCdSKkKzZTLDumF9jcSdwtVrzWhb4,962
+langtrace_python_sdk/langtrace.py,sha256=2VVVeWX3b20jwDY9iqDTxeE-Zunfgt2gfgRs3SreZLU,6993
+langtrace_python_sdk/version.py,sha256=zPJIgPGcoSNiD0qme18OnYJYE3A9VVytlhO-V5DaAW0,22
 langtrace_python_sdk/constants/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/constants/exporter/langtrace_exporter.py,sha256=5MNjnAOg-4am78J3gVMH6FSwq5N8TOj72ugkhsw4vi0,46
 langtrace_python_sdk/constants/instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/constants/instrumentation/anthropic.py,sha256=YX3llt3zwDY6XrYk3CB8WEVqgrzRXEw_ffyk56JoF3k,126
 langtrace_python_sdk/constants/instrumentation/chroma.py,sha256=hiPGYdHS0Yj4Kh3eaYBbuCAl_swqIygu80yFqkOgdak,955
 langtrace_python_sdk/constants/instrumentation/cohere.py,sha256=tf9sDfb5K3qOAHChEE5o8eYWPZ1io58VsOjZDCZPxfw,577
-langtrace_python_sdk/constants/instrumentation/common.py,sha256=v2daYzs3nul5pSClKGnzpe4vIxmXztJ_2ohveV-wMRU,754
+langtrace_python_sdk/constants/instrumentation/common.py,sha256=TCHX8uqIJD2p9sONGrlX6Q3b5Mj859PTP-c729ogKcY,782
 langtrace_python_sdk/constants/instrumentation/groq.py,sha256=VFXmIl4aqGY_fS0PAmjPj_Qm7Tibxbx7Ur_e7rQpqXc,134
 langtrace_python_sdk/constants/instrumentation/openai.py,sha256=9VF6ic9Ed3bpSvdp6iNmrpx2Ppo6DPav3hoUcqSQSv0,1048
 langtrace_python_sdk/constants/instrumentation/pinecone.py,sha256=Xaqqw-xBO0JJLGk75hiCUQGztNm0HiVaLQvjtYK7VJM,472
 langtrace_python_sdk/constants/instrumentation/qdrant.py,sha256=yL7BopNQTXW7L7Z-gVM2PdusKD7r9qqcATvczFd7NtQ,1999
+langtrace_python_sdk/constants/instrumentation/weaviate.py,sha256=Iytf2OpB_irZYEmvOQ7Pf483EdG5Bh59GxaBlXck0yY,1501
 langtrace_python_sdk/extensions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/extensions/langtrace_exporter.py,sha256=s-4zc7lba1va_JUReFdLgyBKlvwV1KmBt_OzVmWReQg,3610
 langtrace_python_sdk/instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -57,7 +59,7 @@ langtrace_python_sdk/instrumentation/anthropic/instrumentation.py,sha256=-srgE8q
 langtrace_python_sdk/instrumentation/anthropic/patch.py,sha256=koAo6bA6nnRDk01fcrArqnyAmgzjxV20ofITMQyKVws,8427
 langtrace_python_sdk/instrumentation/chroma/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/chroma/instrumentation.py,sha256=nT6PS6bsrIOO9kLV5GuUeRjMe6THHHAZGvqWBP1dYog,1807
-langtrace_python_sdk/instrumentation/chroma/patch.py,sha256=qytjkZxntUe7XyAvVoiEvFgUIXNU52o9BnYLAnQ9M20,2809
+langtrace_python_sdk/instrumentation/chroma/patch.py,sha256=1FvA4YvGw9temfBC2g33xGEzIcL1rVdABXwMVTPm3aM,8750
 langtrace_python_sdk/instrumentation/cohere/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/cohere/instrumentation.py,sha256=YQFHZIBd7SSPD4b6Va-ZR0thf_AuBCqj5yzHLHJVWnM,2121
 langtrace_python_sdk/instrumentation/cohere/patch.py,sha256=hdSWArP0cdfGs0Q5KhkaowkwrYQIzxvF8NpY4sg8p2Q,26563
@@ -81,20 +83,24 @@ langtrace_python_sdk/instrumentation/llamaindex/instrumentation.py,sha256=8iAg-O
 langtrace_python_sdk/instrumentation/llamaindex/patch.py,sha256=q99qmEBNH54Oi_sHCIGBQFpIE4zaNKVjIWARlX8fJ-M,4189
 langtrace_python_sdk/instrumentation/openai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/openai/instrumentation.py,sha256=G2HSZfr6DuP2n-8v0h91M60m0DJWFBcru4-1FQJl-5A,2726
-langtrace_python_sdk/instrumentation/openai/patch.py,sha256=kQxc_E1IlOgHZnlc0HbpGjM5rZCDR3YpNFChp2X9GbM,37263
+langtrace_python_sdk/instrumentation/openai/patch.py,sha256=xi87lLHoj81xSx6MiqTAKWr3MLxXpMf84NzJ5uDm134,37288
 langtrace_python_sdk/instrumentation/pinecone/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/pinecone/instrumentation.py,sha256=52GbkfRK-sxXBXhHmHRRO2BGNd6zZVHDp8njLS1CQX0,2314
-langtrace_python_sdk/instrumentation/pinecone/patch.py,sha256=hLgYWDKTytuvTORDd6dupMpKAp19vCgOAXDOdjvWme0,4832
+langtrace_python_sdk/instrumentation/pinecone/patch.py,sha256=VNZ_fCRpVQNXeg0G8hFUidBelV3L91-A_OtkKT7ZWUM,4889
 langtrace_python_sdk/instrumentation/qdrant/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/qdrant/instrumentation.py,sha256=vl2eKSP55aqDo1JiRlvOUBrr6kddvG9Z5dCYew2OG08,1816
-langtrace_python_sdk/instrumentation/qdrant/patch.py,sha256=x95PXaJe_GrOG51zZCDuchVA-bzuA7s5LHqOoWrqSA0,3883
-langtrace_python_sdk/types/__init__.py,sha256=QYK-kRkyfXDdzjjKmXVxi0vZ6Wpbxdf0CoZbTXO0JT8,787
-langtrace_python_sdk/utils/__init__.py,sha256=is6gOLiHcvrN9kPgYnUwyzkgz9vInVQ4jGe7k0zRHrA,150
-langtrace_python_sdk/utils/llm.py,sha256=gnmJ6YWzzdOCaVFg1C-iyNcMGDyiJ623qZeOwDYUfIQ,2162
+langtrace_python_sdk/instrumentation/qdrant/patch.py,sha256=KcUWmvnS3PgPStzs1oWmlZsyomKBl3dmg8nXuR0T1C0,4654
+langtrace_python_sdk/instrumentation/weaviate/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+langtrace_python_sdk/instrumentation/weaviate/instrumentation.py,sha256=i5xi0uWY1pPvGWyedFEvOtbRVT7ybFnMia-9TYFigOA,2309
+langtrace_python_sdk/instrumentation/weaviate/patch.py,sha256=ig2fc33hNydEcH5cJWRycFnMXiXiJr731J-Vg5Ze4No,5634
+langtrace_python_sdk/types/__init__.py,sha256=-j8cuz3bhUdOqj7N2c0w5y-j3UmcxwEgNh8BWeXwHoI,813
+langtrace_python_sdk/utils/__init__.py,sha256=MbBqT4NpDCNTqqvZ0lG4tRiFFZ-tlM8Dwophg4xyrRw,148
+langtrace_python_sdk/utils/llm.py,sha256=CiASOvObFvsN6T7ogWywNXfXGzI__u9misgolLxyeZk,2161
+langtrace_python_sdk/utils/misc.py,sha256=CD9NWRLxLpFd0YwlHJqzlpFNedXVWtAKGOjQWnDCo8k,838
 langtrace_python_sdk/utils/prompt_registry.py,sha256=7FFB4Pj0414qgf02h5zL5vXBZgNBf74g4Iq7GdFaIO0,2689
 langtrace_python_sdk/utils/silently_fail.py,sha256=F_9EteXCO9Cyq-8MA1OT2Zy_dx8n06nt31I7t7ui24E,478
-langtrace_python_sdk/utils/types.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk/utils/with_root_span.py,sha256=MXF-FmG47P0GtJQNffKM_3XGrJHVAC4hMB6GmwtnA4M,2454
+langtrace_python_sdk/utils/types.py,sha256=l-N6o7cnWUyrD6dBvW7W3Pf5CkPo5QaoT__k1XLbrQg,383
+langtrace_python_sdk/utils/with_root_span.py,sha256=9OkVMzoH8Ydqt7OpWPONiGrmHHBnEOXY034St2UToCE,6095
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/conftest.py,sha256=0Jo6iCZTXbdvyJVhG9UpYGkLabL75378oauCzmt-Sa8,603
 tests/utils.py,sha256=hP8sTH-M8WO6zlLkSFHPf6483ZcKEcnxul6JIIb1pLM,1396
@@ -109,10 +115,10 @@ tests/cohere/conftest.py,sha256=jBbyg-tut-ZJN5_5D64sGmaPIhT_nQQQAiW43kl5Rz4,621
 tests/cohere/test_cohere_chat.py,sha256=MioEIXB5e-4t99Vt-KpLlxd4diPbp3S6eNfnPWjEDXM,4564
 tests/cohere/test_cohere_embed.py,sha256=bFKDzUnypMyLefAOHR-6dEgvt0yys2Yw7-F0zc0uQyc,1250
 tests/cohere/test_cohere_rerank.py,sha256=ZDzvXsY9l1fiytOAKGQidpBkop3fxV6d8xEy_gQt9jE,2279
-tests/cohere/cassettes/test_cohere_chat.yaml,sha256=_Az6Kfhp1XmoK6zwTxqbnRYTmsm0k88McOidwQaRp3g,3436
-tests/cohere/cassettes/test_cohere_chat_streaming.yaml,sha256=TdWe64Pyfa5Ho9lqTMIW41YntUmUwmYQc55qWntbvas,10610
-tests/cohere/cassettes/test_cohere_embed.yaml,sha256=YAxJ3otY9N6jJQOKi50nQA4xY9-dknm1oXEP8KiCH5U,27312
-tests/cohere/cassettes/test_cohere_rerank.yaml,sha256=m5EAKAM5JEcGMyUlmjaxU-cNLiXKcenSFQyU-UaUI0U,2392
+tests/cohere/cassettes/test_cohere_chat.yaml,sha256=iQIhJwmWe2AQDmdguL6L0SZOPMD6B3mhDVUKPSLodA4,3271
+tests/cohere/cassettes/test_cohere_chat_streaming.yaml,sha256=tMlz7IPZYRCJUeMCwyXrA4O4zpcqIPpQ5hU9I5f352Q,8174
+tests/cohere/cassettes/test_cohere_embed.yaml,sha256=Svcyw8WkVMVw4cocWQzGOdUVmWi5AM6Gb9t9VLlyOjM,27320
+tests/cohere/cassettes/test_cohere_rerank.yaml,sha256=RsJ8804lTktI0FKEFCw0M4ztStZbCxb_z_VRmrw8ooQ,2393
 tests/langchain/test_langchain.py,sha256=p1YkeL1U23b0E3vhEcXonwMa3QGUswQjKNReMbV0ndA,2576
 tests/langchain/test_langchain_community.py,sha256=B7Z7YRx7v9507wEyDnXV1NKVYo4r2XU0YplugfYVsJ0,2615
 tests/langchain/test_langchain_core.py,sha256=LBeoK27-uDethSvBGykYAlCVfEwljQrN7sZL91TVCPw,4467
@@ -131,7 +137,7 @@ 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-2.0.13.dist-info/METADATA,sha256=AG-wRGpBbg8pDHDvU6lyVf0c3puz-ANxrmJVYlcvX8U,10173
-langtrace_python_sdk-2.0.13.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
-langtrace_python_sdk-2.0.13.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-langtrace_python_sdk-2.0.13.dist-info/RECORD,,
+langtrace_python_sdk-2.1.1.dist-info/METADATA,sha256=QyEMBrnv_vd8SjrZdVVtifuCLr9VlCjUKCefnT28O7c,11722
+langtrace_python_sdk-2.1.1.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
+langtrace_python_sdk-2.1.1.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+langtrace_python_sdk-2.1.1.dist-info/RECORD,,

tests/cohere/cassettes/test_cohere_chat.yaml

@@ -17,7 +17,7 @@ interactions:
       content-type:
       - application/json
       host:
-      - api.cohere.ai
+      - api.cohere.com
       user-agent:
       - python-httpx/0.27.0
       x-fern-language:
@@ -25,27 +25,25 @@ interactions:
       x-fern-sdk-name:
       - cohere
       x-fern-sdk-version:
-      - 5.3.2
+      - 5.5.3
     method: POST
-    uri: https://api.cohere.ai/v1/chat
+    uri: https://api.cohere.com/v1/chat
   response:
     body:
-      string: '{"response_id":"27cf0e2e-b4d5-427b-9866-d108307022de","text":"Argh,
+      string: '{"response_id":"6b2a2c29-2cd4-4e79-a05a-db8a6019b757","text":"Argh,
         many years ago, a bold explorer named Isaac Newton sat beneath an apple tree
-        for many hours, contemplating the nature of the universe and the motions of
-        the heavenly bodies. He posited that all objects in the universe are attracted
-        to each other. Then one day, his theory was proven true when a juicy apple
-        fell from the tree and struck him on the head, inspiring his theory of gravity
-        and changing science forever.","generation_id":"2a7babaa-f1da-468f-9891-5171f137ba04","chat_history":[{"role":"USER","message":"Who
+        for many hours, contemplating the nature of the universe. Suddenly, an apple
+        fell from the sky, striking him on the head. This inspired him to devise a
+        theory about the forces that govern objects on Earth and altered our understanding
+        of the universe forevermore.","generation_id":"f3b5a0d7-409b-45a7-9324-5b89c944457e","chat_history":[{"role":"USER","message":"Who
         discovered gravity?"},{"role":"CHATBOT","message":"The man who is widely credited
         with discovering gravity is Sir Isaac Newton"},{"role":"USER","message":"Tell
         me a story in 3 sentences or less?"},{"role":"CHATBOT","message":"Argh, many
         years ago, a bold explorer named Isaac Newton sat beneath an apple tree for
-        many hours, contemplating the nature of the universe and the motions of the
-        heavenly bodies. He posited that all objects in the universe are attracted
-        to each other. Then one day, his theory was proven true when a juicy apple
-        fell from the tree and struck him on the head, inspiring his theory of gravity
-        and changing science forever."}],"finish_reason":"COMPLETE","meta":{"api_version":{"version":"1"},"billed_units":{"input_tokens":31,"output_tokens":85},"tokens":{"input_tokens":49,"output_tokens":85}},"documents":[]}'
+        many hours, contemplating the nature of the universe. Suddenly, an apple fell
+        from the sky, striking him on the head. This inspired him to devise a theory
+        about the forces that govern objects on Earth and altered our understanding
+        of the universe forevermore."}],"finish_reason":"COMPLETE","meta":{"api_version":{"version":"1"},"billed_units":{"input_tokens":31,"output_tokens":69},"tokens":{"input_tokens":49,"output_tokens":69}},"documents":[]}'
     headers:
       Alt-Svc:
       - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
@@ -56,17 +54,17 @@ interactions:
       cache-control:
       - no-cache, no-store, no-transform, must-revalidate, private, max-age=0
       content-length:
-      - '1420'
+      - '1270'
       content-type:
       - application/json
       date:
-      - Fri, 26 Apr 2024 13:51:11 GMT
+      - Fri, 24 May 2024 09:38:02 GMT
       expires:
       - Thu, 01 Jan 1970 00:00:00 UTC
       num_chars:
       - '220'
       num_tokens:
-      - '116'
+      - '100'
       pragma:
       - no-cache
       server:
@@ -76,15 +74,15 @@ interactions:
       x-accel-expires:
       - '0'
       x-debug-trace-id:
-      - 6e0f777a5f86c8b5453bf374fd3fadbd
+      - 68cf0a1a2924c2a2534e350092097d49
       x-endpoint-monthly-call-limit:
       - '1000'
       x-envoy-upstream-service-time:
-      - '3037'
+      - '2573'
       x-trial-endpoint-call-limit:
-      - '40'
+      - '10'
       x-trial-endpoint-call-remaining:
-      - '39'
+      - '9'
     status:
       code: 200
       message: OK