From 30dfbf350d05703b62f22c1906960ee48dd63ad8 Mon Sep 17 00:00:00 2001 From: Girish Chandrashekar Date: Mon, 26 Jun 2023 16:59:13 +0200 Subject: [PATCH] fix: allow additional_headers parameters to Weaviate client Weaviate v3.9.0 supports `additional_headers` config param to pass in authentication headers. Signed-off-by: Girish Chandrashekar --- docarray/array/storage/weaviate/backend.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/docarray/array/storage/weaviate/backend.py b/docarray/array/storage/weaviate/backend.py index 57c5c811740..2deb517474d 100644 --- 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