partners[milvus]: fix integration test issues (#26136)

fix some integration test issues: https://github.com/langchain-ai/langchain/actions/runs/10688447230/job/29628412258 Signed-off-by: ChengZi <[email protected]> Co-authored-by: Erick Friis <[email protected]>
langchain-ai · Sep 6, 2024 · a03141a · a03141a
1 parent 5c1ebd3
commit a03141a
Show file tree

Hide file tree

Showing 4 changed files with 652 additions and 10 deletions.
diff --git a/libs/partners/milvus/langchain_milvus/vectorstores/milvus.py b/libs/partners/milvus/langchain_milvus/vectorstores/milvus.py
@@ -394,7 +394,7 @@ def _create_connection_alias(self, connection_args: dict) -> str:
                     and (addr["address"] == given_address)
                     and ("user" in addr)
                     and (addr["user"] == tmp_user)
-                    and (addr["db_name"] == db_name)
+                    and (addr.get("db_name", "default") == db_name)
                 ):
                     logger.debug("Using previous connection: %s", con[0])
                     return con[0]