xiaoshu/src/ajax/api/message.js

import request from "../request";
const create = (token, agentId, dialogId) => {
    const ws = new WebSocket(`ws://smartai.com:8000/api/chat/ws/${agentId}/${dialogId}?token=${token}`);
    // 当 WebSocket 打开时...
    ws.onopen = function (event) {
        console.log("WebSocket 连接已打开");
    };

    // 当 WebSocket 关闭时...
    ws.onclose = function (event) {
        console.log(event.code);
    };

    // 当有错误发生时...
    ws.onerror = function (error) {
        console.error("WebSocket 出错: ", error);
    };
    return ws;
};

/**
 * 新建对话
 * @returns {Promise<Object>}
 */
const createDialog = (id) => {
    return request.get(`/api/agent/get-chat-id/${id}`);
};

/**
 * 获取对话
 * @returns {Promise<Object>}
 */
const getDialogs = (id) => {
    return request.get(`/api/agent/${id}`);
};

export default {
    create,
    createDialog,
    getDialogs
};