PyPI - openlit - Versions diffs - 1.33.13__py3-none-any.whl → 1.33.15__py3-none-any.whl - Mend

openlit 1.33.13py3-none-any.whl → 1.33.15py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (19) hide show

openlit/instrumentation/assemblyai/__init__.py +13 -13
openlit/instrumentation/assemblyai/assemblyai.py +35 -28
openlit/instrumentation/astra/__init__.py +82 -83
openlit/instrumentation/astra/astra.py +26 -207
openlit/instrumentation/astra/async_astra.py +27 -208
openlit/instrumentation/astra/utils.py +102 -0
openlit/instrumentation/azure_ai_inference/__init__.py +17 -18
openlit/instrumentation/chroma/chroma.py +10 -10
openlit/instrumentation/dynamiq/dynamiq.py +2 -2
openlit/instrumentation/milvus/milvus.py +10 -10
openlit/instrumentation/pinecone/pinecone.py +8 -8
openlit/instrumentation/qdrant/async_qdrant.py +19 -19
openlit/instrumentation/qdrant/qdrant.py +20 -20
openlit/otel/metrics.py +18 -0
openlit/semcov/__init__.py +26 -13
{openlit-1.33.13.dist-info → openlit-1.33.15.dist-info}/METADATA +1 -1
{openlit-1.33.13.dist-info → openlit-1.33.15.dist-info}/RECORD +19 -18
{openlit-1.33.13.dist-info → openlit-1.33.15.dist-info}/LICENSE +0 -0
{openlit-1.33.13.dist-info → openlit-1.33.15.dist-info}/WHEEL +0 -0

openlit/instrumentation/astra/astra.py CHANGED Viewed

@@ -1,226 +1,45 @@
-# pylint: disable=duplicate-code, broad-exception-caught, too-many-statements, unused-argument, possibly-used-before-assignment, too-many-branches
 """
 Module for monitoring AstraDB.
 """
-import logging
-from opentelemetry.trace import SpanKind, Status, StatusCode
-from opentelemetry.sdk.resources import SERVICE_NAME, TELEMETRY_SDK_NAME, DEPLOYMENT_ENVIRONMENT
-from openlit.__helpers import handle_exception
+import time
+from opentelemetry.trace import SpanKind
+from openlit.instrumentation.astra.utils import (
+    DB_OPERATION_MAP,
+    process_db_operations
+)
 from openlit.semcov import SemanticConvetion
-# Initialize logger for logging potential issues and operations
-logger = logging.getLogger(__name__)
-def object_count(obj):
-    """
-    Counts Length of object if it exists, Else returns None
-    """
-    return len(obj) if obj else None
 def general_wrap(gen_ai_endpoint, version, environment, application_name,
                  tracer, pricing_info, capture_message_content, metrics, disable_metrics):
     """
-    Wraps a AstraDB operation to trace and log its execution metrics.
-    This function is intended to wrap around AstraDB operations in order to
-    measure their execution time, log relevant information, and trace the execution
-    using OpenTelemetry. This helps in monitoring and debugging operations within
-    the AstraDB space.
-    Parameters:
-    - operation (str): The specific AstraDB operation being monitored.
-                                Examples include 'create_index', 'query', 'upsert', etc.
-    - version (str): The version of the application interfacing with AstraDB.
-    - environment (str): The deployment environment, such as 'production' or 'development'.
-    - application_name (str): The name of the application performing the AstraDB operation.
-    - tracer (opentelemetry.trace.Tracer): An object used for OpenTelemetry tracing.
-    - pricing_info (dict): Information about pricing, not used in current implementation.
-    - capture_message_content (bool): A flag indicating whether the content of responses should be traced.
-    Returns:
-    - function: A decorator function that, when applied, wraps the target function with
-                additional functionality for tracing and logging AstraDB operations.
+    Generates a telemetry wrapper for VectorDB function call
     """
     def wrapper(wrapped, instance, args, kwargs):
         """
-        Executes the wrapped AstraDB operation, adding tracing and logging.
-        This inner wrapper function captures the execution of AstraDB operations,
-        annotating the operation with relevant metrics and tracing information, and
-        ensuring any exceptions are caught and logged appropriately.
-        Parameters:
-        - wrapped (Callable): The AstraDB operation to be wrapped and executed.
-        - instance (object): The instance on which the operation is called (for class methods).
-        - args (tuple): Positional arguments for the AstraDB operation.
-        - kwargs (dict): Keyword arguments for the AstraDB operation.
-        Returns:
-        - Any: The result of executing the wrapped AstraDB operation.
+        Wraps the VectorDB function call.
         """
