jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

test: add unit tests for vector stores of Milvus, Qdrant and Weaviate (#3688)

Browse source
This commit is contained in:
Bowen Liang 2024-04-24 21:52:42 +08:00 • committed by GitHub
commit 9cec8c1750
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
15 changed files with 211 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api/core/rag/datasource/vdb/milvus/milvus_vector.py
View file

@ -250,7 +250,7 @@ class MilvusVector(BaseVector):
# Create the collection
collection_name = self._collection_name
self._client.create_collection(collection_name=collection_name,
self._client.create_collection_with_schema(collection_name=collection_name,
schema=schema, index_param=index_params,
consistency_level=self._consistency_level)
redis_client.set(collection_exist_cache_key, 1, ex=3600)