v4.6.2-alpah (#511)
This commit is contained in:
@@ -27,6 +27,7 @@ export async function insertData2Dataset({
|
||||
model: string;
|
||||
}) {
|
||||
if (!q || !datasetId || !collectionId || !model) {
|
||||
console.log(q, a, datasetId, collectionId, model);
|
||||
return Promise.reject('q, datasetId, collectionId, model is required');
|
||||
}
|
||||
if (String(teamId) === String(tmbId)) {
|
||||
|
||||
@@ -153,11 +153,11 @@ export async function searchDatasetData({
|
||||
SET LOCAL hnsw.ef_search = ${global.systemEnv.pgHNSWEfSearch || 100};
|
||||
select id, collection_id, data_id, (vector <#> '[${
|
||||
vectors[0]
|
||||
}]') * -1 AS score from ${PgDatasetTableName} where dataset_id IN (${datasetIds
|
||||
.map((id) => `'${String(id)}'`)
|
||||
.join(',')}) AND vector <#> '[${vectors[0]}]' < -${similarity} order by vector <#> '[${
|
||||
}]') * -1 AS score from ${PgDatasetTableName}
|
||||
where dataset_id IN (${datasetIds.map((id) => `'${String(id)}'`).join(',')}) AND vector <#> '[${
|
||||
vectors[0]
|
||||
}]' limit ${minLimit};
|
||||
}]' < -${similarity}
|
||||
order by score desc limit ${minLimit};
|
||||
COMMIT;`
|
||||
);
|
||||
|
||||
|
||||
Reference in New Issue
Block a user