openlit 1.33.13__py3-none-any.whl → 1.33.15__py3-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
  1. openlit/instrumentation/assemblyai/__init__.py +13 -13
  2. openlit/instrumentation/assemblyai/assemblyai.py +35 -28
  3. openlit/instrumentation/astra/__init__.py +82 -83
  4. openlit/instrumentation/astra/astra.py +26 -207
  5. openlit/instrumentation/astra/async_astra.py +27 -208
  6. openlit/instrumentation/astra/utils.py +102 -0
  7. openlit/instrumentation/azure_ai_inference/__init__.py +17 -18
  8. openlit/instrumentation/chroma/chroma.py +10 -10
  9. openlit/instrumentation/dynamiq/dynamiq.py +2 -2
  10. openlit/instrumentation/milvus/milvus.py +10 -10
  11. openlit/instrumentation/pinecone/pinecone.py +8 -8
  12. openlit/instrumentation/qdrant/async_qdrant.py +19 -19
  13. openlit/instrumentation/qdrant/qdrant.py +20 -20
  14. openlit/otel/metrics.py +18 -0
  15. openlit/semcov/__init__.py +26 -13
  16. {openlit-1.33.13.dist-info → openlit-1.33.15.dist-info}/METADATA +1 -1
  17. {openlit-1.33.13.dist-info → openlit-1.33.15.dist-info}/RECORD +19 -18
  18. {openlit-1.33.13.dist-info → openlit-1.33.15.dist-info}/LICENSE +0 -0
  19. {openlit-1.33.13.dist-info → openlit-1.33.15.dist-info}/WHEEL +0 -0
openlit/instrumentation/azure_ai_inference/__init__.py CHANGED
@@ -1,4 +1,3 @@
1
- # pylint: disable=useless-return, bad-staticmethod-argument, disable=duplicate-code
2
1
  """Initializer of Auto Instrumentation of Azure AI Inference Functions"""
3
2
 
4
3
  from typing import Collection
@@ -14,7 +13,7 @@ from openlit.instrumentation.azure_ai_inference.async_azure_ai_inference import
14
13
  async_complete, async_embedding
15
14
  )
16
15
 
17
- _instruments = ("azure-ai-inference >= 1.0.0b4",)
16
+ _instruments = ('azure-ai-inference >= 1.0.0b4',)
18
17
 
19
18
  class AzureAIInferenceInstrumentor(BaseInstrumentor):