-        with tracer.start_as_current_span(gen_ai_endpoint, kind= SpanKind.CLIENT) as span:
-            response = wrapped(*args, **kwargs)
-            try:
-                span.set_attribute(TELEMETRY_SDK_NAME, "openlit")
-                span.set_attribute(SemanticConvetion.GEN_AI_ENDPOINT,
-                                   gen_ai_endpoint)
-                span.set_attribute(DEPLOYMENT_ENVIRONMENT,
-                                   environment)
-                span.set_attribute(SERVICE_NAME,
-                                   application_name)
-                span.set_attribute(SemanticConvetion.GEN_AI_OPERATION,
-                                   SemanticConvetion.GEN_AI_OPERATION_TYPE_VECTORDB)
-                span.set_attribute(SemanticConvetion.DB_SYSTEM,
-                                   SemanticConvetion.DB_SYSTEM_ASTRA)
-                if gen_ai_endpoint == "astra.create_collection":
-                    db_operation = SemanticConvetion.DB_OPERATION_CREATE_COLLECTION
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       response.name)
-                    span.set_attribute(SemanticConvetion.DB_INDEX_DIMENSION,
-                                       kwargs.get("dimension", ""))
-                    span.set_attribute(SemanticConvetion.DB_INDEX_METRIC,
-                                       str(kwargs.get("metric", "")))
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                        instance.api_endpoint)
+        db_operation = DB_OPERATION_MAP.get(gen_ai_endpoint, "UNKNOWN")
+        if db_operation == SemanticConvetion.DB_OPERATION_REPLACE and kwargs.get('upsert'):
+            db_operation = SemanticConvetion.DB_OPERATION_UPSERT
-                elif gen_ai_endpoint == "astra.drop_collection":
-                    db_operation = SemanticConvetion.DB_OPERATION_DELETE_COLLECTION
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       kwargs.get("name_or_collection", ""))
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                       instance.api_endpoint)
+        span_name = f"{db_operation} {instance.name}"
-                elif gen_ai_endpoint == "astra.insert_one":
-                    db_operation = SemanticConvetion.DB_OPERATION_INSERT
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       instance.name)
-                    span.set_attribute(SemanticConvetion.DB_DOCUMENTS_COUNT,
-                                       1)
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                       instance.database.api_endpoint)
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_ID,
-                                       response.inserted_id)
-                elif gen_ai_endpoint == "astra.insert_many":
-                    db_operation = SemanticConvetion.DB_OPERATION_INSERT
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_DOCUMENTS_COUNT,
-                                       object_count(args[0]))
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       instance.name)
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                       instance.database.api_endpoint)
-                elif gen_ai_endpoint in ["astra.update_one", "astra.update_many"]:
-                    db_operation = SemanticConvetion.DB_OPERATION_UPDATE
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       instance.name)
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                       instance.database.api_endpoint)
-                    span.set_attribute(SemanticConvetion.DB_DOCUMENTS_COUNT,
-                                       response.update_info.get("nModified", 0))
-                elif gen_ai_endpoint == "astra.find_one_and_update":
-                    db_operation = SemanticConvetion.DB_OPERATION_UPDATE
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       instance.name)
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                       instance.database.api_endpoint)
-                elif gen_ai_endpoint == "astra.find":
-                    db_operation = SemanticConvetion.DB_OPERATION_QUERY
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       instance.name)
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                       instance.database.api_endpoint)
-                    span.set_attribute(SemanticConvetion.DB_STATEMENT,
-                                       str(args))
-                elif gen_ai_endpoint == "astra.replace_one":
-                    if kwargs.get("upsert") is True:
-                        db_operation = SemanticConvetion.DB_OPERATION_UPSERT
-                    else:
-                        db_operation = SemanticConvetion.DB_OPERATION_REPLACE
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       instance.name)
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                       instance.database.api_endpoint)
-                    span.set_attribute(SemanticConvetion.DB_STATEMENT,
-                                       str(args))
-                elif gen_ai_endpoint in ["astra.delete_one", "astra.delete_many"]:
-                    db_operation = SemanticConvetion.DB_OPERATION_DELETE
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       instance.name)
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                       instance.database.api_endpoint)
-                    span.set_attribute(SemanticConvetion.DB_STATEMENT,
-                                       str(args))
-                    span.set_attribute(SemanticConvetion.DB_DOCUMENTS_COUNT,
-                                       response.deleted_count)
-                elif gen_ai_endpoint == "astra.find_one_and_delete":
-                    db_operation = SemanticConvetion.DB_OPERATION_DELETE
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       instance.name)
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                       instance.database.api_endpoint)
-                    span.set_attribute(SemanticConvetion.DB_STATEMENT,
-                                       str(args))
-                span.set_status(Status(StatusCode.OK))
-                if disable_metrics is False:
-                    attributes = {
-                        TELEMETRY_SDK_NAME:
-                            "openlit",
-                        SERVICE_NAME:
-                            application_name,
-                        SemanticConvetion.DB_SYSTEM:
-                            SemanticConvetion.DB_SYSTEM_ASTRA,
-                        DEPLOYMENT_ENVIRONMENT:
-                            environment,
-                        SemanticConvetion.GEN_AI_OPERATION:
-                            SemanticConvetion.GEN_AI_OPERATION_TYPE_VECTORDB,
-                        SemanticConvetion.DB_OPERATION:
-                            db_operation
-                    }
-                    metrics["db_requests"].add(1, attributes)
-                # Return original response
-                return response
-            except Exception as e:
-                handle_exception(span, e)
-                logger.error("Error in trace creation: %s", e)
-                # Return original response
-                return response
+        with tracer.start_as_current_span(span_name, kind=SpanKind.CLIENT) as span:
+            start_time = time.time()
+            response = wrapped(*args, **kwargs)
+            server_address = getattr(getattr(instance, 'database', instance), 'api_endpoint', '')
+            server_port = 443
+            collection_name = instance.name
+            response = process_db_operations(
+                response, span, start_time, gen_ai_endpoint,
+                version, environment, application_name, capture_message_content,
+                metrics, disable_metrics, server_address, server_port,
+                collection_name, db_operation, kwargs, args
+            )
+            return response
     return wrapper

