fix: grpc meta auth (#811)

* fix: grpc meta auth * fix: use clip-auth-token env
jina-ai · Aug 30, 2022 · ea23968 · ea23968
1 parent 83a8120
commit ea23968
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/client/clip_client/client.py b/client/clip_client/client.py
@@ -61,7 +61,9 @@ def __init__(self, server: str, credential: dict = {}, **kwargs):
         else:
             raise ValueError(f'{server} is not a valid scheme')
 
-        self._authorization = credential.get('Authorization', None)
+        self._authorization = credential.get(
+            'Authorization', os.environ.get('CLIP_AUTH_TOKEN')
+        )
 
     @overload
     def encode(
@@ -205,8 +207,9 @@ def _get_post_payload(self, content, kwargs):
             request_size=kwargs.get('batch_size', 8),
             total_docs=len(content) if hasattr(content, '__len__') else None,
         )
+
         if self._scheme == 'grpc' and self._authorization:
-            payload.update(metadata=('authorization', self._authorization))
+            payload.update(metadata=(('authorization', self._authorization),))
         elif self._scheme == 'http' and self._authorization:
             payload.update(headers={'Authorization': self._authorization})
         return payload
@@ -413,7 +416,7 @@ def _get_rank_payload(self, content, kwargs):
             total_docs=len(content) if hasattr(content, '__len__') else None,
         )
         if self._scheme == 'grpc' and self._authorization:
-            payload.update(metadata=('authorization', self._authorization))
+            payload.update(metadata=(('authorization', self._authorization),))
         elif self._scheme == 'http' and self._authorization:
             payload.update(headers={'Authorization': self._authorization})
         return payload