getVectorIDsForFile 获取文件相关的所有向量ID
(eid int64, fileID int64)
| 1262 | |
| 1263 | // getVectorIDsForFile 获取文件相关的所有向量ID |
| 1264 | func getVectorIDsForFile(eid int64, fileID int64) ([]string, error) { |
| 1265 | var vectorIDs []string |
| 1266 | |
| 1267 | // 查询文件相关的所有检索块的向量ID |
| 1268 | err := DB.Model(&RetrievalChunk{}). |
| 1269 | Select("vector_id"). |
| 1270 | Where("eid = ? AND file_id = ? AND vector_id IS NOT NULL AND vector_id != ''", eid, fileID). |
| 1271 | Pluck("vector_id", &vectorIDs).Error |
| 1272 | |
| 1273 | if err != nil { |
| 1274 | return nil, fmt.Errorf("查询检索块向量ID失败: %v", err) |
| 1275 | } |
| 1276 | |
| 1277 | return vectorIDs, nil |
| 1278 | } |
| 1279 | |
| 1280 | // deleteVectorsFromDB 从向量数据库中删除向量 |
| 1281 | func deleteVectorsFromDB(eid int64, fileID int64, vectorIDs []string) error { |
no test coverage detected