github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-11 22:09:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: UI component rendering in agent dialog mode (#1083)

Browse Source 
Co-authored-by: csunny <cfqsunny@163.com>
This commit is contained in:
Hzh_97
2024-01-18 11:08:02 +08:00
committed by GitHub
parent 674104eb7c
commit 0936856c3a
46 changed files with 1597 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
web/hooks/use-chat.ts
View File

@@ -10,6 +10,7 @@ type Props = {
type ChatParams = {
chatId: string;
data?: Record<string, any>;
query?: Record<string, string>;
onMessage: (message: string) => void;
onClose?: () => void;
onDone?: () => void;
@@ -22,7 +23,7 @@ const useChat = ({ queryAgentURL = '/api/v1/chat/completions' }: Props) => {
const chat = useCallback(
async ({ data, chatId, onMessage, onClose, onDone, onError }: ChatParams) => {
if (!data?.user_input && !data?.doc_id) {
message.warning(i18n.t('NoContextTip'));
message.warning(i18n.t('no_context_tip'));
return;
}
@@ -58,7 +59,12 @@ const useChat = ({ queryAgentURL = '/api/v1/chat/completions' }: Props) => {
throw new Error(err);
},
onmessage: (event) => {
const message = event.data?.replaceAll('\\n', '\n');
let message = event.data;
try {
message = JSON.parse(message).vis;
} catch (e) {
message.replaceAll('\\n', '\n');
}
if (message === '[DONE]') {
onDone?.();
} else if (message?.startsWith('[ERROR]')) {