import request from '../../utils/request';

export async function getAiModels() {
    const res = await request.get('/ai/model/list');
    if (res.data.code === 200) {
        return res.data.data || [];
    }
    return Promise.reject(new Error(res.data.msg || 'Load models failed'));
}

export async function getAiSessions() {
    const res = await request.get('/ai/session/list');
    if (res.data.code === 200) {
        return res.data.data || [];
    }
    return Promise.reject(new Error(res.data.msg || 'Load sessions failed'));
}

export async function createAiSession(payload = {}) {
    const res = await request.post('/ai/session/create', payload);
    if (res.data.code === 200) {
        return res.data.data;
    }
    return Promise.reject(new Error(res.data.msg || 'Create session failed'));
}

export async function removeAiSession(sessionId) {
    const res = await request.post(`/ai/session/remove/${sessionId}`);
    return res.data;
}

export async function getAiMessages(sessionId) {
    const res = await request.get(`/ai/session/${sessionId}/messages`);
    if (res.data.code === 200) {
        return res.data.data || [];
    }
    return Promise.reject(new Error(res.data.msg || 'Load messages failed'));
}

export async function stopAiChat(payload) {
    const res = await request.post('/ai/chat/stop', payload);
    return res.data;
}