ChatGPT/FastGPT
1
0
Fork 0
mirror of https://github.com/labring/FastGPT.git synced 2025-07-23 05:12:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

v4.6.6-search test adapt diff search mode (#685)

Browse Source
This commit is contained in:
Archer
2024-01-03 15:40:03 +08:00
committed by GitHub
parent c766a0ed8a
commit 13b10720ac
57 changed files with 1101 additions and 612 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
packages/global/core/dataset/constant.ts
View File

@@ -148,5 +148,19 @@ export enum SearchScoreTypeEnum {
reRank = 'reRank',
rrf = 'rrf'
}
export const SearchScoreTypeMap = {
[SearchScoreTypeEnum.embedding]: {
label: 'core.dataset.search.score.embedding'
},
[SearchScoreTypeEnum.fullText]: {
label: 'core.dataset.search.score.fullText'
},
[SearchScoreTypeEnum.reRank]: {
label: 'core.dataset.search.score.reRank'
},
[SearchScoreTypeEnum.rrf]: {
label: 'core.dataset.search.score.rrf'
}
};
export const FolderAvatarSrc = '/imgs/files/folder.svg';

5
packages/global/core/dataset/type.d.ts vendored
View File

@@ -162,7 +162,10 @@ export type DatasetFileSchema = {
};
/* ============= search =============== */
export type SearchDataResponseItemType = Omit<DatasetDataItemType, 'isOwner' | 'canWrite'> & {
export type SearchDataResponseItemType = Omit<
DatasetDataItemType,
'indexes' | 'isOwner' | 'canWrite'
> & {
score: { type: `${SearchScoreTypeEnum}`; value: number; index: number }[];
// score: number;
};

2
packages/global/core/module/template/input.ts
View File

@@ -27,7 +27,7 @@ export const Input_Template_History: FlowNodeInputItemType = {
export const Input_Template_UserChatInput: FlowNodeInputItemType = {
key: ModuleInputKeyEnum.userChatInput,
type: FlowNodeInputTypeEnum.target,
type: FlowNodeInputTypeEnum.hidden,
label: 'core.module.input.label.user question',
required: true,
valueType: ModuleIOValueTypeEnum.string,

8
packages/global/core/module/template/output.ts
View File

@@ -3,6 +3,14 @@ import { ModuleOutputKeyEnum } from '../constants';
import { FlowNodeOutputTypeEnum } from '../node/constant';
import { ModuleIOValueTypeEnum } from '../constants';
export const Output_Template_UserChatInput: FlowNodeOutputItemType = {
key: ModuleOutputKeyEnum.userChatInput,
label: 'core.module.input.label.user question',
type: FlowNodeOutputTypeEnum.hidden,
valueType: ModuleIOValueTypeEnum.string,
targets: []
};
export const Output_Template_Finish: FlowNodeOutputItemType = {
key: ModuleOutputKeyEnum.finish,
label: 'core.module.output.label.running done',

3
packages/global/core/module/template/system/aiChat.ts
View File

@@ -16,7 +16,7 @@ import {
Input_Template_UserChatInput
} from '../input';
import { chatNodeSystemPromptTip } from '../tip';
import { Output_Template_Finish } from '../output';
import { Output_Template_Finish, Output_Template_UserChatInput } from '../output';
export const AiChatModule: FlowModuleTemplateType = {
id: FlowNodeTypeEnum.chatNode,
@@ -131,6 +131,7 @@ export const AiChatModule: FlowModuleTemplateType = {
Input_Template_UserChatInput
],
outputs: [
Output_Template_UserChatInput,
{
key: ModuleOutputKeyEnum.history,
label: '新的上下文',

2
packages/global/core/module/template/system/classifyQuestion.ts
View File

@@ -10,6 +10,7 @@ import {
Input_Template_Switch,
Input_Template_UserChatInput
} from '../input';
import { Output_Template_UserChatInput } from '../output';
export const ClassifyQuestionModule: FlowModuleTemplateType = {
id: FlowNodeTypeEnum.classifyQuestion,
@@ -72,6 +73,7 @@ export const ClassifyQuestionModule: FlowModuleTemplateType = {
}
],
outputs: [
Output_Template_UserChatInput,
// custom output
{
key: 'wqre',

2
packages/global/core/module/template/system/coreferenceResolution.ts
View File

@@ -15,6 +15,7 @@ import {
Input_Template_Switch,
Input_Template_UserChatInput
} from '../input';
import { Output_Template_UserChatInput } from '../output';
export const AiCFR: FlowModuleTemplateType = {
id: FlowNodeTypeEnum.chatNode,
@@ -50,6 +51,7 @@ export const AiCFR: FlowModuleTemplateType = {
Input_Template_UserChatInput
],
outputs: [
Output_Template_UserChatInput,
{
key: ModuleOutputKeyEnum.text,
label: 'core.module.output.label.cfr result',

3
packages/global/core/module/template/system/datasetSearch.ts
View File

@@ -11,7 +11,7 @@ import {
ModuleTemplateTypeEnum
} from '../../constants';
import { Input_Template_Switch, Input_Template_UserChatInput } from '../input';
import { Output_Template_Finish } from '../output';
import { Output_Template_Finish, Output_Template_UserChatInput } from '../output';
import { DatasetSearchModeEnum } from '../../../dataset/constant';
export const DatasetSearchModule: FlowModuleTemplateType = {
@@ -90,6 +90,7 @@ export const DatasetSearchModule: FlowModuleTemplateType = {
Input_Template_UserChatInput
],
outputs: [
Output_Template_UserChatInput,
{
key: ModuleOutputKeyEnum.datasetIsEmpty,
label: '搜索结果为空',

3
packages/global/core/module/template/system/runApp.ts
View File

@@ -15,7 +15,7 @@ import {
Input_Template_Switch,
Input_Template_UserChatInput
} from '../input';
import { Output_Template_Finish } from '../output';
import { Output_Template_Finish, Output_Template_UserChatInput } from '../output';
export const RunAppModule: FlowModuleTemplateType = {
id: FlowNodeTypeEnum.runApp,
@@ -41,6 +41,7 @@ export const RunAppModule: FlowModuleTemplateType = {
Input_Template_UserChatInput
],
outputs: [
Output_Template_UserChatInput,
{
key: ModuleOutputKeyEnum.history,
label: '新的上下文',