admin/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Feat/chat add origin (#1130)

Browse Source
This commit is contained in:
zxhlyh
2023-09-09 19:17:12 +08:00
committed by GitHub
parent 6effcd3755
commit 84c76bc04a
74 changed files with 2454 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
web/service/universal-chat.ts
View File

@@ -1,4 +1,4 @@
import type { IOnCompleted, IOnData, IOnError, IOnThought } from './base'
import type { IOnCompleted, IOnData, IOnError, IOnMessageEnd, IOnThought } from './base'
import {
del, get, patch, post, ssePost,
} from './base'
@@ -10,11 +10,12 @@ function getUrl(url: string) {
return `${baseUrl}/${url.startsWith('/') ? url.slice(1) : url}`
}
export const sendChatMessage = async (body: Record<string, any>, { onData, onCompleted, onError, onThought, getAbortController }: {
export const sendChatMessage = async (body: Record<string, any>, { onData, onCompleted, onError, onThought, onMessageEnd, getAbortController }: {
onData: IOnData
onCompleted: IOnCompleted
onError: IOnError
onThought: IOnThought
onMessageEnd: IOnMessageEnd
getAbortController?: (abortController: AbortController) => void
}) => {
return ssePost(getUrl('messages'), {
@@ -22,7 +23,7 @@ export const sendChatMessage = async (body: Record<string, any>, { onData, onCom
...body,
response_mode: 'streaming',
},
}, { onData, onCompleted, onThought, onError, getAbortController })
}, { onData, onCompleted, onThought, onError, getAbortController, onMessageEnd })
}
export const stopChatMessageResponding = async (taskId: string) => {