ChatGPT/FastGPT
1
0
Fork 0
mirror of https://github.com/labring/FastGPT.git synced 2025-10-18 01:16:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

4.6.7 first pr (#726)

Browse Source
This commit is contained in:
Archer
2024-01-10 23:35:04 +08:00
committed by GitHub
parent 414b693303
commit 006ad17c6a
186 changed files with 2996 additions and 1838 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
packages/service/common/vectorStore/controller.d.ts vendored
View File

@@ -2,6 +2,8 @@ export type DeleteDatasetVectorProps = {
id?: string;
datasetIds?: string[];
collectionIds?: string[];
collectionId?: string;
dataIds?: string[];
};

11
packages/service/common/vectorStore/pg/controller.ts
View File

@@ -101,14 +101,19 @@ export const deleteDatasetDataVector = async (
retry?: number;
}
): Promise<any> => {
const { id, datasetIds, collectionIds, dataIds, retry = 2 } = props;
const { id, datasetIds, collectionIds, collectionId, dataIds, retry = 2 } = props;
const where = await (() => {
if (id) return `id=${id}`;
if (datasetIds) return `dataset_id IN (${datasetIds.map((id) => `'${String(id)}'`).join(',')})`;
if (collectionIds)
if (collectionIds) {
return `collection_id IN (${collectionIds.map((id) => `'${String(id)}'`).join(',')})`;
if (dataIds) return `data_id IN (${dataIds.map((id) => `'${String(id)}'`).join(',')})`;
}
if (collectionId && dataIds) {
return `collection_id='${String(collectionId)}' and data_id IN (${dataIds
.map((id) => `'${String(id)}'`)
.join(',')})`;
}
return Promise.reject('deleteDatasetData: no where');
})();