generalize the generation of new collection name by dataset id (#2620)

2024-02-29 12:47:10 +08:00
parent 0428f44113
commit 801d135390
5 changed files with 12 additions and 8 deletions
--- a/api/core/rag/datasource/vdb/weaviate/weaviate_vector.py
+++ b/api/core/rag/datasource/vdb/weaviate/weaviate_vector.py
@@ -70,7 +70,7 @@ class WeaviateVector(BaseVector):
            return class_prefix

        dataset_id = dataset.id
-        return "Vector_index_" + dataset_id.replace("-", "_") + '_Node'
+        return Dataset.gen_collection_name_by_id(dataset_id)

    def to_index_struct(self) -> dict:
        return {