ソースを参照

feat: Stream the greetings of the agent dialogue #3355 (#3490)

### What problem does this PR solve?

feat: Stream the greetings of the agent dialogue #3355

### Type of change


- [x] New Feature (non-breaking change which adds functionality)
tags/v0.14.0
balibabu 11ヶ月前
コミット
16fbe9920d
コミッターのメールアドレスに関連付けられたアカウントが存在しません
2個のファイルの変更17行の追加176行の削除
  1. 15
    162
      web/src/pages/flow/chat/hooks.ts
  2. 2
    14
      web/src/pages/flow/hooks.tsx

+ 15
- 162
web/src/pages/flow/chat/hooks.ts ファイルの表示

import { useFetchFlow } from '@/hooks/flow-hooks'; import { useFetchFlow } from '@/hooks/flow-hooks';
import { import {
useHandleMessageInputChange, useHandleMessageInputChange,
useScrollToBottom,
useSelectDerivedMessages, useSelectDerivedMessages,
useSendMessageWithSse, useSendMessageWithSse,
} from '@/hooks/logic-hooks'; } from '@/hooks/logic-hooks';
import { IAnswer, Message } from '@/interfaces/database/chat';
import { IMessage } from '@/pages/chat/interface';
import { Message } from '@/interfaces/database/chat';
import api from '@/utils/api'; import api from '@/utils/api';
import { buildMessageUuid } from '@/utils/chat';
import { message } from 'antd'; import { message } from 'antd';
import trim from 'lodash/trim'; import trim from 'lodash/trim';
import { useCallback, useEffect, useState } from 'react';
import { useCallback, useEffect } from 'react';
import { useParams } from 'umi'; import { useParams } from 'umi';
import { v4 as uuid } from 'uuid'; import { v4 as uuid } from 'uuid';
import { receiveMessageError } from '../utils'; import { receiveMessageError } from '../utils';


const antMessage = message; const antMessage = message;


export const useSelectCurrentMessages = () => {
const { id: id } = useParams();
const [currentMessages, setCurrentMessages] = useState<IMessage[]>([]);

const { data: flowDetail, loading } = useFetchFlow();
const messages = flowDetail.dsl.messages;
const reference = flowDetail.dsl.reference;

const ref = useScrollToBottom(currentMessages);

const addNewestQuestion = useCallback(
(message: Message, answer: string = '') => {
setCurrentMessages((pre) => {
return [
...pre,
{
...message,
id: buildMessageUuid(message),
},
{
role: MessageType.Assistant,
content: answer,
id: buildMessageUuid({ ...message, role: MessageType.Assistant }),
},
];
});
},
[],
);

const addNewestAnswer = useCallback((answer: IAnswer) => {
setCurrentMessages((pre) => {
return [
...pre.slice(0, -1),
{
role: MessageType.Assistant,
content: answer.answer,
reference: answer.reference,
id: buildMessageUuid({
id: answer.id,
role: MessageType.Assistant,
}),
},
];
});
}, []);

const removeLatestMessage = useCallback(() => {
setCurrentMessages((pre) => {
const nextMessages = pre?.slice(0, -2) ?? [];
return nextMessages;
return [...pre, ...nextMessages];
});
}, []);

useEffect(() => {
if (id) {
const nextMessages = messages.map((x) => ({ ...x, id: uuid() }));
setCurrentMessages(nextMessages);
}
}, [messages, id]);

return {
currentMessages,
reference,
addNewestQuestion,
removeLatestMessage,
addNewestAnswer,
ref,
loading,
};
};

export const useSelectNextMessages = () => { export const useSelectNextMessages = () => {
const { id: id } = useParams();
const { data: flowDetail, loading } = useFetchFlow(); const { data: flowDetail, loading } = useFetchFlow();
const messages = flowDetail.dsl.messages;
const reference = flowDetail.dsl.reference; const reference = flowDetail.dsl.reference;
const { const {
derivedMessages, derivedMessages,
setDerivedMessages,
ref, ref,
addNewestQuestion, addNewestQuestion,
addNewestAnswer, addNewestAnswer,
removeMessagesAfterCurrentMessage, removeMessagesAfterCurrentMessage,
} = useSelectDerivedMessages(); } = useSelectDerivedMessages();


useEffect(() => {
if (id) {
const nextMessages = messages.map((x) => ({ ...x, id: uuid() }));
setDerivedMessages(nextMessages);
}
}, [messages, id, setDerivedMessages]);

return { return {
reference, reference,
loading, loading,
}; };
}; };


