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

feat:support baidu vector db (#9185)

Browse source
This commit is contained in:
Shili Cao 2024-10-12 23:24:17 +08:00 • committed by GitHub
commit 2ec6ffe478
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
15 changed files with 582 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

8
api/commands.py
View file

@ -347,6 +347,14 @@ def migrate_knowledge_vector_database():
index_name = Dataset.gen_collection_name_by_id(dataset_id)
index_struct_dict = {"type": "elasticsearch", "vector_store": {"class_prefix": index_name}}
dataset.index_struct = json.dumps(index_struct_dict)
elif vector_type == VectorType.BAIDU:
dataset_id = dataset.id
collection_name = Dataset.gen_collection_name_by_id(dataset_id)
index_struct_dict = {
"type": VectorType.BAIDU,
"vector_store": {"class_prefix": collection_name},
}
dataset.index_struct = json.dumps(index_struct_dict)
else:
raise ValueError(f"Vector store {vector_type} is not supported.")