|
|
|
@@ -72,8 +72,18 @@ class MilvusVector(BaseVector): |
|
|
|
self._client = self._init_client(config) |
|
|
|
self._consistency_level = "Session" # Consistency level for Milvus operations |
|
|
|
self._fields: list[str] = [] # List of fields in the collection |
|
|
|
if self._client.has_collection(collection_name): |
|
|
|
self._load_collection_fields() |
|
|
|
self._hybrid_search_enabled = self._check_hybrid_search_support() # Check if hybrid search is supported |
|
|
|
|
|
|
|
def _load_collection_fields(self, fields: Optional[list[str]] = None) -> None: |
|
|
|
if fields is None: |
|
|
|
# Load collection fields from remote server |
|
|
|
collection_info = self._client.describe_collection(self._collection_name) |
|
|
|
fields = [field["name"] for field in collection_info["fields"]] |
|
|
|
# Since primary field is auto-id, no need to track it |
|
|
|
self._fields = [f for f in fields if f != Field.PRIMARY_KEY.value] |
|
|
|
|
|
|
|
def _check_hybrid_search_support(self) -> bool: |
|
|
|
""" |
|
|
|
Check if the current Milvus version supports hybrid search. |
|
|
|
@@ -306,10 +316,7 @@ class MilvusVector(BaseVector): |
|
|
|
) |
|
|
|
schema.add_function(bm25_function) |
|
|
|
|
|
|
|
for x in schema.fields: |
|
|
|
self._fields.append(x.name) |
|
|
|
# Since primary field is auto-id, no need to track it |
|
|
|
self._fields.remove(Field.PRIMARY_KEY.value) |
|
|
|
self._load_collection_fields([f.name for f in schema.fields]) |
|
|
|
|
|
|
|
# Create Index params for the collection |
|
|
|
index_params_obj = IndexParams() |