mirror of
https://github.com/labring/FastGPT.git
synced 2025-08-01 20:27:45 +00:00
37 lines
881 B
TypeScript
37 lines
881 B
TypeScript
import { sseResponseEventEnum, TaskResponseKeyEnum } from '@/constants/chat';
|
|
import { sseResponse } from '@/service/utils/tools';
|
|
import { textAdaptGptResponse } from '@/utils/adapt';
|
|
import type { ModuleDispatchProps } from '@/types/core/modules';
|
|
|
|
export type AnswerProps = ModuleDispatchProps<{
|
|
text: string;
|
|
}>;
|
|
export type AnswerResponse = {
|
|
[TaskResponseKeyEnum.answerText]: string;
|
|
finish: boolean;
|
|
};
|
|
|
|
export const dispatchAnswer = (props: Record<string, any>): AnswerResponse => {
|
|
const {
|
|
res,
|
|
detail,
|
|
stream,
|
|
inputs: { text = '' }
|
|
} = props as AnswerProps;
|
|
|
|
if (stream) {
|
|
sseResponse({
|
|
res,
|
|
event: detail ? sseResponseEventEnum.answer : undefined,
|
|
data: textAdaptGptResponse({
|
|
text: text.replace(/\\n/g, '\n')
|
|
})
|
|
});
|
|
}
|
|
|
|
return {
|
|
[TaskResponseKeyEnum.answerText]: text,
|
|
finish: true
|
|
};
|
|
};
|