85 lines
2.1 KiB
JavaScript
85 lines
2.1 KiB
JavaScript
/**
|
||
* AI 智能体 API
|
||
*/
|
||
import request from '@/api/http'
|
||
import { fetchEventSource } from '@microsoft/fetch-event-source'
|
||
import tokenManager from '@gold/utils/token-manager'
|
||
import { API_BASE } from '@gold/config/api'
|
||
|
||
const BASE_URL = `${API_BASE.APP_TIK}`
|
||
|
||
|
||
/**
|
||
* 获取启用的智能体列表
|
||
*/
|
||
export function getAgentList() {
|
||
return request({
|
||
url: `${BASE_URL}/agent/list`,
|
||
method: 'get'
|
||
})
|
||
}
|
||
|
||
/**
|
||
* 流式对话（SSE）
|
||
* @param {Object} options - 请求配置
|
||
* @param {number} options.agentId - 智能体ID
|
||
* @param {string} options.content - 用户输入内容
|
||
* @param {string} [options.conversationId] - 会话ID（可选，首次对话不传）
|
||
* @param {AbortController} [options.ctrl] - 取消控制器
|
||
* @param {Function} options.onMessage - 消息回调
|
||
* @param {Function} [options.onError] - 错误回调
|
||
* @param {Function} [options.onClose] - 关闭回调
|
||
*/
|
||
export async function sendChatStream(options) {
|
||
const {
|
||
agentId,
|
||
content,
|
||
conversationId,
|
||
ctrl,
|
||
onMessage,
|
||
onError,
|
||
onClose
|
||
} = options || {}
|
||
|
||
const token = tokenManager.getAccessToken()
|
||
|
||
return fetchEventSource(`${BASE_URL}/dify/chat/stream`, {
|
||
method: 'post',
|
||
headers: {
|
||
'Content-Type': 'application/json',
|
||
'Authorization': `Bearer ${token}`,
|
||
'tenant-id': import.meta.env?.VITE_TENANT_ID
|
||
},
|
||
openWhenHidden: true,
|
||
body: JSON.stringify({
|
||
agentId,
|
||
content,
|
||
conversationId
|
||
}),
|
||
onmessage: (event) => {
|
||
if (typeof onMessage === 'function') {
|
||
try {
|
||
const data = JSON.parse(event.data)
|
||
// 解析 CommonResult 包装
|
||
const result = data.code === 0 ? data.data : data
|
||
onMessage(result)
|
||
} catch (e) {
|
||
console.error('解析 SSE 数据失败:', e)
|
||
}
|
||
}
|
||
},
|
||
onerror: (err) => {
|
||
if (typeof onError === 'function') {
|
||
onError(err)
|
||
}
|
||
throw err // 不重试
|
||
},
|
||
onclose: () => {
|
||
if (typeof onClose === 'function') {
|
||
onClose()
|
||
}
|
||
},
|
||
signal: ctrl ? ctrl.signal : undefined
|
||
})
|
||
}
|