langtrace-python-sdk 2.1.2__py3-none-any.whl → 2.1.5__py3-none-any.whl

examples/pinecone_example/basic.py

@@ -29,7 +29,7 @@ def create_index():
 
 
 @with_langtrace_root_span()
-def basic(span_id, trace_id):
+def basic(span_id=None, trace_id=None):
 
     result = client.embeddings.create(
         model="text-embedding-ada-002",
@@ -44,7 +44,6 @@ def basic(span_id, trace_id):
 
     index = pinecone.Index(PINECONE_INDEX_NAME)
     res = index.upsert(vectors=[data_to_upsert], namespace="test-namespace")
-    print("res", res)
 
     resp = index.query(
         vector=embedding, top_k=1, include_values=False, namespace="test-namespace"
@@ -52,8 +51,7 @@ def basic(span_id, trace_id):
     send_user_feedback(
         {"spanId": span_id, "traceId": trace_id, "userScore": 1, "userId": "123"}
     )
-    print(resp)
+    return [res, resp]
 
 
 # create_index()
-basic("span_id", "trace_id")

examples/weaviate_example/query_text.py

@@ -19,8 +19,8 @@ import langtrace_python_sdk.langtrace as langtrace
 from langtrace_python_sdk import with_langtrace_root_span
 
 # Set these environment variables
-WCS_DEMO_URL = "WEVIATE CLUSTER URL"
-WCS_DEMO_RO_KEY = "<WEAVIATE API KEY>"
+WCS_DEMO_URL = os.environ["WCS_DEMO_URL"]
+WCS_DEMO_RO_KEY = os.environ["WCS_DEMO_RO_KEY"]
 
 # Connect to a WCS instance
 client = weaviate.connect_to_wcs(
@@ -30,7 +30,7 @@ client = weaviate.connect_to_wcs(
     headers={"X-OpenAI-Api-Key": os.environ["OPENAI_APIKEY"]},
 )
 
-langtrace.init(api_key="asdf", write_spans_to_console=True, batch=False)
+langtrace.init()
 
 
 @with_langtrace_root_span()

langtrace_python_sdk/extensions/langtrace_exporter.py

@@ -9,6 +9,8 @@ from opentelemetry.trace.span import format_trace_id
 from langtrace_python_sdk.constants.exporter.langtrace_exporter import (
     LANGTRACE_REMOTE_URL,
 )
+from colorama import Fore
+from requests.exceptions import RequestException
 
 
 class LangTraceExporter(SpanExporter):
@@ -55,9 +57,6 @@ class LangTraceExporter(SpanExporter):
         self.api_key = api_key or os.environ.get("LANGTRACE_API_KEY")
         self.api_host: str = api_host or LANGTRACE_REMOTE_URL
 
-        if not self.api_key:
-            raise ValueError("No API key provided")
-
     def export(self, spans: typing.Sequence[ReadableSpan]) -> SpanExportResult:
         """
         Exports a batch of telemetry data.
@@ -68,10 +67,17 @@ class LangTraceExporter(SpanExporter):
         Returns:
             The result of the export SUCCESS or FAILURE
         """
+        if not self.api_key:
+            print(Fore.RED)
+            print(
+                "Missing Langtrace API key, proceed to https://langtrace.ai to create one"
+            )
+            print("Set the API key as an environment variable LANGTRACE_API_KEY")
+            print(Fore.RESET)
+            return
         data = [
             {
                 "traceId": format_trace_id(span.get_span_context().trace_id),
-                "instrumentationLibrary": span.instrumentation_info.__repr__(),
                 "droppedEventsCount": span.dropped_events,
                 "droppedAttributesCount": span.dropped_attributes,
                 "droppedLinksCount": span.dropped_links,
@@ -83,14 +89,23 @@ class LangTraceExporter(SpanExporter):
 
         # Send data to remote URL
         try:
-            requests.post(
+            response = requests.post(
                 url=f"{self.api_host}/api/trace",
                 data=json.dumps(data),
                 headers={"Content-Type": "application/json", "x-api-key": self.api_key},
+                timeout=20,
+            )
+
+            if not response.ok:
+                raise RequestException(response.text)
+
+            print(
+                Fore.GREEN + f"Exported {len(spans)} spans successfully." + Fore.RESET
             )
-            print(f"sent to {self.api_host}/api/trace with {len(data)} spans")
             return SpanExportResult.SUCCESS
-        except Exception as e:
+        except RequestException as err:
+            print(Fore.RED + "Failed to export spans.")
+            print(Fore.RED + f"Error: {err}" + Fore.RESET)
             return SpanExportResult.FAILURE
 
     def shutdown(self) -> None:

langtrace_python_sdk/instrumentation/__init__.py

@@ -0,0 +1,29 @@
+from .anthropic import AnthropicInstrumentation
+from .chroma import ChromaInstrumentation
+from .cohere import CohereInstrumentation
+from .groq import GroqInstrumentation
+from .langchain import LangchainInstrumentation
+from .langchain_community import LangchainCommunityInstrumentation
+from .langchain_core import LangchainCoreInstrumentation
+from .langgraph import LanggraphInstrumentation
+from .llamaindex import LlamaindexInstrumentation
+from .openai import OpenAIInstrumentation
+from .pinecone import PineconeInstrumentation
+from .qdrant import QdrantInstrumentation
+from .weaviate import WeaviateInstrumentation
+
+__all__ = [
+    "AnthropicInstrumentation",
+    "ChromaInstrumentation",
+    "CohereInstrumentation",
+    "GroqInstrumentation",
+    "LangchainInstrumentation",
+    "LangchainCommunityInstrumentation",
+    "LangchainCoreInstrumentation",
+    "LanggraphInstrumentation",
+    "LlamaindexInstrumentation",
+    "OpenAIInstrumentation",
+    "PineconeInstrumentation",
+    "QdrantInstrumentation",
+    "WeaviateInstrumentation",
+]

langtrace_python_sdk/instrumentation/anthropic/__init__.py

@@ -0,0 +1,5 @@
+from .instrumentation import AnthropicInstrumentation
+
+__all__ = [
+    "AnthropicInstrumentation",
+]

langtrace_python_sdk/instrumentation/chroma/__init__.py

@@ -0,0 +1,5 @@
+from .instrumentation import ChromaInstrumentation
+
+__all__ = [
+    "ChromaInstrumentation",
+]

langtrace_python_sdk/instrumentation/cohere/__init__.py

@@ -0,0 +1,5 @@
+from .instrumentation import CohereInstrumentation
+
+__all__ = [
+    "CohereInstrumentation",
+]

langtrace_python_sdk/instrumentation/groq/__init__.py

@@ -0,0 +1,5 @@
+from .instrumentation import GroqInstrumentation
+
+__all__ = [
+    "GroqInstrumentation",
+]

langtrace_python_sdk/instrumentation/langchain/__init__.py

@@ -0,0 +1,5 @@
+from .instrumentation import LangchainInstrumentation
+
+__all__ = [
+    "LangchainInstrumentation",
+]

langtrace_python_sdk/instrumentation/langchain_community/__init__.py

@@ -0,0 +1,5 @@
+from .instrumentation import LangchainCommunityInstrumentation
+
+__all__ = [
+    "LangchainCommunityInstrumentation",
+]

langtrace_python_sdk/instrumentation/langchain_core/__init__.py

@@ -0,0 +1,6 @@
+from .instrumentation import LangchainCoreInstrumentation
+
+
+__all__ = [
+    "LangchainCoreInstrumentation",
+]

langtrace_python_sdk/instrumentation/langgraph/__init__.py

@@ -0,0 +1,5 @@
+from .instrumentation import LanggraphInstrumentation
+
+__all__ = [
+    "LanggraphInstrumentation",
+]

langtrace_python_sdk/instrumentation/llamaindex/__init__.py

@@ -0,0 +1,5 @@
+from .instrumentation import LlamaindexInstrumentation
+
+__all__ = [
+    "LlamaindexInstrumentation",
+]

langtrace_python_sdk/instrumentation/openai/__init__.py

@@ -0,0 +1,5 @@
+from .instrumentation import OpenAIInstrumentation
+
+__all__ = [
+    "OpenAIInstrumentation",
+]

langtrace_python_sdk/instrumentation/pinecone/__init__.py

@@ -0,0 +1,5 @@
+from .instrumentation import PineconeInstrumentation
+
+__all__ = [
+    "PineconeInstrumentation",
+]

langtrace_python_sdk/instrumentation/qdrant/__init__.py

@@ -0,0 +1,5 @@
+from .instrumentation import QdrantInstrumentation
+
+__all__ = [
+    "QdrantInstrumentation",
+]

langtrace_python_sdk/instrumentation/weaviate/__init__.py

@@ -0,0 +1,5 @@
+from .instrumentation import WeaviateInstrumentation
+
+__all__ = [
+    "WeaviateInstrumentation",
+]

langtrace_python_sdk/instrumentation/weaviate/instrumentation.py

@@ -17,7 +17,7 @@ limitations under the License.
 import importlib.metadata
 import logging
 from typing import Collection
-
+from opentelemetry.instrumentation.utils import unwrap
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 from opentelemetry.trace import get_tracer
 from wrapt import wrap_function_wrapper
@@ -28,7 +28,7 @@ from langtrace_python_sdk.instrumentation.weaviate.patch import (
 )
 from langtrace_python_sdk.constants.instrumentation.weaviate import APIS
 
-logging.basicConfig(level=logging.FATAL)
+logging.basicConfig(level=logging.DEBUG)  # Set to DEBUG for detailed logging
 
 
 class WeaviateInstrumentation(BaseInstrumentor):
@@ -47,17 +47,19 @@ class WeaviateInstrumentation(BaseInstrumentor):
         for api_name, api_config in APIS.items():
             module_path, function_name = api_name.rsplit(".", 1)
             if api_config.get("OPERATION") == "query":
-                wrap_function_wrapper(
-                    api_config["MODULE"],
-                    api_config["METHOD"],
-                    generic_query_patch(api_name, version, tracer),
-                )
+                if getattr(api_config["MODULE"], api_config["METHOD"], None):
+                    wrap_function_wrapper(
+                        api_config["MODULE"],
+                        api_config["METHOD"],
+                        generic_query_patch(api_name, version, tracer),
+                    )
             elif api_config.get("OPERATION") == "create":
-                wrap_function_wrapper(
-                    api_config["MODULE"],
-                    api_config["METHOD"],
-                    generic_collection_patch(api_name, version, tracer),
-                )
+                if getattr(api_config["MODULE"], api_config["METHOD"], None):
+                    wrap_function_wrapper(
+                        api_config["MODULE"],
+                        api_config["METHOD"],
+                        generic_collection_patch(api_name, version, tracer),
+                    )
 
     def _instrument_module(self, module_name):
         pass

langtrace_python_sdk/langtrace.py

@@ -28,45 +28,23 @@ import sys
 from opentelemetry.sdk.resources import Resource
 
 from langtrace_python_sdk.extensions.langtrace_exporter import LangTraceExporter
-from langtrace_python_sdk.instrumentation.anthropic.instrumentation import (
+from langtrace_python_sdk.instrumentation import (
     AnthropicInstrumentation,
-)
-from langtrace_python_sdk.instrumentation.chroma.instrumentation import (
     ChromaInstrumentation,
-)
-from langtrace_python_sdk.instrumentation.cohere.instrumentation import (
     CohereInstrumentation,
-)
-from langtrace_python_sdk.instrumentation.groq.instrumentation import (
     GroqInstrumentation,
-)
-from langtrace_python_sdk.instrumentation.langchain.instrumentation import (
     LangchainInstrumentation,
-)
-from langtrace_python_sdk.instrumentation.langchain_community.instrumentation import (
     LangchainCommunityInstrumentation,
-)
-from langtrace_python_sdk.instrumentation.langchain_core.instrumentation import (
     LangchainCoreInstrumentation,
-)
-from langtrace_python_sdk.instrumentation.langgraph.instrumentation import (
     LanggraphInstrumentation,
-)
-from langtrace_python_sdk.instrumentation.llamaindex.instrumentation import (
     LlamaindexInstrumentation,
-)
-from langtrace_python_sdk.instrumentation.openai.instrumentation import (
     OpenAIInstrumentation,
-)
-from langtrace_python_sdk.instrumentation.pinecone.instrumentation import (
     PineconeInstrumentation,
-)
-from langtrace_python_sdk.instrumentation.qdrant.instrumentation import (
     QdrantInstrumentation,
-)
-from langtrace_python_sdk.instrumentation.weaviate.instrumentation import (
     WeaviateInstrumentation,
 )
+from opentelemetry.instrumentation.sqlalchemy import SQLAlchemyInstrumentor
+from colorama import Fore
 
 
 def init(
@@ -77,6 +55,7 @@ def init(
     api_host: Optional[str] = None,
     disable_instrumentations: Optional[DisableInstrumentations] = None,
 ):
+    print(Fore.GREEN + "Initializing Langtrace SDK.." + Fore.RESET)
     provider = TracerProvider(resource=Resource.create({"service.name": sys.argv[0]}))
 
     remote_write_exporter = (
@@ -90,20 +69,24 @@ def init(
     simple_processor_console = SimpleSpanProcessor(console_exporter)
 
     if write_spans_to_console:
+        print(Fore.BLUE + "Writing spans to console" + Fore.RESET)
         provider.add_span_processor(simple_processor_console)
 
     elif custom_remote_exporter is not None:
+        print(Fore.BLUE + "Exporting spans to custom remote exporter.." + Fore.RESET)
         if batch:
             provider.add_span_processor(batch_processor_remote)
         else:
             provider.add_span_processor(simple_processor_remote)
 
     elif api_host is not None:
+        print(Fore.BLUE + f"Exporting spans to custom host: {api_host}.." + Fore.RESET)
         if batch:
             provider.add_span_processor(batch_processor_remote)
         else:
             provider.add_span_processor(simple_processor_remote)
     else:
+        print(Fore.BLUE + "Exporting spans to Langtrace cloud.." + Fore.RESET)
         provider.add_span_processor(batch_processor_remote)
 
     # Initialize tracer
@@ -123,6 +106,7 @@ def init(
         "anthropic": AnthropicInstrumentation(),
         "cohere": CohereInstrumentation(),
         "weaviate": WeaviateInstrumentation(),
+        "sqlalchemy": SQLAlchemyInstrumentor(),
     }
 
     init_instrumentations(disable_instrumentations, all_instrumentations)

langtrace_python_sdk/utils/with_root_span.py

@@ -126,7 +126,6 @@ def send_user_feedback(data: EvaluationAPIData) -> None:
                 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
@@ -137,7 +136,6 @@ def send_user_feedback(data: EvaluationAPIData) -> None:
                 headers=headers,
                 timeout=None,
             )
-            print("Response", response.status_code, response.text)
             response.raise_for_status()
 
     except requests.RequestException as err:

langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "2.1.2"
+__version__ = "2.1.5"

{langtrace_python_sdk-2.1.2.dist-info → langtrace_python_sdk-2.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 2.1.2
+Version: 2.1.5
 Summary: Python SDK for LangTrace
 Project-URL: Homepage, https://github.com/Scale3-Labs/langtrace-python-sdk
 Author-email: Scale3 Labs <engineering@scale3labs.com>
@@ -10,9 +10,11 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.9
+Requires-Dist: colorama>=0.4.6
 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-sqlalchemy>=0.45b0
 Requires-Dist: opentelemetry-instrumentation>=0.45b0
 Requires-Dist: opentelemetry-sdk>=1.24.0
 Requires-Dist: tiktoken>=0.1.1

{langtrace_python_sdk-2.1.2.dist-info → langtrace_python_sdk-2.1.5.dist-info}/RECORD

@@ -32,13 +32,13 @@ 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=9iG5Tk8QExe4cFFI6_mt1pgd5BLCZh64tsU1oPBFXTg,1464
+examples/pinecone_example/basic.py,sha256=b-5-cSf3_N6Gk4zGVecGF47mBcFY9ZrC5ztrIZVgTQo,1430
 examples/qdrant_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 examples/qdrant_example/basic.py,sha256=DCMjHSuBZKkhEjCkwy5d5La9WMyW0lCWqtcZWiFCEm4,1425
-examples/weaviate_example/query_text.py,sha256=s3f0hYYi-xdbh_LlTBDHegQiVi-4bW_eACb2EnEj8J4,64533
+examples/weaviate_example/query_text.py,sha256=yhBCmJ61llKL5T92s1_R7yihS6koys2AJQWZXSYIT9w,64491
 langtrace_python_sdk/__init__.py,sha256=R4dv73QARUw4WRfCdSKkKzZTLDumF9jcSdwtVrzWhb4,962
-langtrace_python_sdk/langtrace.py,sha256=2VVVeWX3b20jwDY9iqDTxeE-Zunfgt2gfgRs3SreZLU,6993
-langtrace_python_sdk/version.py,sha256=UiuBcRXPtXxPUBDdp0ZDvWl0U9Db1kMNfT3oAfhxqLg,22
+langtrace_python_sdk/langtrace.py,sha256=h168V4kyqSezozDnMVar2Xgbi764WvVA_PpSgVuBzz0,6560
+langtrace_python_sdk/version.py,sha256=Ol8KeLnnX1kAXAFgJsZj4d_cZMypHQtaiyICJpuzp64,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
@@ -52,46 +52,46 @@ langtrace_python_sdk/constants/instrumentation/pinecone.py,sha256=Xaqqw-xBO0JJLG
 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
-langtrace_python_sdk/instrumentation/anthropic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+langtrace_python_sdk/extensions/langtrace_exporter.py,sha256=Fj--DzBr4AwTM_iS3lRz0NhqESEyTOZ3SBCA4OevFkE,4127
+langtrace_python_sdk/instrumentation/__init__.py,sha256=htP583cfv32IUvWeck6edRiPQxhk0uzUa1l1vgbtjtY,1042
+langtrace_python_sdk/instrumentation/anthropic/__init__.py,sha256=donrurJAGYlxrSRA3BIf76jGeUcAx9Tq8CVpah68S0Y,101
 langtrace_python_sdk/instrumentation/anthropic/instrumentation.py,sha256=-srgE8qumAn0ulQYZxMa8ch-9IBH0XgBW_rfEnGk6LI,1684
 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/__init__.py,sha256=pNZ5UO8Q-d5VkXSobBf79reB6AmEl_usnnTp5Itv818,95
 langtrace_python_sdk/instrumentation/chroma/instrumentation.py,sha256=nT6PS6bsrIOO9kLV5GuUeRjMe6THHHAZGvqWBP1dYog,1807
 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/__init__.py,sha256=sGUSLdTUyYf36Tm6L5jQflhzCqvmWrhnBOMYHjvp6Hs,95
 langtrace_python_sdk/instrumentation/cohere/instrumentation.py,sha256=YQFHZIBd7SSPD4b6Va-ZR0thf_AuBCqj5yzHLHJVWnM,2121
 langtrace_python_sdk/instrumentation/cohere/patch.py,sha256=hdSWArP0cdfGs0Q5KhkaowkwrYQIzxvF8NpY4sg8p2Q,26563
-langtrace_python_sdk/instrumentation/groq/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+langtrace_python_sdk/instrumentation/groq/__init__.py,sha256=ZXeq_nrej6Lm_uoMFEg8wbSejhjB2UJ5IoHQBPc2-C0,91
 langtrace_python_sdk/instrumentation/groq/instrumentation.py,sha256=Ttf07XVKhdYY1_fqJc7QWiSdmgEhEVyQB_3Az2_wqYo,1832
 langtrace_python_sdk/instrumentation/groq/patch.py,sha256=Fo5QbNwScFwhvmXZQezbjF0AdajsHQ8a0f8nKsVFjF8,26068
-langtrace_python_sdk/instrumentation/langchain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+langtrace_python_sdk/instrumentation/langchain/__init__.py,sha256=-7ZkqQFu64F-cxSFd1ZPrciODKqmUIyUbQQ-eHuQPyM,101
 langtrace_python_sdk/instrumentation/langchain/instrumentation.py,sha256=_Z4AeNb2hBPSCvMRxE-mUfmkUO_wP_tGGtu-jppWPiI,3462
 langtrace_python_sdk/instrumentation/langchain/patch.py,sha256=2BT_-DjfSPTmyHDqnpj-gVrMFxRSZugVC-YRPXnuupU,3796
-langtrace_python_sdk/instrumentation/langchain_community/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+langtrace_python_sdk/instrumentation/langchain_community/__init__.py,sha256=mj5RR_cfkjMql7W9OyyhmviT2GZ-4Pv9XJfGwJufp_E,119
 langtrace_python_sdk/instrumentation/langchain_community/instrumentation.py,sha256=-9E2-8-XR5PQ5ON-_ib53p5Tp0uD4MT4Ioh4MlWlF2k,5192
 langtrace_python_sdk/instrumentation/langchain_community/patch.py,sha256=uyRUp02M9mYjLsr36ZBCvJC7s40AGry3R_EcjvXaoWk,3751
-langtrace_python_sdk/instrumentation/langchain_core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+langtrace_python_sdk/instrumentation/langchain_core/__init__.py,sha256=kumE_reeqgM-ZvEZ6-XxyT-F-HAdKq_v_PKvsLb4EZQ,110
 langtrace_python_sdk/instrumentation/langchain_core/instrumentation.py,sha256=KNXHtlnq80akLACF-vbPCo9oVXhnVjYIM1Lo7kNRwEg,5937
 langtrace_python_sdk/instrumentation/langchain_core/patch.py,sha256=Kqpv77ZKrZ4QMit36JtN59ubtWE8pX6t8gg6OOmIii0,8458
-langtrace_python_sdk/instrumentation/langgraph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+langtrace_python_sdk/instrumentation/langgraph/__init__.py,sha256=eitlHloY-aZ4ZuIEJx61AadEA3G7siyecP-V-lziAr8,101
 langtrace_python_sdk/instrumentation/langgraph/instrumentation.py,sha256=SUZZhWSIbcfsF1S5NtEqW8QzkRM_pKAuXB7pwk5tsOU,2526
 langtrace_python_sdk/instrumentation/langgraph/patch.py,sha256=fr90kuPokwxP2BGP2tU793ReDOp2FtehKMcsxS9tS_0,4589
-langtrace_python_sdk/instrumentation/llamaindex/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+langtrace_python_sdk/instrumentation/llamaindex/__init__.py,sha256=rHvuqpuQKLj57Ow7vuKRqxAN5jT0b5NBeHwhXbbnRa4,103
 langtrace_python_sdk/instrumentation/llamaindex/instrumentation.py,sha256=8iAg-Oxwf2W4S60qRfO5mvzORYxublgq7FdGWqUB4q8,2965
 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/__init__.py,sha256=VPHRNCQEdkizIVP2d0Uw_a7t8XOTSTprEIB8oboJFbs,95
 langtrace_python_sdk/instrumentation/openai/instrumentation.py,sha256=G2HSZfr6DuP2n-8v0h91M60m0DJWFBcru4-1FQJl-5A,2726
 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/__init__.py,sha256=DzXyGh9_MGWveJvXULkFwdkf7PbG2s3bAWtT1Dmz7Ok,99
 langtrace_python_sdk/instrumentation/pinecone/instrumentation.py,sha256=mxQXe3oAOPLsMJGlEzAe6zpgK7RtWfqmcNmGW_gQXX4,1900
 langtrace_python_sdk/instrumentation/pinecone/patch.py,sha256=8uenVh4TCTJVuBGepm7nhTZBuW7HyjlZS4wPx7zi9cg,4879
-langtrace_python_sdk/instrumentation/qdrant/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+langtrace_python_sdk/instrumentation/qdrant/__init__.py,sha256=TaIGSAEPysrL23KJ5FcEL1tfQogrKCtEQ75_u62eqso,95
 langtrace_python_sdk/instrumentation/qdrant/instrumentation.py,sha256=vl2eKSP55aqDo1JiRlvOUBrr6kddvG9Z5dCYew2OG08,1816
 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/__init__.py,sha256=Mc-Je6evPo-kKQzerTG7bd1XO5JOh4YGTE3wBxaUBwg,99
+langtrace_python_sdk/instrumentation/weaviate/instrumentation.py,sha256=4BnHRzXOKEauy2TNTZwpIagDFUp03CUyBZCdE2vinxk,2596
 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
@@ -100,7 +100,7 @@ langtrace_python_sdk/utils/misc.py,sha256=CD9NWRLxLpFd0YwlHJqzlpFNedXVWtAKGOjQWn
 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=l-N6o7cnWUyrD6dBvW7W3Pf5CkPo5QaoT__k1XLbrQg,383
-langtrace_python_sdk/utils/with_root_span.py,sha256=H20e-LeRFHHSmdXlZhkWas33ImgEKihlVqAtoYi4iNA,6103
+langtrace_python_sdk/utils/with_root_span.py,sha256=rSPrTnKndJo6cFUWL4907IkJFbtLwqMS87paADUxD6A,5957
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/conftest.py,sha256=0Jo6iCZTXbdvyJVhG9UpYGkLabL75378oauCzmt-Sa8,603
 tests/utils.py,sha256=hP8sTH-M8WO6zlLkSFHPf6483ZcKEcnxul6JIIb1pLM,1396
@@ -119,9 +119,9 @@ tests/cohere/cassettes/test_cohere_chat.yaml,sha256=iQIhJwmWe2AQDmdguL6L0SZOPMD6
 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
+tests/langchain/conftest.py,sha256=f29apdevxg7AM0mPQ1LoEd-yStGruqGLTQUp29heLJo,1116
+tests/langchain/test_langchain.py,sha256=QP11RdNDX_ztF1ppPnUiui3SsxYYdzVsDgtI-1OGH48,893
+tests/langchain/cassettes/test_langchain.yaml,sha256=KPBTVIYMUPFaSNpwrTDgWzsu4p3hHj_yNDoudDa-Jis,3755
 tests/openai/conftest.py,sha256=BkehS6heg-O91Nzoc4546OSiAzy8KgSgk7VCO3A11zM,700
 tests/openai/test_chat_completion.py,sha256=-TwCAF2czsiGZqETQvMOgmIJgdFWycJ9nOeO5q65AJs,5804
 tests/openai/test_embeddings.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -137,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.1.2.dist-info/METADATA,sha256=4kaLgCo0f7qBqUWa7gRKBEj-_Sl57Iw-MuN1lqCTwAY,11738
-langtrace_python_sdk-2.1.2.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
-langtrace_python_sdk-2.1.2.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-langtrace_python_sdk-2.1.2.dist-info/RECORD,,
+langtrace_python_sdk-2.1.5.dist-info/METADATA,sha256=YMAZgu-2MfYrXYBr3CWOi7FMHV2OOmcyuKknp4Za7Nc,11833
+langtrace_python_sdk-2.1.5.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
+langtrace_python_sdk-2.1.5.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+langtrace_python_sdk-2.1.5.dist-info/RECORD,,

tests/langchain/cassettes/test_langchain.yaml

@@ -0,0 +1,106 @@
+interactions:
+- request:
+    body: '{"messages": [{"content": "You are world class technical documentation
+      writer.", "role": "system"}, {"content": "how can langsmith help with testing?",
+      "role": "user"}], "model": "gpt-3.5-turbo", "n": 1, "stream": false, "temperature":
+      0.7}'
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '240'
+      content-type:
+      - application/json
+      host:
+      - api.openai.com
+      user-agent:
+      - OpenAI/Python 1.30.5
+      x-stainless-arch:
+      - arm64
+      x-stainless-async:
+      - 'false'
+      x-stainless-lang:
+      - python
+      x-stainless-os:
+      - MacOS
+      x-stainless-package-version:
+      - 1.30.5
+      x-stainless-runtime:
+      - CPython
+      x-stainless-runtime-version:
+      - 3.12.3
+    method: POST
+    uri: https://api.openai.com/v1/chat/completions
+  response:
+    body:
+      string: !!binary |
+        H4sIAAAAAAAAA3xVTW8bOQy9+1cQc7aNfLR16tv2A4tFt9hL9rRdBLSGnlGikVSSsuMt8t8X1Iwd
+        pwh6MWxRfHp8fKR/zAAa3zZraFyP6oYcFu//br/s3+271XUeVvfb+0f6cvv4x8MHyjf4tZlbRtrc
+        k9Nj1tKlIQdSn+IYdkyoZKiXq8vVxc3NxeqmBobUUrC0Luvievl2oYU3aXFxefV2yuyTdyTNGv6Z
+        AQD8qJ/GMbb02KzhYn48GUgEO2rWp0sADadgJw2KeFGM2syfgy5FpVhp/4mxk8FrD14AIac98bYE
+        0JQCaI8KDiP0FDLs7ZaSqI8d+Ag7ZJ+KwB4Psv4Wv8XLJfxWNA1WMtyOF9fw/IIhbQiKUAuaYM9e
+        CfCUYdAgjn1WgW1iOKTCIGmre2QCzDl4hyauLOHDAQLtiLGrbFQgohbGAAFjV7AjyJwciVjcYcaN
+        D149ydxwK5eRQH3WoZBYUTmgj/A5dsFLP4cBH0Z8IBRPXHnFFBdKro/eYQAlHGCgYUMsVpaJy35T
+        DDqB9nTSbCK0NK2ullUh+ISK8DtF4lrZz3J1Y2Ri2drlDZp+KUJmamnrI7XAJZAAxhYyqhJHmdee
+        2atW7ffi3UM4wOjHZzChc6lfKnzbezm2LCOrdyUgh4P1zxxiacfKxFE0N8hoGabvxVvLICB3BDik
+        EhXSFlq/Ixaqj1cdricdPqIQfMWIHQ0U9Re2Sdxh9P9RLXeoGWc9fKVl1gVG91B7kTl1TCJG5sge
+        nfpd9UbFpCiFCWyWmXqK4nf2q5qNLO8XvrSS3izhU3LFyni1pxgknU1UO901Jud11O9MUoLKvBJL
+        2tNJ8wVTqGPj4zbxUF+qYzENlHkzHI4GMvD2nJR5HUGUi9PC1MII8jwcJxWiLRCK7lA5uIDs9WDp
+        VYajhi/Aqwx/mWAhzH+qXpQJh+AjvTYbc/BD5rQ7tnRSfT41psfojFMIuEnjxJi5qnJnU+jjLoVd
+        1eaVJi+baRM+nVZoSF3mtLF1G0sIp/Otj176OyaUFG1diqY8pj/NAP6tq7q82L5N5jRkvdP0QNEA
+        r1YjXPP853AWvHw/RTUphrPAm3eziWEjB1Ea7rY+dsSZfd3cxnP2NPsfAAD//wMAt4ZeGbgGAAA=
+    headers:
+      CF-Cache-Status:
+      - DYNAMIC
+      CF-RAY:
+      - 88c03f46bf3b23c5-LHR
+      Connection:
+      - keep-alive
+      Content-Encoding:
+      - gzip
+      Content-Type:
+      - application/json
+      Date:
+      - Thu, 30 May 2024 16:54:42 GMT
+      Server:
+      - cloudflare
+      Set-Cookie:
+      - __cf_bm=9en.DN2CPQUgM8CubXIqqiVZkm43q.oBCL4n7bXxViw-1717088082-1.0.1.1-nl9dbPbwwGsUORjX1Y8Ta7S2v.MBG3o1JWUgYQU98tFUNcY9hFcmfcXWFu_bcohUX1nun6S55Wg2rg3MX7PHPQ;
+        path=/; expires=Thu, 30-May-24 17:24:42 GMT; domain=.api.openai.com; HttpOnly;
+        Secure; SameSite=None
+      - _cfuvid=bJ_po6.pyb_9JWu27O8yfh1AX1atGjC5_sFN497eMI8-1717088082420-0.0.1.1-604800000;
+        path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
+      Transfer-Encoding:
+      - chunked
+      alt-svc:
+      - h3=":443"; ma=86400
+      openai-organization:
+      - scale3-1
+      openai-processing-ms:
+      - '4295'
+      openai-version:
+      - '2020-10-01'
+      strict-transport-security:
+      - max-age=15724800; includeSubDomains
+      x-ratelimit-limit-requests:
+      - '10000'
+      x-ratelimit-limit-tokens:
+      - '2000000'
+      x-ratelimit-remaining-requests:
+      - '9999'
+      x-ratelimit-remaining-tokens:
+      - '1999960'
+      x-ratelimit-reset-requests:
+      - 6ms
+      x-ratelimit-reset-tokens:
+      - 1ms
+      x-request-id:
+      - req_4b43c2d3cf0dc839647c8b775b75d6f5
+    status:
+      code: 200
+      message: OK
+version: 1

tests/langchain/conftest.py

@@ -0,0 +1,42 @@
+"""Unit tests configuration module."""
+
+import os
+import pytest
+
+from langtrace_python_sdk.instrumentation.openai.instrumentation import (
+    OpenAIInstrumentation,
+)
+from langtrace_python_sdk.instrumentation.langchain.instrumentation import (
+    LangchainInstrumentation,
+)
+from langtrace_python_sdk.instrumentation.langchain_core.instrumentation import (
+    LangchainCoreInstrumentation,
+)
+
+from langtrace_python_sdk.instrumentation.langchain_community.instrumentation import (
+    LangchainCommunityInstrumentation,
+)
+
+
+@pytest.fixture(autouse=True)
+def clear_exporter(exporter):
+    exporter.clear()
+
+
+@pytest.fixture(autouse=True)
+def environment():
+    if not os.environ.get("OPENAI_API_KEY"):
+        os.environ["OPENAI_API_KEY"] = "test"
+
+
+@pytest.fixture(scope="module")
+def vcr_config():
+    return {"filter_headers": ["authorization", "x-api-key"]}
+
+
+@pytest.fixture(scope="session", autouse=True)
+def instrument():
+    OpenAIInstrumentation().instrument()
+    LangchainInstrumentation().instrument()
+    LangchainCoreInstrumentation().instrument()
+    LangchainCommunityInstrumentation().instrument()

tests/langchain/test_langchain.py

@@ -1,82 +1,28 @@
-import unittest
-from unittest.mock import MagicMock, call
-from langtrace_python_sdk.instrumentation.langchain.patch import generic_patch
-from opentelemetry.trace import SpanKind
-from opentelemetry.trace import get_tracer
-import importlib.metadata
-from langtrace_python_sdk.constants.instrumentation.openai import APIS
-from opentelemetry.trace.status import Status, StatusCode
-from tests.utils import common_setup
-import json
-
-
-class TestGenericPatch(unittest.TestCase):
-    data = {"key": "value"}
-
-    def setUp(self):
-        self.langchain_mock, self.tracer, self.span = common_setup(self.data, None)
-
-    def tearDown(self):
-        # Clean up after each test case
-        pass
-
-    def test_generic_patch(self):
-        # Arrange
-        method_name = "example_method"
-        trace_output = False
-        trace_input = False  # Change as per your requirement
-        args = (1, 2, 3)
-        task = "split_text"
-        kwargs = {"key": "value"}
-        version = importlib.metadata.version("langchain")
-
-        # Act
-        wrapped_function = generic_patch(
-            "langchain.text_splitter",
-            task,
-            self.tracer,
-            version,
-            trace_output,
-            trace_input,
-        )
-        result = wrapped_function(self.langchain_mock, MagicMock(), args, kwargs)
-
-        # Assert
-        self.assertTrue(
-            self.tracer.start_as_current_span.called_once_with(
-                method_name, kind=SpanKind.CLIENT
-            )
-        )
-
-        service_provider = "Langchain"
-        expected_attributes = {
-            "langtrace.sdk.name": "langtrace-python-sdk",
-            "langtrace.service.name": service_provider,
-            "langtrace.service.type": "framework",
-            "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
-            "langchain.task.name": task,
-        }
-
-        self.assertTrue(
-            self.span.set_attribute.has_calls(
-                [call(key, value) for key, value in expected_attributes.items()],
-                any_order=True,
-            )
-        )
-
-        actual_calls = self.span.set_attribute.call_args_list
-
-        for key, value in expected_attributes.items():
-            self.assertIn(call(key, value), actual_calls)
-
-        self.assertEqual(self.span.set_status.call_count, 1)
-        self.assertTrue(self.span.set_status.has_calls([call(Status(StatusCode.OK))]))
-
-        expected_result_data = {"key": "value"}
-
-        self.assertEqual(result.key, expected_result_data["key"])
-
-
-if __name__ == "__main__":
-    unittest.main()
+import pytest
+from langchain_openai import ChatOpenAI
+from langchain_core.prompts.chat import ChatPromptTemplate
+from langchain_core.output_parsers import StrOutputParser
+
+
+@pytest.mark.vcr()
+def test_langchain(exporter):
+    llm = ChatOpenAI()
+    prompt = ChatPromptTemplate.from_messages(
+        [
+            ("system", "You are world class technical documentation writer."),
+            ("user", "{input}"),
+        ]
+    )
+    output_parser = StrOutputParser()
+    chain = prompt | llm | output_parser
+    chain.invoke({"input": "how can langsmith help with testing?"})
+    spans = exporter.get_finished_spans()
+
+    assert [
+        "ChatPromptTemplate.invoke",
+        "openai.chat.completions.create",
+        "StrOutputParser.parse",
+        "StrOutputParser.parse_result",
+        "StrOutputParser.invoke",
+        "RunnableSequence.invoke",
+    ] == [span.name for span in spans]

tests/langchain/test_langchain_community.py

@@ -1,82 +0,0 @@
-import unittest
-from unittest.mock import MagicMock, Mock, patch, call
-from langtrace_python_sdk.instrumentation.langchain_community.patch import generic_patch
-from opentelemetry.trace import SpanKind
-from opentelemetry.trace import get_tracer
-import importlib.metadata
-import openai
-from langtrace_python_sdk.constants.instrumentation.openai import APIS
-from opentelemetry.trace.status import Status, StatusCode
-import json
-from tests.utils import common_setup
-
-
-class TestGenericPatch(unittest.TestCase):
-    data = {"key": "value"}
-
-    def setUp(self):
-        self.langchain_mock, self.tracer, self.span = common_setup(self.data, None)
-
-    def tearDown(self):
-        # Clean up after each test case
-        pass
-
-    def test_generic_patch(self):
-        # Arrange
-        method_name = "example_method"
-        trace_output = False
-        trace_input = False
-        args = (1, 2, 3)
-        task = "vector_store"
-        kwargs = {"key": "value"}
-        version = importlib.metadata.version("langchain-community")
-
-        # Act
-        wrapped_function = generic_patch(
-            "langchain_community.vectorstores.faiss",
-            task,
-            self.tracer,
-            version,
-            trace_output,
-            trace_input,
-        )
-        result = wrapped_function(self.langchain_mock, MagicMock(), args, kwargs)
-
-        # Assert
-        self.assertTrue(
-            self.tracer.start_as_current_span.called_once_with(
-                method_name, kind=SpanKind.CLIENT
-            )
-        )
-
-        service_provider = "Langchain Community"
-        expected_attributes = {
-            "langtrace.sdk.name": "langtrace-python-sdk",
-            "langtrace.service.name": service_provider,
-            "langtrace.service.type": "framework",
-            "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
-            "langchain.task.name": task,
-        }
-
-        self.assertTrue(
-            self.span.set_attribute.has_calls(
-                [call(key, value) for key, value in expected_attributes.items()],
-                any_order=True,
-            )
-        )
-
-        actual_calls = self.span.set_attribute.call_args_list
-
-        for key, value in expected_attributes.items():
-            self.assertIn(call(key, value), actual_calls)
-
-        self.assertEqual(self.span.set_status.call_count, 1)
-        self.assertTrue(self.span.set_status.has_calls([call(Status(StatusCode.OK))]))
-
-        expected_result_data = {"key": "value"}
-        self.assertEqual(result.key, expected_result_data["key"])
-
-
-if __name__ == "__main__":
-    unittest.main()

tests/langchain/test_langchain_core.py

@@ -1,141 +0,0 @@
-import unittest
-from unittest.mock import MagicMock, Mock, patch, call
-from langtrace_python_sdk.instrumentation.langchain_core.patch import (
-    generic_patch,
-    runnable_patch,
-)
-from opentelemetry.trace import SpanKind
-from opentelemetry.trace import get_tracer
-import importlib.metadata
-import openai
-from langtrace_python_sdk.constants.instrumentation.openai import APIS
-from opentelemetry.trace.status import Status, StatusCode
-import json
-from tests.utils import common_setup
-
-
-class TestGenericPatch(unittest.TestCase):
-    data = {"items": "value"}
-
-    def setUp(self):
-        self.langchain_mock, self.tracer, self.span = common_setup(self.data, None)
-
-    def tearDown(self):
-        # Clean up after each test case
-        pass
-
-    def test_generic_patch(self):
-        # Arrange
-        method_name = "example_method"
-        trace_output = False
-        trace_input = True
-        task = "retriever"
-        args = (1, 2, 3)
-        kwargs = {"key": "value"}
-        version = importlib.metadata.version("langchain-core")
-
-        # Act
-        wrapped_function = generic_patch(
-            "langchain_core.retrievers",
-            task,
-            self.tracer,
-            version,
-            trace_output,
-            trace_input,
-        )
-        result = wrapped_function(self.langchain_mock, MagicMock(), args, kwargs)
-
-        # Assert
-        self.assertTrue(
-            self.tracer.start_as_current_span.called_once_with(
-                method_name, kind=SpanKind.CLIENT
-            )
-        )
-
-        service_provider = "Langchain Core"
-        expected_attributes = {
-            "langtrace.sdk.name": "langtrace-python-sdk",
-            "langtrace.service.name": service_provider,
-            "langtrace.service.type": "framework",
-            "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
-            "langchain.task.name": task,
-        }
-
-        self.assertTrue(
-            self.span.set_attribute.has_calls(
-                [call(key, value) for key, value in expected_attributes.items()],
-                any_order=True,
-            )
-        )
-
-        actual_calls = self.span.set_attribute.call_args_list
-        for key, value in expected_attributes.items():
-            self.assertIn(call(key, value), actual_calls)
-
-        self.assertEqual(self.span.set_status.call_count, 1)
-        self.assertTrue(self.span.set_status.has_calls([call(Status(StatusCode.OK))]))
-
-        expected_result_data = {"items": "value"}
-        self.assertEqual(result.items, expected_result_data["items"])
-
-    def test_runnable_patch(self):
-        # Arrange
-        method_name = "example_method"
-        trace_output = False
-        trace_input = True
-        args = (1, 2, 3)
-        kwargs = {"key": "value"}
-        version = importlib.metadata.version("langchain-core")
-
-        # Act
-        wrapped_function = runnable_patch(
-            "langchain_core.runnables.passthrough",
-            "runnablepassthrough",
-            self.tracer,
-            version,
-            trace_output,
-            trace_input,
-        )
-
-        result = wrapped_function(self.langchain_mock, MagicMock(), args, kwargs)
-
-        # Assert
-        self.assertTrue(
-            self.tracer.start_as_current_span.called_once_with(
-                method_name, kind=SpanKind.CLIENT
-            )
-        )
-
-        service_provider = "Langchain Core"
-        expected_attributes = {
-            "langtrace.sdk.name": "langtrace-python-sdk",
-            "langtrace.service.name": service_provider,
-            "langtrace.service.type": "framework",
-            "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
-            "langchain.task.name": "runnablepassthrough",
-        }
-
-        self.assertTrue(
-            self.span.set_attribute.has_calls(
-                [call(key, value) for key, value in expected_attributes.items()],
-                any_order=True,
-            )
-        )
-
-        actual_calls = self.span.set_attribute.call_args_list
-
-        for key, value in expected_attributes.items():
-            self.assertIn(call(key, value), actual_calls)
-
-        self.assertEqual(self.span.set_status.call_count, 1)
-        self.assertTrue(self.span.set_status.has_calls([call(Status(StatusCode.OK))]))
-
-        expected_result_data = {"items": "value"}
-
-        self.assertEqual(result.items, expected_result_data["items"])
-
-
-if __name__ == "__main__":
-    unittest.main()