community: add collection_properties parameter to Milvus (#15788)

- **Description:** add collection_properties parameter to Milvus. See
[pymilvus set_properties()
description](https://milvus.io/api-reference/pymilvus/v2.3.x/Collection/set_properties().md)
  - **Issue:** None
  - **Dependencies:** None
  - **Twitter handle:** None
This commit is contained in:
axiangcoding
2024-01-11 12:29:01 +08:00
committed by GitHub
parent 9e1ed17bfb
commit d5aa277b94
2 changed files with 13 additions and 0 deletions

View File

@@ -18,6 +18,7 @@ class MilvusRetriever(BaseRetriever):
embedding_function: Embeddings
collection_name: str = "LangChainCollection"
collection_properties: Optional[Dict[str, Any]] = None
connection_args: Optional[Dict[str, Any]] = None
consistency_level: str = "Session"
search_params: Optional[dict] = None
@@ -31,6 +32,7 @@ class MilvusRetriever(BaseRetriever):
values["store"] = Milvus(
values["embedding_function"],
values["collection_name"],
values["collection_properties"],
values["connection_args"],
values["consistency_level"],
)