langtrace-python-sdk 2.3.9__tar.gz → 2.3.11__tar.gz

{langtrace_python_sdk-2.3.9 → langtrace_python_sdk-2.3.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 2.3.9
+Version: 2.3.11
 Summary: Python SDK for LangTrace
 Project-URL: Homepage, https://github.com/Scale3-Labs/langtrace-python-sdk
 Author-email: Scale3 Labs <engineering@scale3labs.com>
@@ -18,6 +18,7 @@ Requires-Dist: opentelemetry-exporter-otlp-proto-http>=1.25.0
 Requires-Dist: opentelemetry-instrumentation-sqlalchemy>=0.46b0
 Requires-Dist: opentelemetry-instrumentation>=0.47b0
 Requires-Dist: opentelemetry-sdk>=1.25.0
+Requires-Dist: sentry-sdk>=2.14.0
 Requires-Dist: sqlalchemy
 Requires-Dist: tiktoken>=0.1.1
 Requires-Dist: trace-attributes==7.0.4
@@ -196,6 +197,10 @@ langtrace.init(custom_remote_exporter=<your_exporter>, batch=<True or False>)
 | `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']}` |
 
+### Error Reporting to Langtrace
+
+By default all sdk errors are reported to langtrace via Sentry. This can be disabled by setting the following enviroment variable to `False` like so `LANGTRACE_ERROR_REPORTING=False`
+
 ### 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.
@@ -277,6 +282,7 @@ prompt = get_prompt_from_registry(<Registry ID>, options={"prompt_version": 1, "
 ```
 
 ### Opt out of tracing prompt and completion data
+
 By default, prompt and completion data are captured. If you would like to opt out of it, set the following env var,
 
 `TRACE_PROMPT_COMPLETION_DATA=false`

{langtrace_python_sdk-2.3.9 → langtrace_python_sdk-2.3.11}/README.md

@@ -148,6 +148,10 @@ langtrace.init(custom_remote_exporter=<your_exporter>, batch=<True or False>)
 | `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']}` |
 
+### Error Reporting to Langtrace
+
+By default all sdk errors are reported to langtrace via Sentry. This can be disabled by setting the following enviroment variable to `False` like so `LANGTRACE_ERROR_REPORTING=False`
+
 ### 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.
@@ -229,6 +233,7 @@ prompt = get_prompt_from_registry(<Registry ID>, options={"prompt_version": 1, "
 ```
 
 ### Opt out of tracing prompt and completion data
+
 By default, prompt and completion data are captured. If you would like to opt out of it, set the following env var,
 
 `TRACE_PROMPT_COMPLETION_DATA=false`

{langtrace_python_sdk-2.3.9 → langtrace_python_sdk-2.3.11}/pyproject.toml

@@ -30,6 +30,7 @@ dependencies = [
   'sqlalchemy',
   'fsspec>=2024.6.0',
   "transformers>=4.11.3",
+  "sentry-sdk>=2.14.0",
 ]
 
 requires-python = ">=3.9"

langtrace_python_sdk-2.3.11/src/langtrace_python_sdk/constants/__init__.py

@@ -0,0 +1,2 @@
+LANGTRACE_SDK_NAME = "langtrace-python-sdk"
+SENTRY_DSN = "https://7f8eed3a1237fb2efef0f5e96ab407af@o1419498.ingest.us.sentry.io/4507929133056000"

{langtrace_python_sdk-2.3.9 → langtrace_python_sdk-2.3.11}/src/langtrace_python_sdk/instrumentation/crewai/patch.py

@@ -51,7 +51,6 @@ def patch_memory(operation_name, version, tracer: Tracer):
                     )
                 if result:
                     span.set_status(Status(StatusCode.OK))
-                span.end()
                 return result
 
             except Exception as err:
@@ -102,7 +101,6 @@ def patch_crew(operation_name, version, tracer: Tracer):
                                 span.set_attribute(
                                     f"crewai.crew.{attr}", str(getattr(result, attr))
                                 )
-                span.end()
                 return result
 
             except Exception as err:
@@ -139,21 +137,27 @@ class CrewAISpanAttributes:
             for key, value in self.crew.items():
                 key = f"crewai.crew.{key}"
                 if value is not None:
-                    set_span_attribute(self.span, key, value)
+                    set_span_attribute(
+                        self.span, key, str(value) if isinstance(value, list) else value
+                    )
 
         elif instance_name == "Agent":
             agent = self.set_agent_attributes()
             for key, value in agent.items():
                 key = f"crewai.agent.{key}"
                 if value is not None:
-                    set_span_attribute(self.span, key, value)
+                    set_span_attribute(
+                        self.span, key, str(value) if isinstance(value, list) else value
+                    )
 
         elif instance_name == "Task":
             task = self.set_task_attributes()
             for key, value in task.items():
                 key = f"crewai.task.{key}"
                 if value is not None:
-                    set_span_attribute(self.span, key, value)
+                    set_span_attribute(
+                        self.span, key, str(value) if isinstance(value, list) else value
+                    )
 
     def set_crew_attributes(self):
         for key, value in self.instance.__dict__.items():

{langtrace_python_sdk-2.3.9 → langtrace_python_sdk-2.3.11}/src/langtrace_python_sdk/langtrace.py

@@ -17,8 +17,9 @@ limitations under the License.
 import os
 import sys
 from typing import Optional
-
+import importlib.util
 from colorama import Fore
+from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME, SENTRY_DSN
 from opentelemetry import trace
 from opentelemetry.instrumentation.sqlalchemy import SQLAlchemyInstrumentor
 from opentelemetry.sdk.resources import SERVICE_NAME, Resource
@@ -60,8 +61,9 @@ from langtrace_python_sdk.types import (
     InstrumentationMethods,
     InstrumentationType,
 )
-from langtrace_python_sdk.utils import check_if_sdk_is_outdated
+from langtrace_python_sdk.utils import check_if_sdk_is_outdated, get_sdk_version
 from langtrace_python_sdk.utils.langtrace_sampler import LangtraceSampler
+import sentry_sdk
 
 
 def init(
@@ -120,25 +122,25 @@ def init(
     all_instrumentations = {
         "openai": OpenAIInstrumentation(),
         "groq": GroqInstrumentation(),
-        "pinecone": PineconeInstrumentation(),
-        "llamaindex": LlamaindexInstrumentation(),
-        "chroma": ChromaInstrumentation(),
+        "pinecone-client": PineconeInstrumentation(),
+        "llama-index": LlamaindexInstrumentation(),
+        "chromadb": ChromaInstrumentation(),
         "embedchain": EmbedchainInstrumentation(),
-        "qdrant": QdrantInstrumentation(),
+        "qdrant-client": QdrantInstrumentation(),
         "langchain": LangchainInstrumentation(),
-        "langchain_core": LangchainCoreInstrumentation(),
-        "langchain_community": LangchainCommunityInstrumentation(),
+        "langchain-core": LangchainCoreInstrumentation(),
+        "langchain-community": LangchainCommunityInstrumentation(),
         "langgraph": LanggraphInstrumentation(),
         "anthropic": AnthropicInstrumentation(),
         "cohere": CohereInstrumentation(),
-        "weaviate": WeaviateInstrumentation(),
+        "weaviate-client": WeaviateInstrumentation(),
         "sqlalchemy": SQLAlchemyInstrumentor(),
         "ollama": OllamaInstrumentor(),
-        "dspy": DspyInstrumentation(),
+        "dspy-ai": DspyInstrumentation(),
         "crewai": CrewAIInstrumentation(),
-        "vertexai": VertexAIInstrumentation(),
-        "gemini": GeminiInstrumentation(),
-        "mistral": MistralInstrumentation(),
+        "google-cloud-aiplatform": VertexAIInstrumentation(),
+        "google-generativeai": GeminiInstrumentation(),
+        "mistralai": MistralInstrumentation(),
     }
 
     init_instrumentations(disable_instrumentations, all_instrumentations)
@@ -164,6 +166,27 @@ def init(
         provider.add_span_processor(batch_processor_remote)
 
     sys.stdout = sys.__stdout__
+    if os.environ.get("LANGTRACE_ERROR_REPORTING", "True") == "True":
+        sentry_sdk.init(
+            dsn=SENTRY_DSN,
+            traces_sample_rate=1.0,
+            profiles_sample_rate=1.0,
+        )
+        sdk_options = {
+            "service_name": os.environ.get("OTEL_SERVICE_NAME")
+            or service_name
+            or sys.argv[0],
+            "disable_logging": disable_logging,
+            "disable_instrumentations": disable_instrumentations,
+            "disable_tracing_for_functions": disable_tracing_for_functions,
+            "batch": batch,
+            "write_spans_to_console": write_spans_to_console,
+            "custom_remote_exporter": custom_remote_exporter,
+            "sdk_name": LANGTRACE_SDK_NAME,
+            "sdk_version": get_sdk_version(),
+            "api_host": host,
+        }
+        sentry_sdk.set_context("sdk_init_options", sdk_options)
 
 
 def init_instrumentations(
@@ -171,8 +194,9 @@ def init_instrumentations(
 ):
     if disable_instrumentations is None:
         for idx, (name, v) in enumerate(all_instrumentations.items()):
+            if is_package_installed(name):
+                v.instrument()
 
-            v.instrument()
     else:
 
         validate_instrumentations(disable_instrumentations)
@@ -229,3 +253,10 @@ def validate_instrumentations(disable_instrumentations):
             raise ValueError(
                 "Cannot specify both only and all_except in disable_instrumentations"
             )
+
+
+def is_package_installed(package_name):
+    import pkg_resources
+
+    installed_packages = {p.key for p in pkg_resources.working_set}
+    return package_name in installed_packages

{langtrace_python_sdk-2.3.9 → langtrace_python_sdk-2.3.11}/src/langtrace_python_sdk/utils/__init__.py

@@ -31,3 +31,7 @@ def set_event_prompt(span: Span, prompt):
 def check_if_sdk_is_outdated():
     SDKVersionChecker().check()
     return
+
+
+def get_sdk_version():
+    return SDKVersionChecker().get_sdk_version()

{langtrace_python_sdk-2.3.9 → langtrace_python_sdk-2.3.11}/src/langtrace_python_sdk/utils/sdk_version_checker.py

@@ -44,6 +44,9 @@ class SDKVersionChecker:
             return self._current_version < latest_version
         return False
 
+    def get_sdk_version(self):
+        return self._current_version
+
     def check(self):
         if self.is_outdated():
             print(

langtrace_python_sdk-2.3.11/src/langtrace_python_sdk/version.py

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

langtrace_python_sdk-2.3.9/src/langtrace_python_sdk/constants/__init__.py

@@ -1 +0,0 @@
-LANGTRACE_SDK_NAME = "langtrace-python-sdk"

langtrace_python_sdk-2.3.9/src/langtrace_python_sdk/version.py

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