ChatGPT/FastGPT
1
0
Fork 0
mirror of https://github.com/labring/FastGPT.git synced 2025-07-22 20:37:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: embedding recall drop-dead halt (#1415)

Browse Source
This commit is contained in:
Archer
2024-05-09 16:13:06 +08:00
committed by GitHub
parent afe5039cd3
commit d4169bf066
3 changed files with 15 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
packages/service/core/dataset/search/controller.ts
View File

@@ -59,19 +59,19 @@ export async function searchDatasetData(props: SearchDatasetDataProps) {
const countRecallLimit = () => {
if (searchMode === DatasetSearchModeEnum.embedding) {
return {
embeddingLimit: 150,
embeddingLimit: 100,
fullTextLimit: 0
};
}
if (searchMode === DatasetSearchModeEnum.fullTextRecall) {
return {
embeddingLimit: 0,
fullTextLimit: 150
fullTextLimit: 100
};
}
return {
embeddingLimit: 100,
fullTextLimit: 80
embeddingLimit: 80,
fullTextLimit: 60
};
};
const embeddingRecall = async ({ query, limit }: { query: string; limit: number }) => {
@@ -82,9 +82,10 @@ export async function searchDatasetData(props: SearchDatasetDataProps) {
});
const { results } = await recallFromVectorStore({
teamId,
datasetIds,
vectors,
limit,
datasetIds,
efSearch: global.systemEnv?.pgHNSWEfSearch
});