fix: allow additional_headers parameters to Weaviate client (#1673)

girishc13 · web-flow · commit debbe3f9d1b7 · 2023-06-26T17:48:13.000+02:00
Signed-off-by: Girish Chandrashekar &lt;girish.chandrashekar@jina.ai&gt;
diff --git a/docarray/array/storage/weaviate/backend.py b/docarray/array/storage/weaviate/backend.py
@@ -9,6 +9,7 @@
     Union,
     Tuple,
     List,
+    Any,
 )
 
 import numpy as np
@@ -31,6 +32,7 @@ class WeaviateConfig:
     host: Optional[str] = field(default='localhost')
     port: Optional[int] = field(default=8080)
     protocol: Optional[str] = field(default='http')
+    additional_headers: Optional[Dict[str, Any]] = field(default_factory=dict)
     name: Optional[str] = None
     list_like: bool = True
     serialize_config: Dict = field(default_factory=dict)
@@ -124,6 +126,7 @@ def _init_storage(
         self._client = weaviate.Client(
             f'{config.protocol}://{config.host}:{config.port}',
             timeout_config=config.timeout_config,
+            additional_headers=config.additional_headers,
         )
         self._config = config
 