export const useSendMessage = (
addNewestQuestion: (message: Message, answer?: string) => void,
removeLatestMessage: () => void,
addNewestAnswer: (answer: IAnswer) => void,
) => {
const { id: flowId } = useParams();
const { handleInputChange, value, setValue } = useHandleMessageInputChange();
const { refetch } = useFetchFlow();

const { send, answer, done } = useSendMessageWithSse(api.runCanvas);

const sendMessage = useCallback(
async (message: Message) => {
const params: Record<string, unknown> = {
id: flowId,
};
if (message.content) {
params.message = message.content;
params.message_id = message.id;
}
const res = await send(params);

if (receiveMessageError(res)) {
antMessage.error(res?.data?.message);

// cancel loading
setValue(message.content);
removeLatestMessage();
} else {
refetch(); // pull the message list after sending the message successfully
}
},
[flowId, removeLatestMessage, setValue, send, refetch],
);

const handleSendMessage = useCallback(
async (message: Message) => {
sendMessage(message);
},
[sendMessage],
);

useEffect(() => {
if (answer.answer) {
addNewestAnswer(answer);
}
}, [answer, addNewestAnswer]);

const handlePressEnter = useCallback(() => {
if (trim(value) === '') return;
const id = uuid();
if (done) {
setValue('');
handleSendMessage({ id, content: value.trim(), role: MessageType.User });
}
addNewestQuestion({
content: value,
id,
role: MessageType.User,
});
}, [addNewestQuestion, handleSendMessage, done, setValue, value]);

return {
handlePressEnter,
handleInputChange,
value,
loading: !done,
};
};

export const useSendNextMessage = () => { export const useSendNextMessage = () => {
const { const {
reference, reference,
} = useSelectNextMessages(); } = useSelectNextMessages();
const { id: flowId } = useParams(); const { id: flowId } = useParams();
const { handleInputChange, value, setValue } = useHandleMessageInputChange(); const { handleInputChange, value, setValue } = useHandleMessageInputChange();
const { refetch } = useFetchFlow();


const { send, answer, done } = useSendMessageWithSse(api.runCanvas); const { send, answer, done } = useSendMessageWithSse(api.runCanvas);


// cancel loading // cancel loading
setValue(message.content); setValue(message.content);
removeLatestMessage(); removeLatestMessage();
} else {
refetch(); // pull the message list after sending the message successfully
} }
}, },
[flowId, removeLatestMessage, setValue, send, refetch],
[flowId, removeLatestMessage, setValue, send],
); );


const handleSendMessage = useCallback( const handleSendMessage = useCallback(
}); });
}, [addNewestQuestion, handleSendMessage, done, setValue, value]); }, [addNewestQuestion, handleSendMessage, done, setValue, value]);


const fetchPrologue = useCallback(async () => {
// fetch prologue
const sendRet = await send({ id: flowId });
if (receiveMessageError(sendRet)) {
message.error(sendRet?.data?.message);
}
}, [flowId, send]);

useEffect(() => {
fetchPrologue();
}, [fetchPrologue]);

return { return {
handlePressEnter, handlePressEnter,
handleInputChange, handleInputChange,

+ 2
- 14
web/src/pages/flow/hooks.tsx ファイルの表示

ModelVariableType, ModelVariableType,
settledModelVariableMap, settledModelVariableMap,
} from '@/constants/knowledge'; } from '@/constants/knowledge';
import { useFetchModelId, useSendMessageWithSse } from '@/hooks/logic-hooks';
import { useFetchModelId } from '@/hooks/logic-hooks';
import { Variable } from '@/interfaces/database/chat'; import { Variable } from '@/interfaces/database/chat';
import api from '@/utils/api';
import { useDebounceEffect } from 'ahooks'; import { useDebounceEffect } from 'ahooks';
import { FormInstance, message } from 'antd'; import { FormInstance, message } from 'antd';
import { DefaultOptionType } from 'antd/es/select'; import { DefaultOptionType } from 'antd/es/select';
generateNodeNamesWithIncreasingIndex, generateNodeNamesWithIncreasingIndex,
generateSwitchHandleText, generateSwitchHandleText,
getNodeDragHandle, getNodeDragHandle,
receiveMessageError,
replaceIdWithText, replaceIdWithText,
} from './utils'; } from './utils';


}; };


export const useSaveGraphBeforeOpeningDebugDrawer = (show: () => void) => { export const useSaveGraphBeforeOpeningDebugDrawer = (show: () => void) => {
const { id } = useParams();
const { saveGraph, loading } = useSaveGraph(); const { saveGraph, loading } = useSaveGraph();
const { resetFlow } = useResetFlow(); const { resetFlow } = useResetFlow();
const { refetch } = useFetchFlow();
const { send } = useSendMessageWithSse(api.runCanvas);


const handleRun = useCallback( const handleRun = useCallback(
async (nextNodes?: Node[]) => { async (nextNodes?: Node[]) => {
// After resetting, all previous messages will be cleared. // After resetting, all previous messages will be cleared.
if (resetRet?.code === 0) { if (resetRet?.code === 0) {
show(); show();
// fetch prologue
const sendRet = await send({ id });
if (receiveMessageError(sendRet)) {
message.error(sendRet?.data?.message);
} else {
refetch();
}
} }
} }
}, },
[saveGraph, resetFlow, send, id, refetch, show],
[saveGraph, resetFlow, show],
); );


return { handleRun, loading }; return { handleRun, loading };

読み込み中…
キャンセル
保存