Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-18 09:25:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

community: add a flag that determines whether to load the milvus collection (#15693)

Browse Source 
fix https://github.com/langchain-ai/langchain/issues/15694

---------

Co-authored-by: hyungwookyang <hyungwookyang@worksmobile.com>
Co-authored-by: Bagatur <baskaryan@gmail.com>
This commit is contained in:
YHW 2024-01-16 04:25:23 +09:00 committed by GitHub
parent 9e779ca846
commit ce21392a21
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
libs/community/langchain_community/vectorstores/milvus.py
View File

@ -446,9 +446,15 @@ class Milvus(VectorStore):
timeout: Optional[float] = None,
) -> None:
"""Load the collection if available."""
from pymilvus import Collection
from pymilvus import Collection, utility
from pymilvus.client.types import LoadState
if isinstance(self.col, Collection) and self._get_index() is not None:
if (
isinstance(self.col, Collection)
and self._get_index() is not None
and utility.load_state(self.collection_name, using=self.alias)
== LoadState.NotLoad
):
self.col.load(
partition_names=partition_names,
replica_number=replica_number,