langtrace-python-sdk 2.1.0__tar.gz → 2.1.2__tar.gz

{langtrace_python_sdk-2.1.0 → langtrace_python_sdk-2.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 2.1.0
+Version: 2.1.2
 Summary: Python SDK for LangTrace
 Project-URL: Homepage, https://github.com/Scale3-Labs/langtrace-python-sdk
 Author-email: Scale3 Labs <engineering@scale3labs.com>
@@ -10,15 +10,13 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.9
-Requires-Dist: opentelemetry-api
+Requires-Dist: opentelemetry-api>=1.24.0
 Requires-Dist: opentelemetry-exporter-otlp-proto-grpc>=1.24.0
 Requires-Dist: opentelemetry-exporter-otlp-proto-http>=1.24.0
-Requires-Dist: opentelemetry-instrumentation
-Requires-Dist: opentelemetry-sdk
-Requires-Dist: pinecone-client
-Requires-Dist: tiktoken
-Requires-Dist: trace-attributes>=4.0.2
-Requires-Dist: weaviate-client
+Requires-Dist: opentelemetry-instrumentation>=0.45b0
+Requires-Dist: opentelemetry-sdk>=1.24.0
+Requires-Dist: tiktoken>=0.1.1
+Requires-Dist: trace-attributes>=4.0.4
 Provides-Extra: dev
 Requires-Dist: anthropic; extra == 'dev'
 Requires-Dist: chromadb; extra == 'dev'
@@ -27,6 +25,7 @@ Requires-Dist: langchain; extra == 'dev'
 Requires-Dist: langchain-community; extra == 'dev'
 Requires-Dist: langchain-openai; extra == 'dev'
 Requires-Dist: openai; extra == 'dev'
+Requires-Dist: pinecone-client; extra == 'dev'
 Requires-Dist: python-dotenv; extra == 'dev'
 Requires-Dist: qdrant-client; extra == 'dev'
 Requires-Dist: weaviate-client; extra == 'dev'

{langtrace_python_sdk-2.1.0 → langtrace_python_sdk-2.1.2}/pyproject.toml

@@ -18,15 +18,13 @@ classifiers=[
     "Operating System :: OS Independent",
 ]
 dependencies = [
-  'trace-attributes>=4.0.2',
-  'opentelemetry-api',
-  'opentelemetry-sdk',
-  'opentelemetry-instrumentation',
-  'pinecone-client',
-  'tiktoken',
+  'trace-attributes>=4.0.4',
+  'opentelemetry-api>=1.24.0',
+  'opentelemetry-sdk>=1.24.0',
+  'opentelemetry-instrumentation>=0.45b0',
+  'tiktoken>=0.1.1',
   'opentelemetry-exporter-otlp-proto-http>=1.24.0',
   'opentelemetry-exporter-otlp-proto-grpc>=1.24.0',
-  'weaviate-client',
 ]
 
 requires-python = ">=3.9"
@@ -38,6 +36,7 @@ dev = [
     "chromadb",
     'qdrant-client',
     "python-dotenv",
+    'pinecone-client',
     "langchain",
     'langchain-community',
     "langchain-openai",

{langtrace_python_sdk-2.1.0 → langtrace_python_sdk-2.1.2}/src/examples/pinecone_example/basic.py

@@ -2,13 +2,12 @@
 This example demonstrates how to use Pinecone with Langtrace.
 """
 
-from langtrace_python_sdk import langtrace
 from dotenv import find_dotenv, load_dotenv
 from openai import OpenAI
-from pinecone import Pinecone
+from pinecone import Pinecone, ServerlessSpec
 
-
-from langtrace_python_sdk import with_langtrace_root_span, send_user_feedback
+from langtrace_python_sdk import langtrace, with_langtrace_root_span
+from langtrace_python_sdk.utils.with_root_span import send_user_feedback
 
 _ = load_dotenv(find_dotenv())
 
@@ -17,6 +16,17 @@ langtrace.init()
 client = OpenAI()
 pinecone = Pinecone()
 
+PINECONE_INDEX_NAME = "test-index"
+
+
+def create_index():
+    pinecone.create_index(
+        name=PINECONE_INDEX_NAME,
+        dimension=1536,
+        metric="cosine",
+        spec=ServerlessSpec(cloud="aws", region="us-east-1"),
+    )
+
 
 @with_langtrace_root_span()
 def basic(span_id, trace_id):
@@ -32,7 +42,7 @@ def basic(span_id, trace_id):
     unique_id = "unique_random_id"
     data_to_upsert = {"id": unique_id, "values": embedding}
 
-    index = pinecone.Index("test-index")
+    index = pinecone.Index(PINECONE_INDEX_NAME)
     res = index.upsert(vectors=[data_to_upsert], namespace="test-namespace")
     print("res", res)
 
@@ -43,3 +53,7 @@ def basic(span_id, trace_id):
         {"spanId": span_id, "traceId": trace_id, "userScore": 1, "userId": "123"}
     )
     print(resp)
+
+
+# create_index()
+basic("span_id", "trace_id")

{langtrace_python_sdk-2.1.0 → langtrace_python_sdk-2.1.2}/src/langtrace_python_sdk/instrumentation/pinecone/instrumentation.py

@@ -18,7 +18,6 @@ import importlib.metadata
 import logging
 from typing import Collection
 
-import pinecone
 from langtrace.trace_attributes import PineconeMethods
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 from opentelemetry.trace import get_tracer
@@ -42,21 +41,12 @@ class PineconeInstrumentation(BaseInstrumentor):
         tracer = get_tracer(__name__, "", tracer_provider)
         version = importlib.metadata.version("pinecone-client")
         for operation_name, details in APIS.items():
-            method_ref = details["METHOD"]
-            method = None
-            if method_ref is PineconeMethods.UPSERT.value:
-                method = pinecone.Index.upsert
-            elif method_ref is PineconeMethods.QUERY.value:
-                method = pinecone.Index.query
-            elif method_ref is PineconeMethods.DELETE.value:
-                method = pinecone.Index.delete
             operation = details["OPERATION"]
-
             # Dynamically creating the patching call
             wrap_function_wrapper(
                 "pinecone.data.index",
                 f"Index.{operation}",
-                generic_patch(method, operation_name, version, tracer),
+                generic_patch(operation_name, version, tracer),
             )
 
     def _uninstrument(self, **kwargs):

{langtrace_python_sdk-2.1.0 → langtrace_python_sdk-2.1.2}/src/langtrace_python_sdk/instrumentation/pinecone/patch.py

@@ -14,9 +14,9 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
+import json
+
 from langtrace.trace_attributes import DatabaseSpanAttributes
-from langtrace_python_sdk.utils import set_span_attribute
-from langtrace_python_sdk.utils.silently_fail import silently_fail
 from opentelemetry import baggage
 from opentelemetry.trace import SpanKind
 from opentelemetry.trace.status import Status, StatusCode
@@ -26,15 +26,16 @@ from langtrace_python_sdk.constants.instrumentation.common import (
     SERVICE_PROVIDERS,
 )
 from langtrace_python_sdk.constants.instrumentation.pinecone import APIS
-import json
+from langtrace_python_sdk.utils import set_span_attribute
+from langtrace_python_sdk.utils.silently_fail import silently_fail
 
 
-def generic_patch(original_method, method, version, tracer):
+def generic_patch(operation_name, version, tracer):
     """
     A generic patch method that wraps a function with a span"""
 
     def traced_method(wrapped, instance, args, kwargs):
-        api = APIS[method]
+        api = APIS[operation_name]
         service_provider = SERVICE_PROVIDERS["PINECONE"]
         extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
 
@@ -56,7 +57,7 @@ def generic_patch(original_method, method, version, tracer):
 
             if span.is_recording():
                 set_span_attribute(span, "server.address", instance._config.host)
-                if method == "QUERY":
+                if operation_name == "QUERY":
                     set_query_input_attributes(span, kwargs)
 
             for field, value in attributes.model_dump(by_alias=True).items():
@@ -64,7 +65,7 @@ def generic_patch(original_method, method, version, tracer):
                     span.set_attribute(field, value)
             try:
                 # Attempt to call the original method
-                result = original_method(instance, *args, **kwargs)
+                result = wrapped(*args, **kwargs)
                 if result:
                     if span.is_recording():
                         set_query_response_attributes(span, result)

{langtrace_python_sdk-2.1.0 → langtrace_python_sdk-2.1.2}/src/langtrace_python_sdk/instrumentation/weaviate/instrumentation.py

@@ -26,7 +26,7 @@ from langtrace_python_sdk.instrumentation.weaviate.patch import (
     generic_collection_patch,
     generic_query_patch,
 )
-from src.langtrace_python_sdk.constants.instrumentation.weaviate import APIS
+from langtrace_python_sdk.constants.instrumentation.weaviate import APIS
 
 logging.basicConfig(level=logging.FATAL)
 

{langtrace_python_sdk-2.1.0 → langtrace_python_sdk-2.1.2}/src/langtrace_python_sdk/utils/with_root_span.py

@@ -15,25 +15,25 @@ limitations under the License.
 """
 
 import asyncio
+import os
 from functools import wraps
 from typing import Optional
 
+import requests
+from opentelemetry import baggage, context, trace
+from opentelemetry.trace import SpanKind
+
 from langtrace_python_sdk.constants.exporter.langtrace_exporter import (
     LANGTRACE_REMOTE_URL,
 )
+from langtrace_python_sdk.constants.instrumentation.common import (
+    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY,
+)
 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(
@@ -161,7 +161,7 @@ def _get_evaluation(span_id: str) -> Optional[LangTraceEvaluation]:
             headers={"x-api-key": os.environ["LANGTRACE_API_KEY"]},
             timeout=None,
         )
-        evaluations = response.json()["evaluations"]
+        evaluations = response.json().get("evaluations", [])
         response.raise_for_status()
         return None if not evaluations else evaluations[0]
 

langtrace_python_sdk-2.1.2/src/langtrace_python_sdk/version.py

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

langtrace_python_sdk-2.1.0/src/langtrace_python_sdk/version.py

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