1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
| import { get, post, ssePost } from './base'
| import type { IOnCompleted, IOnData, IOnError, IOnFile, IOnMessageEnd, IOnMessageReplace, IOnThought } from './base'
| import type { ChatPromptConfig, CompletionPromptConfig } from '@/models/debug'
| import type { ModelModeType } from '@/types/app'
| import type { ModelParameterRule } from '@/app/components/header/account-setting/model-provider-page/declarations'
| export type AutomaticRes = {
| prompt: string
| variables: string[]
| opening_statement: string
| error?: string
| }
| export type CodeGenRes = {
| code: string
| language: string[]
| error?: string
| }
|
| export const sendChatMessage = async (appId: string, body: Record<string, any>, { onData, onCompleted, onThought, onFile, onError, getAbortController, onMessageEnd, onMessageReplace }: {
| onData: IOnData
| onCompleted: IOnCompleted
| onFile: IOnFile
| onThought: IOnThought
| onMessageEnd: IOnMessageEnd
| onMessageReplace: IOnMessageReplace
| onError: IOnError
| getAbortController?: (abortController: AbortController) => void
| }) => {
| return ssePost(`apps/${appId}/chat-messages`, {
| body: {
| ...body,
| response_mode: 'streaming',
| },
| }, { onData, onCompleted, onThought, onFile, onError, getAbortController, onMessageEnd, onMessageReplace })
| }
|
| export const stopChatMessageResponding = async (appId: string, taskId: string) => {
| return post(`apps/${appId}/chat-messages/${taskId}/stop`)
| }
|
| export const sendCompletionMessage = async (appId: string, body: Record<string, any>, { onData, onCompleted, onError, onMessageReplace }: {
| onData: IOnData
| onCompleted: IOnCompleted
| onError: IOnError
| onMessageReplace: IOnMessageReplace
| }) => {
| return ssePost(`apps/${appId}/completion-messages`, {
| body: {
| ...body,
| response_mode: 'streaming',
| },
| }, { onData, onCompleted, onError, onMessageReplace })
| }
|
| export const fetchSuggestedQuestions = (appId: string, messageId: string, getAbortController?: any) => {
| return get(
| `apps/${appId}/chat-messages/${messageId}/suggested-questions`,
| {},
| {
| getAbortController,
| },
| )
| }
|
| export const fetchConversationMessages = (appId: string, conversation_id: string, getAbortController?: any) => {
| return get(`apps/${appId}/chat-messages`, {
| params: {
| conversation_id,
| },
| }, {
| getAbortController,
| })
| }
|
| export const generateRule = (body: Record<string, any>) => {
| return post<AutomaticRes>('/rule-generate', {
| body,
| })
| }
| export const generateRuleCode = (body: Record<string, any>) => {
| return post<CodeGenRes>('/rule-code-generate', {
| body,
| })
| }
|
| export const fetchModelParams = (providerName: string, modelId: string) => {
| return get(`workspaces/current/model-providers/${providerName}/models/parameter-rules`, {
| params: {
| model: modelId,
| },
| }) as Promise<{ data: ModelParameterRule[] }>
| }
|
| export const fetchPromptTemplate = ({
| appMode,
| mode,
| modelName,
| hasSetDataSet,
| }: { appMode: string; mode: ModelModeType; modelName: string; hasSetDataSet: boolean }) => {
| return get<Promise<{ chat_prompt_config: ChatPromptConfig; completion_prompt_config: CompletionPromptConfig; stop: [] }>>('/app/prompt-templates', {
| params: {
| app_mode: appMode,
| model_mode: mode,
| model_name: modelName,
| has_context: hasSetDataSet,
| },
| })
| }
|
| export const fetchTextGenerationMessage = ({
| appId,
| messageId,
| }: { appId: string; messageId: string }) => {
| return get<Promise<any>>(`/apps/${appId}/messages/${messageId}`)
| }
|
|
