enhance: Use hyphen instead of underscore in grpc header (#2730)

Related to milvus-io/milvus#41357

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>

Author: congqixia

pymilvus/client/async_grpc_handler.py

@@ -199,13 +199,13 @@ def _setup_grpc_channel(self, **kwargs):
             self._final_channel._unary_unary_interceptors.append(async_db_interceptor)
         if self._log_level:
             async_log_level_interceptor = async_header_adder_interceptor(
-                ["log_level"], [self._log_level]
+                ["log-level"], [self._log_level]
             )
             self._final_channel._unary_unary_interceptors.append(async_log_level_interceptor)
             self._log_level = None
         if self._request_id:
             async_request_id_interceptor = async_header_adder_interceptor(
-                ["client_request_id"], [self._request_id]
+                ["client-request-id"], [self._request_id]
             )
             self._final_channel._unary_unary_interceptors.append(async_request_id_interceptor)
             self._request_id = None

pymilvus/client/grpc_handler.py

@@ -259,7 +259,7 @@ def _setup_grpc_channel(self):
             self._log_level = None
         if self._request_id:
             request_id_interceptor = interceptor.header_adder_interceptor(
-                ["client_request_id"], [self._request_id]
+                ["client-request-id"], [self._request_id]
             )
             self._final_channel = grpc.intercept_channel(
                 self._final_channel, request_id_interceptor

pymilvus/decorators.py

@@ -167,8 +167,8 @@ def tracing_request():
     def wrapper(func: Callable):
         @functools.wraps(func)
         def handler(self: Callable, *args, **kwargs):
-            level = kwargs.get("log_level")
-            req_id = kwargs.get("client_request_id")
+            level = kwargs.get("log-level", kwargs.get("log_level"))
+            req_id = kwargs.get("client-request-id", kwargs.get("client_request_id"))
             if level:
                 self.set_onetime_loglevel(level)
             if req_id: