From 65b6c2706e1d6f97dcabd96cc195668bf39d2e9d Mon Sep 17 00:00:00 2001 From: chyezh Date: Thu, 27 Apr 2023 09:57:29 +0800 Subject: [PATCH] fix connection bug for zilliz uri on milvus (#3278) Co-authored-by: Reinier van der Leer Co-authored-by: merwanehamadi --- autogpt/memory/milvus.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/autogpt/memory/milvus.py b/autogpt/memory/milvus.py index 085f50b4..41f9ed13 100644 --- a/autogpt/memory/milvus.py +++ b/autogpt/memory/milvus.py @@ -61,13 +61,13 @@ class MilvusMemory(MemoryProviderSingleton): if self.uri.startswith("https"): self.secure = True - # Zilliz Cloud requires AutoIndex. - if re.match(r"^https://(.*)\.zillizcloud\.(com|cn)", self.address) is not None: - self.index_params = { - "metric_type": "IP", - "index_type": "AUTOINDEX", - "params": {}, - } + # Zilliz Cloud requires AutoIndex. + if re.match(r"^https://(.*)\.zillizcloud\.(com|cn)", self.uri) is not None: + self.index_params = { + "metric_type": "IP", + "index_type": "AUTOINDEX", + "params": {}, + } def init_collection(self) -> None: """Initialize collection in vector database."""