Skip to content

Commit eae4c5b

Browse files
authored
enhance: [2.5] Use hyphen instead of underscore in grpc header (#2730) (#2731)
Cherry-pick from master pr: #2730 Related to milvus-io/milvus#41357 Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
1 parent 57f54b0 commit eae4c5b

File tree

3 files changed

+5
-5
lines changed

3 files changed

+5
-5
lines changed

pymilvus/client/async_grpc_handler.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -199,13 +199,13 @@ def _setup_grpc_channel(self, **kwargs):
199199
self._final_channel._unary_unary_interceptors.append(async_db_interceptor)
200200
if self._log_level:
201201
async_log_level_interceptor = async_header_adder_interceptor(
202-
["log_level"], [self._log_level]
202+
["log-level"], [self._log_level]
203203
)
204204
self._final_channel._unary_unary_interceptors.append(async_log_level_interceptor)
205205
self._log_level = None
206206
if self._request_id:
207207
async_request_id_interceptor = async_header_adder_interceptor(
208-
["client_request_id"], [self._request_id]
208+
["client-request-id"], [self._request_id]
209209
)
210210
self._final_channel._unary_unary_interceptors.append(async_request_id_interceptor)
211211
self._request_id = None

pymilvus/client/grpc_handler.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -251,7 +251,7 @@ def _setup_grpc_channel(self):
251251
self._log_level = None
252252
if self._request_id:
253253
request_id_interceptor = interceptor.header_adder_interceptor(
254-
["client_request_id"], [self._request_id]
254+
["client-request-id"], [self._request_id]
255255
)
256256
self._final_channel = grpc.intercept_channel(
257257
self._final_channel, request_id_interceptor

pymilvus/decorators.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -167,8 +167,8 @@ def tracing_request():
167167
def wrapper(func: Callable):
168168
@functools.wraps(func)
169169
def handler(self: Callable, *args, **kwargs):
170-
level = kwargs.get("log_level")
171-
req_id = kwargs.get("client_request_id")
170+
level = kwargs.get("log-level", kwargs.get("log_level"))
171+
req_id = kwargs.get("client-request-id", kwargs.get("client_request_id"))
172172
if level:
173173
self.set_onetime_loglevel(level)
174174
if req_id:

0 commit comments

Comments
 (0)