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

add meta.doc_id index for tidb (#4963)

Browse source
This commit is contained in:
Jyong 2024-06-05 20:45:43 +08:00 • committed by GitHub
commit 237b8fe3d9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api/core/rag/datasource/vdb/tidb_vector/tidb_vector.py
View file

@ -84,6 +84,8 @@ class TiDBVector(BaseVector):
id CHAR(36) PRIMARY KEY,
text TEXT NOT NULL,
meta JSON NOT NULL,
doc_id VARCHAR(64) AS (JSON_UNQUOTE(JSON_EXTRACT(meta, '$.doc_id'))) STORED,
KEY (doc_id),
vector VECTOR<FLOAT>({dimension}) NOT NULL COMMENT "hnsw(distance={self._distance_func})",
create_time DATETIME DEFAULT CURRENT_TIMESTAMP,
update_time DATETIME DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP