From f8d941f607669b19917a0e0474f4dcd23a16a031 Mon Sep 17 00:00:00 2001 From: Ali Waleed Date: Tue, 19 Nov 2024 14:58:37 +0200 Subject: [PATCH 1/2] add langtrace attrs --- src/langtrace_python_sdk/instrumentation/milvus/patch.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/langtrace_python_sdk/instrumentation/milvus/patch.py b/src/langtrace_python_sdk/instrumentation/milvus/patch.py index 2c19ac79..354ebef4 100644 --- a/src/langtrace_python_sdk/instrumentation/milvus/patch.py +++ b/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), From db17b94965b2c607aaf5d671f789f48a2e1ef7a1 Mon Sep 17 00:00:00 2001 From: Ali Waleed Date: Tue, 19 Nov 2024 14:59:22 +0200 Subject: [PATCH 2/2] bump version --- src/langtrace_python_sdk/version.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/langtrace_python_sdk/version.py b/src/langtrace_python_sdk/version.py index 1549c123..1de04600 100644 --- a/src/langtrace_python_sdk/version.py +++ b/src/langtrace_python_sdk/version.py @@ -1 +1 @@ -__version__ = "3.3.6" +__version__ = "3.3.7"