openlit/instrumentation/astra/async_astra.py CHANGED Viewed

@@ -1,226 +1,45 @@
-# pylint: disable=duplicate-code, broad-exception-caught, too-many-statements, unused-argument, possibly-used-before-assignment, too-many-branches
 """
 Module for monitoring AstraDB.
 """
-import logging
-from opentelemetry.trace import SpanKind, Status, StatusCode
-from opentelemetry.sdk.resources import SERVICE_NAME, TELEMETRY_SDK_NAME, DEPLOYMENT_ENVIRONMENT
-from openlit.__helpers import handle_exception
+import time
+from opentelemetry.trace import SpanKind
+from openlit.instrumentation.astra.utils import (
+    DB_OPERATION_MAP,
+    process_db_operations
+)
 from openlit.semcov import SemanticConvetion
-# Initialize logger for logging potential issues and operations
-logger = logging.getLogger(__name__)
-def object_count(obj):
-    """
-    Counts Length of object if it exists, Else returns None
-    """
-    return len(obj) if obj else None
-def general_wrap(gen_ai_endpoint, version, environment, application_name,
+def async_general_wrap(gen_ai_endpoint, version, environment, application_name,
                  tracer, pricing_info, capture_message_content, metrics, disable_metrics):
     """
-    Wraps a AstraDB operation to trace and log its execution metrics.
-    This function is intended to wrap around AstraDB operations in order to
-    measure their execution time, log relevant information, and trace the execution
-    using OpenTelemetry. This helps in monitoring and debugging operations within
-    the AstraDB space.
-    Parameters:
-    - operation (str): The specific AstraDB operation being monitored.
-                                Examples include 'create_index', 'query', 'upsert', etc.
-    - version (str): The version of the application interfacing with AstraDB.
-    - environment (str): The deployment environment, such as 'production' or 'development'.
-    - application_name (str): The name of the application performing the AstraDB operation.
-    - tracer (opentelemetry.trace.Tracer): An object used for OpenTelemetry tracing.
-    - pricing_info (dict): Information about pricing, not used in current implementation.
-    - capture_message_content (bool): A flag indicating whether the content of responses should be traced.
-    Returns:
-    - function: A decorator function that, when applied, wraps the target function with
-                additional functionality for tracing and logging AstraDB operations.
+    Generates a telemetry wrapper for VectorDB function call
     """
     async def wrapper(wrapped, instance, args, kwargs):
         """
-        Executes the wrapped AstraDB operation, adding tracing and logging.
-        This inner wrapper function captures the execution of AstraDB operations,
-        annotating the operation with relevant metrics and tracing information, and
-        ensuring any exceptions are caught and logged appropriately.
-        Parameters:
-        - wrapped (Callable): The AstraDB operation to be wrapped and executed.
-        - instance (object): The instance on which the operation is called (for class methods).
-        - args (tuple): Positional arguments for the AstraDB operation.
-        - kwargs (dict): Keyword arguments for the AstraDB operation.
-        Returns:
-        - Any: The result of executing the wrapped AstraDB operation.
+        Wraps the VectorDB function call.
         """
-        with tracer.start_as_current_span(gen_ai_endpoint, kind= SpanKind.CLIENT) as span:
-            response = await wrapped(*args, **kwargs)
-            try:
-                span.set_attribute(TELEMETRY_SDK_NAME, "openlit")
-                span.set_attribute(SemanticConvetion.GEN_AI_ENDPOINT,
-                                   gen_ai_endpoint)
-                span.set_attribute(DEPLOYMENT_ENVIRONMENT,
-                                   environment)
-                span.set_attribute(SERVICE_NAME,
-                                   application_name)
-                span.set_attribute(SemanticConvetion.GEN_AI_OPERATION,
-                                   SemanticConvetion.GEN_AI_OPERATION_TYPE_VECTORDB)
-                span.set_attribute(SemanticConvetion.DB_SYSTEM,
-                                   SemanticConvetion.DB_SYSTEM_ASTRA)
-                if gen_ai_endpoint == "astra.create_collection":
-                    db_operation = SemanticConvetion.DB_OPERATION_CREATE_COLLECTION
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       response.name)
-                    span.set_attribute(SemanticConvetion.DB_INDEX_DIMENSION,
-                                       kwargs.get("dimension", ""))
-                    span.set_attribute(SemanticConvetion.DB_INDEX_METRIC,
-                                       str(kwargs.get("metric", "")))
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                        instance.api_endpoint)
+        db_operation = DB_OPERATION_MAP.get(gen_ai_endpoint, "UNKNOWN")
+        if db_operation == SemanticConvetion.DB_OPERATION_REPLACE and kwargs.get('upsert'):
+            db_operation = SemanticConvetion.DB_OPERATION_UPSERT
-                elif gen_ai_endpoint == "astra.drop_collection":
-                    db_operation = SemanticConvetion.DB_OPERATION_DELETE_COLLECTION
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       kwargs.get("name_or_collection", ""))
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                       instance.api_endpoint)
+        span_name = f"{db_operation} {instance.name}"
-                elif gen_ai_endpoint == "astra.insert_one":
-                    db_operation = SemanticConvetion.DB_OPERATION_INSERT
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       instance.name)
-                    span.set_attribute(SemanticConvetion.DB_DOCUMENTS_COUNT,
-                                       1)
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                       instance.database.api_endpoint)
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_ID,
-                                       response.inserted_id)
-                elif gen_ai_endpoint == "astra.insert_many":
-                    db_operation = SemanticConvetion.DB_OPERATION_INSERT
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_DOCUMENTS_COUNT,
-                                       object_count(args[0]))
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       instance.name)
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                       instance.database.api_endpoint)
-                elif gen_ai_endpoint in ["astra.update_one", "astra.update_many"]:
-                    db_operation = SemanticConvetion.DB_OPERATION_UPDATE
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       instance.name)
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                       instance.database.api_endpoint)
-                    span.set_attribute(SemanticConvetion.DB_DOCUMENTS_COUNT,
-                                       response.update_info.get("nModified", 0))
-                elif gen_ai_endpoint == "astra.find_one_and_update":
-                    db_operation = SemanticConvetion.DB_OPERATION_UPDATE
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       instance.name)
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                       instance.database.api_endpoint)
-                elif gen_ai_endpoint == "astra.find":
-                    db_operation = SemanticConvetion.DB_OPERATION_QUERY
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       instance.name)
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                       instance.database.api_endpoint)
-                    span.set_attribute(SemanticConvetion.DB_STATEMENT,
-                                       str(args))
-                elif gen_ai_endpoint == "astra.replace_one":
-                    if kwargs.get("upsert") is True:
-                        db_operation = SemanticConvetion.DB_OPERATION_UPSERT
-                    else:
-                        db_operation = SemanticConvetion.DB_OPERATION_REPLACE
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       instance.name)
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                       instance.database.api_endpoint)
-                    span.set_attribute(SemanticConvetion.DB_STATEMENT,
-                                       str(args))
-                elif gen_ai_endpoint in ["astra.delete_one", "astra.delete_many"]:
-                    db_operation = SemanticConvetion.DB_OPERATION_DELETE
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       instance.name)
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                       instance.database.api_endpoint)
-                    span.set_attribute(SemanticConvetion.DB_STATEMENT,
-                                       str(args))
-                    span.set_attribute(SemanticConvetion.DB_DOCUMENTS_COUNT,
-                                       response.deleted_count)
-                elif gen_ai_endpoint == "astra.find_one_and_delete":
-                    db_operation = SemanticConvetion.DB_OPERATION_DELETE
-                    span.set_attribute(SemanticConvetion.DB_OPERATION,
-                                       db_operation)
-                    span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
-                                       instance.name)
-                    span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
-                                       instance.database.api_endpoint)
-                    span.set_attribute(SemanticConvetion.DB_STATEMENT,
-                                       str(args))
-                span.set_status(Status(StatusCode.OK))
-                if disable_metrics is False:
-                    attributes = {
-                        TELEMETRY_SDK_NAME:
-                            "openlit",
-                        SERVICE_NAME:
-                            application_name,
-                        SemanticConvetion.DB_SYSTEM:
-                            SemanticConvetion.DB_SYSTEM_ASTRA,
-                        DEPLOYMENT_ENVIRONMENT:
-                            environment,
-                        SemanticConvetion.GEN_AI_OPERATION:
-                            SemanticConvetion.GEN_AI_OPERATION_TYPE_VECTORDB,
-                        SemanticConvetion.DB_OPERATION:
-                            db_operation
-                    }
-                    metrics["db_requests"].add(1, attributes)
-                # Return original response
-                return response
-            except Exception as e:
-                handle_exception(span, e)
-                logger.error("Error in trace creation: %s", e)
-                # Return original response
-                return response
+        with tracer.start_as_current_span(span_name, kind=SpanKind.CLIENT) as span:
+            start_time = time.time()
+            response = await wrapped(*args, **kwargs)
+            server_address = getattr(getattr(instance, 'database', instance), 'api_endpoint', '')
+            server_port = 443
+            collection_name = instance.name
+            response = process_db_operations(
+                response, span, start_time, gen_ai_endpoint,
+                version, environment, application_name, capture_message_content,
+                metrics, disable_metrics, server_address, server_port,
+                collection_name, db_operation, kwargs, args
+            )
+            return response
     return wrapper

