mirror of
https://gitee.com/dify_ai/dify.git
synced 2024-12-10 17:27:34 +08:00
222 lines
9.3 KiB
TypeScript
222 lines
9.3 KiB
TypeScript
import type { IOnCompleted, IOnData, IOnError, IOnFile, IOnIterationFinished, IOnIterationNexted, IOnIterationStarted, IOnMessageEnd, IOnMessageReplace, IOnNodeFinished, IOnNodeStarted, IOnTextChunk, IOnTextReplace, IOnThought, IOnWorkflowFinished, IOnWorkflowStarted } from './base'
|
|
import {
|
|
del as consoleDel, get as consoleGet, patch as consolePatch, post as consolePost,
|
|
delPublic as del, getPublic as get, patchPublic as patch, postPublic as post, ssePost,
|
|
} from './base'
|
|
import type { Feedbacktype } from '@/app/components/base/chat/chat/type'
|
|
import type {
|
|
AppConversationData,
|
|
AppData,
|
|
AppMeta,
|
|
ConversationItem,
|
|
} from '@/models/share'
|
|
import type { ChatConfig } from '@/app/components/base/chat/types'
|
|
import type { SystemFeatures } from '@/types/feature'
|
|
|
|
function getAction(action: 'get' | 'post' | 'del' | 'patch', isInstalledApp: boolean) {
|
|
switch (action) {
|
|
case 'get':
|
|
return isInstalledApp ? consoleGet : get
|
|
case 'post':
|
|
return isInstalledApp ? consolePost : post
|
|
case 'patch':
|
|
return isInstalledApp ? consolePatch : patch
|
|
case 'del':
|
|
return isInstalledApp ? consoleDel : del
|
|
}
|
|
}
|
|
|
|
export function getUrl(url: string, isInstalledApp: boolean, installedAppId: string) {
|
|
return isInstalledApp ? `installed-apps/${installedAppId}/${url.startsWith('/') ? url.slice(1) : url}` : url
|
|
}
|
|
|
|
export const sendChatMessage = async (body: Record<string, any>, { onData, onCompleted, onThought, onFile, onError, getAbortController, onMessageEnd, onMessageReplace }: {
|
|
onData: IOnData
|
|
onCompleted: IOnCompleted
|
|
onFile: IOnFile
|
|
onThought: IOnThought
|
|
onError: IOnError
|
|
onMessageEnd?: IOnMessageEnd
|
|
onMessageReplace?: IOnMessageReplace
|
|
getAbortController?: (abortController: AbortController) => void
|
|
}, isInstalledApp: boolean, installedAppId = '') => {
|
|
return ssePost(getUrl('chat-messages', isInstalledApp, installedAppId), {
|
|
body: {
|
|
...body,
|
|
response_mode: 'streaming',
|
|
},
|
|
}, { onData, onCompleted, onThought, onFile, isPublicAPI: !isInstalledApp, onError, getAbortController, onMessageEnd, onMessageReplace })
|
|
}
|
|
|
|
export const stopChatMessageResponding = async (appId: string, taskId: string, isInstalledApp: boolean, installedAppId = '') => {
|
|
return getAction('post', isInstalledApp)(getUrl(`chat-messages/${taskId}/stop`, isInstalledApp, installedAppId))
|
|
}
|
|
|
|
export const sendCompletionMessage = async (body: Record<string, any>, { onData, onCompleted, onError, onMessageReplace }: {
|
|
onData: IOnData
|
|
onCompleted: IOnCompleted
|
|
onError: IOnError
|
|
onMessageReplace: IOnMessageReplace
|
|
}, isInstalledApp: boolean, installedAppId = '') => {
|
|
return ssePost(getUrl('completion-messages', isInstalledApp, installedAppId), {
|
|
body: {
|
|
...body,
|
|
response_mode: 'streaming',
|
|
},
|
|
}, { onData, onCompleted, isPublicAPI: !isInstalledApp, onError, onMessageReplace })
|
|
}
|
|
|
|
export const sendWorkflowMessage = async (
|
|
body: Record<string, any>,
|
|
{
|
|
onWorkflowStarted,
|
|
onNodeStarted,
|
|
onNodeFinished,
|
|
onWorkflowFinished,
|
|
onIterationStart,
|
|
onIterationNext,
|
|
onIterationFinish,
|
|
onTextChunk,
|
|
onTextReplace,
|
|
}: {
|
|
onWorkflowStarted: IOnWorkflowStarted
|
|
onNodeStarted: IOnNodeStarted
|
|
onNodeFinished: IOnNodeFinished
|
|
onWorkflowFinished: IOnWorkflowFinished
|
|
onIterationStart: IOnIterationStarted
|
|
onIterationNext: IOnIterationNexted
|
|
onIterationFinish: IOnIterationFinished
|
|
onTextChunk: IOnTextChunk
|
|
onTextReplace: IOnTextReplace
|
|
},
|
|
isInstalledApp: boolean,
|
|
installedAppId = '',
|
|
) => {
|
|
return ssePost(getUrl('workflows/run', isInstalledApp, installedAppId), {
|
|
body: {
|
|
...body,
|
|
response_mode: 'streaming',
|
|
},
|
|
}, { onNodeStarted, onWorkflowStarted, onWorkflowFinished, isPublicAPI: !isInstalledApp, onNodeFinished, onIterationStart, onIterationNext, onIterationFinish, onTextChunk, onTextReplace })
|
|
}
|
|
|
|
export const fetchAppInfo = async () => {
|
|
return get('/site') as Promise<AppData>
|
|
}
|
|
|
|
export const fetchConversations = async (isInstalledApp: boolean, installedAppId = '', last_id?: string, pinned?: boolean, limit?: number) => {
|
|
return getAction('get', isInstalledApp)(getUrl('conversations', isInstalledApp, installedAppId), { params: { ...{ limit: limit || 20 }, ...(last_id ? { last_id } : {}), ...(pinned !== undefined ? { pinned } : {}) } }) as Promise<AppConversationData>
|
|
}
|
|
|
|
export const pinConversation = async (isInstalledApp: boolean, installedAppId = '', id: string) => {
|
|
return getAction('patch', isInstalledApp)(getUrl(`conversations/${id}/pin`, isInstalledApp, installedAppId))
|
|
}
|
|
|
|
export const unpinConversation = async (isInstalledApp: boolean, installedAppId = '', id: string) => {
|
|
return getAction('patch', isInstalledApp)(getUrl(`conversations/${id}/unpin`, isInstalledApp, installedAppId))
|
|
}
|
|
|
|
export const delConversation = async (isInstalledApp: boolean, installedAppId = '', id: string) => {
|
|
return getAction('del', isInstalledApp)(getUrl(`conversations/${id}`, isInstalledApp, installedAppId))
|
|
}
|
|
|
|
export const renameConversation = async (isInstalledApp: boolean, installedAppId = '', id: string, name: string) => {
|
|
return getAction('post', isInstalledApp)(getUrl(`conversations/${id}/name`, isInstalledApp, installedAppId), { body: { name } })
|
|
}
|
|
|
|
export const generationConversationName = async (isInstalledApp: boolean, installedAppId = '', id: string) => {
|
|
return getAction('post', isInstalledApp)(getUrl(`conversations/${id}/name`, isInstalledApp, installedAppId), { body: { auto_generate: true } }) as Promise<ConversationItem>
|
|
}
|
|
|
|
export const fetchChatList = async (conversationId: string, isInstalledApp: boolean, installedAppId = '') => {
|
|
return getAction('get', isInstalledApp)(getUrl('messages', isInstalledApp, installedAppId), { params: { conversation_id: conversationId, limit: 20, last_id: '' } }) as any
|
|
}
|
|
|
|
// Abandoned API interface
|
|
// export const fetchAppVariables = async () => {
|
|
// return get(`variables`)
|
|
// }
|
|
|
|
// init value. wait for server update
|
|
export const fetchAppParams = async (isInstalledApp: boolean, installedAppId = '') => {
|
|
return (getAction('get', isInstalledApp))(getUrl('parameters', isInstalledApp, installedAppId)) as Promise<ChatConfig>
|
|
}
|
|
|
|
export const fetchSystemFeatures = async () => {
|
|
return (getAction('get', false))(getUrl('system-features', false, '')) as Promise<SystemFeatures>
|
|
}
|
|
|
|
export const fetchWebSAMLSSOUrl = async (appCode: string, redirectUrl: string) => {
|
|
return (getAction('get', false))(getUrl('/enterprise/sso/saml/login', false, ''), {
|
|
params: {
|
|
app_code: appCode,
|
|
redirect_url: redirectUrl,
|
|
},
|
|
}) as Promise<{ url: string }>
|
|
}
|
|
|
|
export const fetchWebOIDCSSOUrl = async (appCode: string, redirectUrl: string) => {
|
|
return (getAction('get', false))(getUrl('/enterprise/sso/oidc/login', false, ''), {
|
|
params: {
|
|
app_code: appCode,
|
|
redirect_url: redirectUrl,
|
|
},
|
|
|
|
}) as Promise<{ url: string }>
|
|
}
|
|
|
|
export const fetchWebOAuth2SSOUrl = async (appCode: string, redirectUrl: string) => {
|
|
return (getAction('get', false))(getUrl('/enterprise/sso/oauth2/login', false, ''), {
|
|
params: {
|
|
app_code: appCode,
|
|
redirect_url: redirectUrl,
|
|
},
|
|
}) as Promise<{ url: string }>
|
|
}
|
|
|
|
export const fetchAppMeta = async (isInstalledApp: boolean, installedAppId = '') => {
|
|
return (getAction('get', isInstalledApp))(getUrl('meta', isInstalledApp, installedAppId)) as Promise<AppMeta>
|
|
}
|
|
|
|
export const updateFeedback = async ({ url, body }: { url: string; body: Feedbacktype }, isInstalledApp: boolean, installedAppId = '') => {
|
|
return (getAction('post', isInstalledApp))(getUrl(url, isInstalledApp, installedAppId), { body })
|
|
}
|
|
|
|
export const fetchMoreLikeThis = async (messageId: string, isInstalledApp: boolean, installedAppId = '') => {
|
|
return (getAction('get', isInstalledApp))(getUrl(`/messages/${messageId}/more-like-this`, isInstalledApp, installedAppId), {
|
|
params: {
|
|
response_mode: 'blocking',
|
|
},
|
|
})
|
|
}
|
|
|
|
export const saveMessage = (messageId: string, isInstalledApp: boolean, installedAppId = '') => {
|
|
return (getAction('post', isInstalledApp))(getUrl('/saved-messages', isInstalledApp, installedAppId), { body: { message_id: messageId } })
|
|
}
|
|
|
|
export const fetchSavedMessage = async (isInstalledApp: boolean, installedAppId = '') => {
|
|
return (getAction('get', isInstalledApp))(getUrl('/saved-messages', isInstalledApp, installedAppId))
|
|
}
|
|
|
|
export const removeMessage = (messageId: string, isInstalledApp: boolean, installedAppId = '') => {
|
|
return (getAction('del', isInstalledApp))(getUrl(`/saved-messages/${messageId}`, isInstalledApp, installedAppId))
|
|
}
|
|
|
|
export const fetchSuggestedQuestions = (messageId: string, isInstalledApp: boolean, installedAppId = '') => {
|
|
return (getAction('get', isInstalledApp))(getUrl(`/messages/${messageId}/suggested-questions`, isInstalledApp, installedAppId))
|
|
}
|
|
|
|
export const audioToText = (url: string, isPublicAPI: boolean, body: FormData) => {
|
|
return (getAction('post', !isPublicAPI))(url, { body }, { bodyStringify: false, deleteContentType: true }) as Promise<{ text: string }>
|
|
}
|
|
|
|
export const textToAudio = (url: string, isPublicAPI: boolean, body: FormData) => {
|
|
return (getAction('post', !isPublicAPI))(url, { body }, { bodyStringify: false, deleteContentType: true }) as Promise<{ data: string }>
|
|
}
|
|
|
|
export const fetchAccessToken = async (appCode: string) => {
|
|
const headers = new Headers()
|
|
headers.append('X-App-Code', appCode)
|
|
return get('/passport', { headers }) as Promise<{ access_token: string }>
|
|
}
|