Merge pull request #411 from Scale3-Labs/development

Release: add langtrace attributes for `milvus`

Author: alizenhom

src/langtrace_python_sdk/instrumentation/milvus/patch.py

@@ -1,8 +1,10 @@
+from langtrace_python_sdk.constants.instrumentation.common import SERVICE_PROVIDERS
 from langtrace_python_sdk.utils.silently_fail import silently_fail
 from opentelemetry.trace import Tracer
 from opentelemetry.trace import SpanKind
 from langtrace_python_sdk.utils import handle_span_error, set_span_attribute
 from langtrace_python_sdk.utils.llm import (
+    get_langtrace_attributes,
     get_extra_attributes,
     set_span_attributes,
 )
@@ -16,6 +18,11 @@ def traced_method(wrapped, instance, args, kwargs):
         with tracer.start_as_current_span(span_name, kind=SpanKind.CLIENT) as span:
             try:
                 span_attributes = {
+                    **get_langtrace_attributes(
+                        service_provider=SERVICE_PROVIDERS["MILVUS"],
+                        version=version,
+                        vendor_type="Vector Database",
+                    ),
                     "db.system": "milvus",
                     "db.operation": operation,
                     "db.name": kwargs.get("collection_name", None),

src/langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "3.3.6"
+__version__ = "3.3.7"