debug.ts 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. import type { IOnCompleted, IOnData, IOnError } from './base'
  2. import { get, post, ssePost } from './base'
  3. export const sendChatMessage = async (appId: string, body: Record<string, any>, { onData, onCompleted, onError, getAbortController }: {
  4. onData: IOnData
  5. onCompleted: IOnCompleted
  6. onError: IOnError
  7. getAbortController?: (abortController: AbortController) => void
  8. }) => {
  9. return ssePost(`apps/${appId}/chat-messages`, {
  10. body: {
  11. ...body,
  12. response_mode: 'streaming',
  13. },
  14. }, { onData, onCompleted, onError, getAbortController })
  15. }
  16. export const stopChatMessageResponding = async (appId: string, taskId: string) => {
  17. return post(`apps/${appId}/chat-messages/${taskId}/stop`)
  18. }
  19. export const sendCompletionMessage = async (appId: string, body: Record<string, any>, { onData, onCompleted, onError }: {
  20. onData: IOnData
  21. onCompleted: IOnCompleted
  22. onError: IOnError
  23. }) => {
  24. return ssePost(`apps/${appId}/completion-messages`, {
  25. body: {
  26. ...body,
  27. response_mode: 'streaming',
  28. },
  29. }, { onData, onCompleted, onError })
  30. }
  31. export const fetchSuggestedQuestions = (appId: string, messageId: string) => {
  32. return get(`apps/${appId}/chat-messages/${messageId}/suggested-questions`)
  33. }
  34. export const fetchConvesationMessages = (appId: string, conversation_id: string) => {
  35. return get(`apps/${appId}/chat-messages`, {
  36. params: {
  37. conversation_id,
  38. },
  39. })
  40. }
  41. export const generateRule = (body: Record<string, any>) => {
  42. return post('/rule-generate', {
  43. body,
  44. })
  45. }