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

chore: reuse existing test functions with upstash vdb (#9679)

Browse source
This commit is contained in:
ice yao 2024-10-23 10:42:11 +08:00 • committed by GitHub
commit ceb2c4f3ef
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 7 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api/core/rag/datasource/vdb/upstash/upstash_vector.py
View file

@ -117,7 +117,7 @@ class UpstashVectorFactory(AbstractVectorFactory):
collection_name = class_prefix.lower()
else:
dataset_id = dataset.id
collection_name = Dataset.gen_collection_name_by_id(dataset_id)
collection_name = Dataset.gen_collection_name_by_id(dataset_id).lower()
dataset.index_struct = json.dumps(self.gen_index_struct_dict(VectorType.UPSTASH, collection_name))
return UpstashVector(