openlit/instrumentation/astra/utils.py ADDED Viewed

@@ -0,0 +1,102 @@
+"""
+Astra OpenTelemetry instrumentation utility functions
+"""
+import time
+import logging
+from opentelemetry.trace import Status, StatusCode
+from opentelemetry.sdk.resources import SERVICE_NAME, TELEMETRY_SDK_NAME, DEPLOYMENT_ENVIRONMENT
+from openlit.__helpers import handle_exception
+from openlit.semcov import SemanticConvetion
+# Initialize logger for logging potential issues and operations
+logger = logging.getLogger(__name__)
+def object_count(obj):
+    """Counts Length of object if it exists, Else returns None."""
+    return len(obj) if isinstance(obj, list) else 1
+DB_OPERATION_MAP = {
+    'astra.create_collection': SemanticConvetion.DB_OPERATION_CREATE_COLLECTION,
+    'astra.drop_collection': SemanticConvetion.DB_OPERATION_DELETE_COLLECTION,
+    'astra.insert': SemanticConvetion.DB_OPERATION_INSERT,
+    'astra.update': SemanticConvetion.DB_OPERATION_UPDATE,
+    'astra.find': SemanticConvetion.DB_OPERATION_SELECT,
+    'astra.find_one_and_update': SemanticConvetion.DB_OPERATION_REPLACE,
+    'astra.replace_one': SemanticConvetion.DB_OPERATION_REPLACE,
+    'astra.delete': SemanticConvetion.DB_OPERATION_DELETE,
+    'astra.find_one_and_delete': SemanticConvetion.DB_OPERATION_FIND_AND_DELETE
+}
+def process_db_operations(response, span, start_time, gen_ai_endpoint,
+        version, environment, application_name,
+        capture_message_content, metrics, disable_metrics, server_address,
+        server_port, collection_name, db_operation, kwargs, args):
+    """
+    Process DB operation and generate Telemetry
+    """
+    end_time = time.time()
+    try:
+        span.set_attribute(TELEMETRY_SDK_NAME, 'openlit')
+        span.set_attribute(SemanticConvetion.GEN_AI_OPERATION, SemanticConvetion.GEN_AI_OPERATION_TYPE_VECTORDB)
+        span.set_attribute(SemanticConvetion.DB_SYSTEM_NAME, SemanticConvetion.DB_SYSTEM_ASTRA)
+        span.set_attribute(SemanticConvetion.DB_CLIENT_OPERATION_DURATION, end_time - start_time)
+        span.set_attribute(SemanticConvetion.SERVER_ADDRESS, server_address)
+        span.set_attribute(SemanticConvetion.SERVER_PORT, server_port)
+        span.set_attribute(DEPLOYMENT_ENVIRONMENT, environment)
+        span.set_attribute(SERVICE_NAME, application_name)
+        span.set_attribute(SemanticConvetion.DB_OPERATION_NAME, db_operation)
+        span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME, collection_name)
+        span.set_attribute(SemanticConvetion.DB_SDK_VERSION, version)
+        if db_operation == SemanticConvetion.DB_OPERATION_CREATE_COLLECTION:
+            span.set_attribute(SemanticConvetion.DB_NAMESPACE, response.keyspace)
+            span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME, response.name)
+            span.set_attribute(SemanticConvetion.DB_INDEX_DIMENSION, kwargs.get('dimension', ''))
+            span.set_attribute(SemanticConvetion.DB_INDEX_METRIC, str(kwargs.get('metric', '')))
+        if db_operation == SemanticConvetion.DB_OPERATION_INSERT:
+            span.set_attribute(SemanticConvetion.DB_DOCUMENTS_COUNT, object_count(args[0]))
+            span.set_attribute(SemanticConvetion.DB_QUERY_TEXT, str(args[0] or kwargs.get('documents', {})))
+        elif db_operation == SemanticConvetion.DB_OPERATION_UPDATE:
+            span.set_attribute(SemanticConvetion.DB_RESPONSE_RETURNED_ROWS, response.update_info.get('nModified', 0))
+            span.set_attribute(SemanticConvetion.DB_QUERY_TEXT, str(args[1] or kwargs.get('update', {})))
+        elif db_operation == SemanticConvetion.DB_OPERATION_DELETE:
+            span.set_attribute(SemanticConvetion.DB_RESPONSE_RETURNED_ROWS, response.deleted_count)
+            span.set_attribute(SemanticConvetion.DB_QUERY_TEXT, str(args[0] or kwargs.get('filter', {})))
+        elif db_operation in [
+            SemanticConvetion.DB_OPERATION_SELECT,
+            SemanticConvetion.DB_OPERATION_FIND_AND_DELETE,
+            SemanticConvetion.DB_OPERATION_REPLACE
+        ]:
+            span.set_attribute(SemanticConvetion.DB_QUERY_TEXT, str(args or kwargs.get('filter', {})))
+        span.set_status(Status(StatusCode.OK))
+        if not disable_metrics:
+            attributes = {
+                TELEMETRY_SDK_NAME: 'openlit',
+                SERVICE_NAME: application_name,
+                SemanticConvetion.DB_SYSTEM_NAME: SemanticConvetion.DB_SYSTEM_ASTRA,
+                DEPLOYMENT_ENVIRONMENT: environment,
+                SemanticConvetion.GEN_AI_OPERATION: SemanticConvetion.GEN_AI_OPERATION_TYPE_VECTORDB,
+                SemanticConvetion.DB_OPERATION_NAME: db_operation
+            }
+            metrics['db_requests'].add(1, attributes)
+            metrics['db_client_operation_duration'].record(end_time - start_time, attributes)
+        # Return original response
+        return response
+    except Exception as e:
+        handle_exception(span, e)
+        logger.error('Error in trace creation: %s', e)
+        # Return original response
+        return response

openlit 1.33.13__py3-none-any.whl → 1.33.15__py3-none-any.whl

openlit 1.33.13py3-none-any.whl → 1.33.15py3-none-any.whl