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

2024-10-23 10:42:11 +08:00
parent d5a93a6400
commit ceb2c4f3ef
3 changed files with 7 additions and 42 deletions
--- a/api/core/rag/datasource/vdb/upstash/upstash_vector.py
+++ b/api/core/rag/datasource/vdb/upstash/upstash_vector.py
@@ -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(