update

langgenius · Dec 24, 2024 · 297116a · 297116a
1 parent c9fb21f
commit 297116a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/api/core/rag/datasource/vdb/tidb_vector/tidb_vector.py b/api/core/rag/datasource/vdb/tidb_vector/tidb_vector.py
@@ -108,11 +108,10 @@ def _create_collection(self, dimension: int):
                         create_time DATETIME DEFAULT CURRENT_TIMESTAMP,
                         update_time DATETIME DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
                         KEY (doc_id),
-                        VECTOR INDEX idx_vector ((VEC_COSINE_DISTANCE(vector)))
+                        VECTOR INDEX idx_vector ((VEC_COSINE_DISTANCE(vector))) USING HNSW
                     );
                 """)
                 session.execute(create_statement)
-                # tidb vector not support 'CREATE/ADD INDEX' now
                 session.commit()
             redis_client.set(collection_exist_cache_key, 1, ex=3600)