From 9647151bd0a43f536632da5e1c282665037f6098 Mon Sep 17 00:00:00 2001 From: hsm207 Date: Mon, 4 Mar 2024 16:09:24 +0000 Subject: [PATCH] update test case --- integrations/weaviate/tests/test_document_store.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/integrations/weaviate/tests/test_document_store.py b/integrations/weaviate/tests/test_document_store.py index 394e0e14d..6e78d1da6 100644 --- a/integrations/weaviate/tests/test_document_store.py +++ b/integrations/weaviate/tests/test_document_store.py @@ -27,7 +27,7 @@ from weaviate.collections.classes.data import DataObject # from weaviate.auth import AuthApiKey as WeaviateAuthApiKey -from weaviate.config import AdditionalConfig, ConnectionConfig +from weaviate.config import AdditionalConfig, ConnectionConfig, Proxies, Timeout from weaviate.embedded import ( DEFAULT_BINARY_PATH, DEFAULT_GRPC_PORT, @@ -314,8 +314,8 @@ def test_from_dict(self, _mock_weaviate, monkeypatch): ], } assert document_store._auth_client_secret == AuthApiKey() - assert document_store._additional_config.timeout == (10, 60) - assert document_store._additional_config.proxies == {"http": "http://proxy:1234"} + assert document_store._additional_config.timeout == Timeout(query=10, insert=60) + assert document_store._additional_config.proxies == Proxies(http="http://proxy:1234", https=None, grpc=None) assert not document_store._additional_config.trust_env assert document_store._additional_headers == {"X-HuggingFace-Api-Key": "MY_HUGGINGFACE_KEY"} assert document_store._embedded_options.persistence_data_path == DEFAULT_PERSISTENCE_DATA_PATH