20
19
  """
@@ -25,43 +24,43 @@ class AzureAIInferenceInstrumentor(BaseInstrumentor):
25
24
  return _instruments
26
25
 
27
26
  def _instrument(self, **kwargs):
28
- application_name = kwargs.get("application_name", "default_application")
29
- environment = kwargs.get("environment", "default_environment")
30
- tracer = kwargs.get("tracer")
31
- metrics = kwargs.get("metrics_dict")
32
- pricing_info = kwargs.get("pricing_info", {})
33
- capture_message_content = kwargs.get("capture_message_content", False)
34
- disable_metrics = kwargs.get("disable_metrics")
35
- version = importlib.metadata.version("azure-ai-inference")
27
+ application_name = kwargs.get('application_name', 'default')
28
+ environment = kwargs.get('environment', 'default')
29
+ tracer = kwargs.get('tracer')
30
+ metrics = kwargs.get('metrics_dict')
31
+ pricing_info = kwargs.get('pricing_info', {})
32
+ capture_message_content = kwargs.get('capture_message_content', False)
33
+ disable_metrics = kwargs.get('disable_metrics')
34
+ version = importlib.metadata.version('azure-ai-inference')
36
35
 
37
36
  # sync generate
38
37
  wrap_function_wrapper(
39
- "azure.ai.inference",
40
- "ChatCompletionsClient.complete",
38
+ 'azure.ai.inference',
39
+ 'ChatCompletionsClient.complete',
41
40
  complete(version, environment, application_name,
42
41
  tracer, pricing_info, capture_message_content, metrics, disable_metrics),
43
42
  )
44
43
 
45
44
  # sync embedding
46
45
  wrap_function_wrapper(
47
- "azure.ai.inference",
48
- "EmbeddingsClient.embed",
46
+ 'azure.ai.inference',
47
+ 'EmbeddingsClient.embed',
49
48
  embedding(version, environment, application_name,
50
49
  tracer, pricing_info, capture_message_content, metrics, disable_metrics),
51
50
  )
52
51
 
53
52
  # async generate
54
53
  wrap_function_wrapper(
55
- "azure.ai.inference.aio",
56
- "ChatCompletionsClient.complete",
54
+ 'azure.ai.inference.aio',
55
+ 'ChatCompletionsClient.complete',
57
56
  async_complete(version, environment, application_name,
58
57
  tracer, pricing_info, capture_message_content, metrics, disable_metrics),
59
58
  )
60
59
 
61
60
  # async embedding
62
61
  wrap_function_wrapper(
63
- "azure.ai.inference.aio",
64
- "EmbeddingsClient.embed",
62
+ 'azure.ai.inference.aio',
63
+ 'EmbeddingsClient.embed',
65
64
  async_embedding(version, environment, application_name,
66
65
  tracer, pricing_info, capture_message_content, metrics, disable_metrics),
67
66
  )
openlit/instrumentation/chroma/chroma.py CHANGED
@@ -79,14 +79,14 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
79
79
  application_name)
80
80
  span.set_attribute(SemanticConvetion.GEN_AI_OPERATION,
81
81
  SemanticConvetion.GEN_AI_OPERATION_TYPE_VECTORDB)
82
- span.set_attribute(SemanticConvetion.DB_SYSTEM,
82
+ span.set_attribute(SemanticConvetion.DB_SYSTEM_NAME,
83
83
  SemanticConvetion.DB_SYSTEM_CHROMA)
84
84
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
85
85
  instance.name)
86
86
 
87
87
  if gen_ai_endpoint == "chroma.add":
88
88
  db_operation = SemanticConvetion.DB_OPERATION_ADD
89
- span.set_attribute(SemanticConvetion.DB_OPERATION,
89
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
90
90
  SemanticConvetion.DB_OPERATION_ADD)
91
91
  span.set_attribute(SemanticConvetion.DB_ID_COUNT,
92
92
  object_count(kwargs.get("ids", [])))
@@ -99,7 +99,7 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
99
99
 
100
100
  elif gen_ai_endpoint == "chroma.get":
101
101
  db_operation = SemanticConvetion.DB_OPERATION_GET
102
- span.set_attribute(SemanticConvetion.DB_OPERATION,
102
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
103
103
  SemanticConvetion.DB_OPERATION_GET)
104
104
  span.set_attribute(SemanticConvetion.DB_ID_COUNT,
105
105
  object_count(kwargs.get("ids", [])))
@@ -112,7 +112,7 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
112
112
 
113
113
  elif gen_ai_endpoint == "chroma.query":
114
114
  db_operation = SemanticConvetion.DB_OPERATION_QUERY
115
- span.set_attribute(SemanticConvetion.DB_OPERATION,
115
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
116
116
  SemanticConvetion.DB_OPERATION_QUERY)
117
117
  span.set_attribute(SemanticConvetion.DB_STATEMENT,
118
118
  str(kwargs.get("query_texts", "")))
@@ -125,7 +125,7 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
125
125
 
126
126
  elif gen_ai_endpoint == "chroma.update":
127
127
  db_operation = SemanticConvetion.DB_OPERATION_UPDATE
128
- span.set_attribute(SemanticConvetion.DB_OPERATION,
128
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
129
129
  SemanticConvetion.DB_OPERATION_UPDATE)
130
130
  span.set_attribute(SemanticConvetion.DB_VECTOR_COUNT,
131
131
  object_count(kwargs.get("embeddings", [])))
@@ -138,7 +138,7 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
138
138
 
139
139
  elif gen_ai_endpoint == "chroma.upsert":
140
140
  db_operation = SemanticConvetion.DB_OPERATION_UPSERT
141
- span.set_attribute(SemanticConvetion.DB_OPERATION,
141
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
142
142
  SemanticConvetion.DB_OPERATION_UPSERT)
143
143
  span.set_attribute(SemanticConvetion.DB_VECTOR_COUNT,
144
144
  object_count(kwargs.get("embeddings", [])))
@@ -151,7 +151,7 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
151
151
 
152
152
  elif gen_ai_endpoint == "chroma.delete":
153
153
  db_operation = SemanticConvetion.DB_OPERATION_DELETE
154
- span.set_attribute(SemanticConvetion.DB_OPERATION,
154
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
155
155
  SemanticConvetion.DB_OPERATION_DELETE)
156
156
  span.set_attribute(SemanticConvetion.DB_ID_COUNT,
157
157
  object_count(kwargs.get("ids", [])))
@@ -164,7 +164,7 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
164
164
 
165
165
  elif gen_ai_endpoint == "chroma.peek":
166
166
  db_operation = SemanticConvetion.DB_OPERATION_PEEK
167
- span.set_attribute(SemanticConvetion.DB_OPERATION,
167
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
168
168
  SemanticConvetion.DB_OPERATION_PEEK)
169
169
 
170
170
  span.set_status(Status(StatusCode.OK))
@@ -175,13 +175,13 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
175
175
  "openlit",
176
176
  SERVICE_NAME:
177
177
  application_name,
178
- SemanticConvetion.DB_SYSTEM:
178
+ SemanticConvetion.DB_SYSTEM_NAME:
179
179
  SemanticConvetion.DB_SYSTEM_CHROMA,
180
180
  DEPLOYMENT_ENVIRONMENT:
181
181
  environment,
182
182
  SemanticConvetion.GEN_AI_OPERATION:
183
183
  SemanticConvetion.GEN_AI_OPERATION_TYPE_VECTORDB,
184
- SemanticConvetion.DB_OPERATION:
184
+ SemanticConvetion.DB_OPERATION_NAME:
185
185
  db_operation
186
186
  }
187
187
 
openlit/instrumentation/dynamiq/dynamiq.py CHANGED
@@ -84,13 +84,13 @@ def dynamiq_wrap(gen_ai_endpoint, version, environment, application_name,
84
84
  getattr(getattr(instance.flow, 'nodes', [None])[0], 'model', 'default_model'))
85
85
 
86
86
  elif gen_ai_endpoint == "dynamiq.memory_add":
87
- span.set_attribute(SemanticConvetion.DB_OPERATION,
87
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
88
88
  SemanticConvetion.DB_OPERATION_ADD)
89
89
  span.set_attribute(SemanticConvetion.DB_METADATA, str(kwargs.get('metadata', '')))
90
90
 
91
91
  elif gen_ai_endpoint == "dynamiq.memory_search":
92
92
  query_value = kwargs.get('query', '') or (args[0] if args else '')
93
- span.set_attribute(SemanticConvetion.DB_OPERATION,
93
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
94
94
  SemanticConvetion.DB_OPERATION_GET)
95
95
  span.set_attribute(SemanticConvetion.DB_FILTER, str(kwargs.get('filters', '')))
96
96
  span.set_attribute(SemanticConvetion.DB_STATEMENT, query_value)
openlit/instrumentation/milvus/milvus.py CHANGED
@@ -79,12 +79,12 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
79
79
  application_name)
80
80
  span.set_attribute(SemanticConvetion.GEN_AI_OPERATION,
81
81
  SemanticConvetion.GEN_AI_OPERATION_TYPE_VECTORDB)
82
- span.set_attribute(SemanticConvetion.DB_SYSTEM,
82
+ span.set_attribute(SemanticConvetion.DB_SYSTEM_NAME,
83
83
  SemanticConvetion.DB_SYSTEM_MILVUS)
84
84
 
85
85
  if gen_ai_endpoint == "milvus.create_collection":
86
86
  db_operation = SemanticConvetion.DB_OPERATION_CREATE_COLLECTION
87
- span.set_attribute(SemanticConvetion.DB_OPERATION,
87
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
88
88
  SemanticConvetion.DB_OPERATION_CREATE_COLLECTION)
89
89
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
90
90
  kwargs.get("collection_name", ""))
@@ -93,14 +93,14 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
93
93
 
94
94
  elif gen_ai_endpoint == "milvus.drop_collection":
95
95
  db_operation = SemanticConvetion.DB_OPERATION_DELETE_COLLECTION
96
- span.set_attribute(SemanticConvetion.DB_OPERATION,
96
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
97
97
  SemanticConvetion.DB_OPERATION_DELETE_COLLECTION)
98
98
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
99
99
  kwargs.get("collection_name", ""))
100
100
 
101
101
  elif gen_ai_endpoint == "milvus.insert":
102
102
  db_operation = SemanticConvetion.DB_OPERATION_ADD
103
- span.set_attribute(SemanticConvetion.DB_OPERATION,
103
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
104
104
  SemanticConvetion.DB_OPERATION_ADD)
105
105
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
106
106
  kwargs.get("collection_name", ""))
@@ -111,7 +111,7 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
111
111
 
112
112
  elif gen_ai_endpoint == "milvus.search":
113
113
  db_operation = SemanticConvetion.DB_OPERATION_QUERY
114
- span.set_attribute(SemanticConvetion.DB_OPERATION,
114
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
115
115
  SemanticConvetion.DB_OPERATION_QUERY)
116
116
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
117
117
  kwargs.get("collection_name", ""))
@@ -120,7 +120,7 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
120
120
 
121
121
  elif gen_ai_endpoint in ["milvus.query", "milvus.get"]:
122
122
  db_operation = SemanticConvetion.DB_OPERATION_QUERY
123
- span.set_attribute(SemanticConvetion.DB_OPERATION,
123
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
124
124
  SemanticConvetion.DB_OPERATION_QUERY)
125
125
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
126
126
  kwargs.get("collection_name", ""))
@@ -129,7 +129,7 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
129
129
 
130
130
  elif gen_ai_endpoint == "milvus.upsert":
131
131
  db_operation = SemanticConvetion.DB_OPERATION_ADD
132
- span.set_attribute(SemanticConvetion.DB_OPERATION,
132
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
133
133
  SemanticConvetion.DB_OPERATION_UPSERT)
134
134
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
135
135
  kwargs.get("collection_name", ""))
@@ -140,7 +140,7 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
140
140
 
141
141
  elif gen_ai_endpoint == "milvus.delete":
142
142
  db_operation = SemanticConvetion.DB_OPERATION_DELETE
143
- span.set_attribute(SemanticConvetion.DB_OPERATION,
143
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
144
144
  SemanticConvetion.DB_OPERATION_DELETE)
145
145
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
146
146
  kwargs.get("collection_name", ""))
@@ -155,13 +155,13 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
155
155
  "openlit",
156
156
  SERVICE_NAME:
157
157
  application_name,
158
- SemanticConvetion.DB_SYSTEM:
158
+ SemanticConvetion.DB_SYSTEM_NAME:
159
159
  SemanticConvetion.DB_SYSTEM_MILVUS,
160
160
  DEPLOYMENT_ENVIRONMENT:
161
161
  environment,
162
162
  SemanticConvetion.GEN_AI_OPERATION:
163
163
  SemanticConvetion.GEN_AI_OPERATION_TYPE_VECTORDB,
164
- SemanticConvetion.DB_OPERATION:
164
+ SemanticConvetion.DB_OPERATION_NAME:
165
165
  db_operation
166
166
  }
167
167
 
openlit/instrumentation/pinecone/pinecone.py CHANGED
@@ -78,12 +78,12 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
78
78
  application_name)
79
79
  span.set_attribute(SemanticConvetion.GEN_AI_OPERATION,
80
80
  SemanticConvetion.GEN_AI_OPERATION_TYPE_VECTORDB)
81
- span.set_attribute(SemanticConvetion.DB_SYSTEM,
81
+ span.set_attribute(SemanticConvetion.DB_SYSTEM_NAME,
82
82
  SemanticConvetion.DB_SYSTEM_PINECONE)
83
83
 
84
84
  if gen_ai_endpoint == "pinecone.create_index":
85
85
  db_operation = SemanticConvetion.DB_OPERATION_CREATE_INDEX
86
- span.set_attribute(SemanticConvetion.DB_OPERATION,
86
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
87
87
  SemanticConvetion.DB_OPERATION_CREATE_INDEX)
88
88
  span.set_attribute(SemanticConvetion.DB_INDEX_NAME,
89
89
  kwargs.get("name", ""))
@@ -96,7 +96,7 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
96
96
 
97
97
  elif gen_ai_endpoint == "pinecone.query":
98
98
  db_operation = SemanticConvetion.DB_OPERATION_QUERY
99
- span.set_attribute(SemanticConvetion.DB_OPERATION,
99
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
100
100
  SemanticConvetion.DB_OPERATION_QUERY)
101
101
  span.set_attribute(SemanticConvetion.DB_STATEMENT,
102
102
  str(kwargs.get("vector")))
@@ -109,7 +109,7 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
109
109
 
110
110
  elif gen_ai_endpoint == "pinecone.update":
111
111
  db_operation = SemanticConvetion.DB_OPERATION_UPDATE
112
- span.set_attribute(SemanticConvetion.DB_OPERATION,
112
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
113
113
  SemanticConvetion.DB_OPERATION_UPDATE)
114
114
  span.set_attribute(SemanticConvetion.DB_UPDATE_ID,
115
115
  kwargs.get("id",""))
@@ -122,14 +122,14 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
122
122
 
123
123
  elif gen_ai_endpoint == "pinecone.upsert":
124
124
  db_operation = SemanticConvetion.DB_OPERATION_UPSERT
125
- span.set_attribute(SemanticConvetion.DB_OPERATION,
125
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
126
126
  SemanticConvetion.DB_OPERATION_UPSERT)
127
127
  span.set_attribute(SemanticConvetion.DB_VECTOR_COUNT,
128
128
  object_count(kwargs.get("vectors")))
129
129
 
130
130
  elif gen_ai_endpoint == "pinecone.delete":
131
131
  db_operation = SemanticConvetion.DB_OPERATION_DELETE
132
- span.set_attribute(SemanticConvetion.DB_OPERATION,
132
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
133
133
  SemanticConvetion.DB_OPERATION_DELETE)
134
134
  span.set_attribute(SemanticConvetion.DB_ID_COUNT,
135
135
  object_count(kwargs.get("ids")))
@@ -148,13 +148,13 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
148
148
  "openlit",
149
149
  SERVICE_NAME:
150
150
  application_name,
151
- SemanticConvetion.DB_SYSTEM:
151
+ SemanticConvetion.DB_SYSTEM_NAME:
152
152
  SemanticConvetion.DB_SYSTEM_PINECONE,
153
153
  DEPLOYMENT_ENVIRONMENT:
154
154
  environment,
155
155
  SemanticConvetion.GEN_AI_OPERATION:
156
156
  SemanticConvetion.GEN_AI_OPERATION_TYPE_VECTORDB,
157
- SemanticConvetion.DB_OPERATION:
157
+ SemanticConvetion.DB_OPERATION_NAME:
158
158
  db_operation
159
159
  }
160
160
 
openlit/instrumentation/qdrant/async_qdrant.py CHANGED
@@ -79,40 +79,40 @@ def async_general_wrap(gen_ai_endpoint, version, environment, application_name,
79
79
  application_name)
80
80
  span.set_attribute(SemanticConvetion.GEN_AI_OPERATION,
81
81
  SemanticConvetion.GEN_AI_OPERATION_TYPE_VECTORDB)
82
- span.set_attribute(SemanticConvetion.DB_SYSTEM,
82
+ span.set_attribute(SemanticConvetion.DB_SYSTEM_NAME,
83
83
  SemanticConvetion.DB_SYSTEM_QDRANT)
84
84
 
85
85
  if gen_ai_endpoint == "qdrant.create_collection":
86
86
  db_operation = SemanticConvetion.DB_OPERATION_CREATE_COLLECTION
87
- span.set_attribute(SemanticConvetion.DB_OPERATION,
87
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
88
88
  SemanticConvetion.DB_OPERATION_CREATE_COLLECTION)
89
89
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
90
90
  kwargs.get("collection_name", ""))
91
91
 
92
92
  elif gen_ai_endpoint == "qdrant.upload_collection":
93
93
  db_operation = SemanticConvetion.DB_OPERATION_CREATE_COLLECTION
94
- span.set_attribute(SemanticConvetion.DB_OPERATION,
94
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
95
95
  SemanticConvetion.DB_OPERATION_CREATE_COLLECTION)
96
96
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
97
97
  kwargs.get("collection_name", ""))
98
98
 
99
99
  elif gen_ai_endpoint == "qdrant.delete_collection":
100
100
  db_operation = SemanticConvetion.DB_OPERATION_DELETE_COLLECTION
101
- span.set_attribute(SemanticConvetion.DB_OPERATION,
101
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
102
102
  SemanticConvetion.DB_OPERATION_DELETE_COLLECTION)
103
103
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
104
104
  kwargs.get("collection_name", ""))
105
105
 
106
106
  elif gen_ai_endpoint == "qdrant.update_collection":
107
107
  db_operation = SemanticConvetion.DB_OPERATION_UPDATE_COLLECTION
108
- span.set_attribute(SemanticConvetion.DB_OPERATION,
108
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
109
109
  SemanticConvetion.DB_OPERATION_UPDATE_COLLECTION)
110
110
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
111
111
  kwargs.get("collection_name", ""))
112
112
 
113
113
  elif gen_ai_endpoint == "qdrant.set_payload":
114
114
  db_operation = SemanticConvetion.DB_OPERATION_ADD
115
- span.set_attribute(SemanticConvetion.DB_OPERATION,
115
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
116
116
  SemanticConvetion.DB_OPERATION_ADD)
117
117
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
118
118
  kwargs.get("collection_name", ""))
@@ -125,7 +125,7 @@ def async_general_wrap(gen_ai_endpoint, version, environment, application_name,
125
125
 
126
126
  elif gen_ai_endpoint == "qdrant.retrieve":
127
127
  db_operation = SemanticConvetion.DB_OPERATION_QUERY
128
- span.set_attribute(SemanticConvetion.DB_OPERATION,
128
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
129
129
  SemanticConvetion.DB_OPERATION_QUERY)
130
130
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
131
131
  kwargs.get("collection_name", ""))
@@ -134,7 +134,7 @@ def async_general_wrap(gen_ai_endpoint, version, environment, application_name,
134
134
 
135
135
  elif gen_ai_endpoint == "qdrant.scroll":
136
136
  db_operation = SemanticConvetion.DB_OPERATION_QUERY
137
- span.set_attribute(SemanticConvetion.DB_OPERATION,
137
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
138
138
  SemanticConvetion.DB_OPERATION_QUERY)
139
139
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
140
140
  kwargs.get("collection_name", ""))
@@ -143,7 +143,7 @@ def async_general_wrap(gen_ai_endpoint, version, environment, application_name,
143
143
 
144
144
  elif gen_ai_endpoint in ["qdrant.search", "qdrant.search_groups"]:
145
145
  db_operation = SemanticConvetion.DB_OPERATION_QUERY
146
- span.set_attribute(SemanticConvetion.DB_OPERATION,
146
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
147
147
  SemanticConvetion.DB_OPERATION_QUERY)
148
148
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
149
149
  kwargs.get("collection_name", ""))
@@ -152,7 +152,7 @@ def async_general_wrap(gen_ai_endpoint, version, environment, application_name,
152
152
 
153
153
  elif gen_ai_endpoint == "qdrant.recommend":
154
154
  db_operation = SemanticConvetion.DB_OPERATION_QUERY
155
- span.set_attribute(SemanticConvetion.DB_OPERATION,
155
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
156
156
  SemanticConvetion.DB_OPERATION_QUERY)
157
157
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
158
158
  kwargs.get("collection_name", ""))
@@ -162,7 +162,7 @@ def async_general_wrap(gen_ai_endpoint, version, environment, application_name,
162
162
 
163
163
  elif gen_ai_endpoint == "qdrant.upload_points":
164
164
  db_operation = SemanticConvetion.DB_OPERATION_ADD
165
- span.set_attribute(SemanticConvetion.DB_OPERATION,
165
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
166
166
  SemanticConvetion.DB_OPERATION_ADD)
167
167
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
168
168
  kwargs.get("collection_name", ""))
@@ -171,7 +171,7 @@ def async_general_wrap(gen_ai_endpoint, version, environment, application_name,
171
171
 
172
172
  elif gen_ai_endpoint == "qdrant.update_vectors":
173
173
  db_operation = SemanticConvetion.DB_OPERATION_UPDATE
174
- span.set_attribute(SemanticConvetion.DB_OPERATION,
174
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
175
175
  SemanticConvetion.DB_OPERATION_UPDATE)
176
176
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
177
177
  kwargs.get("collection_name", ""))
@@ -182,7 +182,7 @@ def async_general_wrap(gen_ai_endpoint, version, environment, application_name,
182
182
 
183
183
  elif gen_ai_endpoint == "qdrant.overwrite_payload":
184
184
  db_operation = SemanticConvetion.DB_OPERATION_UPDATE
185
- span.set_attribute(SemanticConvetion.DB_OPERATION,
185
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
186
186
  SemanticConvetion.DB_OPERATION_UPDATE)
187
187
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
188
188
  kwargs.get("collection_name", ""))
@@ -197,7 +197,7 @@ def async_general_wrap(gen_ai_endpoint, version, environment, application_name,
197
197
  db_operation = SemanticConvetion.DB_OPERATION_UPSERT
198
198
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
199
199
  kwargs.get("collection_name", ""))
200
- span.set_attribute(SemanticConvetion.DB_OPERATION,
200
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
201
201
  SemanticConvetion.DB_OPERATION_UPSERT)
202
202
  span.set_attribute(SemanticConvetion.DB_OPERATION_STATUS,
203
203
  response.status)
@@ -206,7 +206,7 @@ def async_general_wrap(gen_ai_endpoint, version, environment, application_name,
206
206
 
207
207
  elif gen_ai_endpoint in ["qdrant.delete_payload", "qdrant.delete_vectors"]:
208
208
  db_operation = SemanticConvetion.DB_OPERATION_DELETE
209
- span.set_attribute(SemanticConvetion.DB_OPERATION,
209
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
210
210
  SemanticConvetion.DB_OPERATION_DELETE)
211
211
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
212
212
  kwargs.get("collection_name", ""))
@@ -217,7 +217,7 @@ def async_general_wrap(gen_ai_endpoint, version, environment, application_name,
217
217
 
218
218
  elif gen_ai_endpoint in ["qdrant.clear_payload", "qdrant.delete"]:
219
219
  db_operation = SemanticConvetion.DB_OPERATION_DELETE
220
- span.set_attribute(SemanticConvetion.DB_OPERATION,
220
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
221
221
  SemanticConvetion.DB_OPERATION_DELETE)
222
222
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
223
223
  kwargs.get("collection_name", ""))
@@ -228,7 +228,7 @@ def async_general_wrap(gen_ai_endpoint, version, environment, application_name,
228
228
 
229
229
  elif gen_ai_endpoint == "qdrant.create_payload_index":
230
230
  db_operation = SemanticConvetion.DB_OPERATION_CREATE_INDEX
231
- span.set_attribute(SemanticConvetion.DB_OPERATION,
231
+ span.set_attribute(SemanticConvetion.DB_OPERATION_NAME,
232
232
  SemanticConvetion.DB_OPERATION_CREATE_INDEX)
233
233
  span.set_attribute(SemanticConvetion.DB_COLLECTION_NAME,
234
234
  kwargs.get("collection_name", ""))
@@ -243,13 +243,13 @@ def async_general_wrap(gen_ai_endpoint, version, environment, application_name,
243
243
  "openlit",
244
244
  SERVICE_NAME:
245
245
  application_name,
246
- SemanticConvetion.DB_SYSTEM:
246
+ SemanticConvetion.DB_SYSTEM_NAME:
247
247
  SemanticConvetion.DB_SYSTEM_QDRANT,
248
248
  DEPLOYMENT_ENVIRONMENT:
249
249
  environment,
250
250
  SemanticConvetion.GEN_AI_OPERATION:
251
251
  SemanticConvetion.GEN_AI_OPERATION_TYPE_VECTORDB,
252
- SemanticConvetion.DB_OPERATION:
252
+ SemanticConvetion.DB_OPERATION_NAME:
253
253
  db_operation
254
254
  }
255
255