54 lines
1.5 KiB
JavaScript
54 lines
1.5 KiB
JavaScript
|
import { apiAdminGet, apiAdminPost } from './base'
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 思维导图API模块
|
|||
|
|
* 提供思维导图相关的接口功能
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
// =============================================================================
|
|||
|
|
// === 知识库管理 ===
|
|||
|
|
// =============================================================================
|
|||
|
|
|
|||
|
|
export const mindmapApi = {
|
|||
|
|
/**
|
|||
|
|
* 获取所有知识库概览（用于选择）
|
|||
|
|
* @returns {Promise} - 知识库列表
|
|||
|
|
*/
|
|||
|
|
getDatabases: async () => {
|
|||
|
|
return apiAdminGet('/api/mindmap/databases')
|
|||
|
|
},
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取指定知识库的文件列表
|
|||
|
|
* @param {string} dbId - 知识库ID
|
|||
|
|
* @returns {Promise} - 文件列表
|
|||
|
|
*/
|
|||
|
|
getDatabaseFiles: async (dbId) => {
|
|||
|
|
return apiAdminGet(`/api/mindmap/databases/${dbId}/files`)
|
|||
|
|
},
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* AI生成思维导图
|
|||
|
|
* @param {string} dbId - 知识库ID
|
|||
|
|
* @param {Array<string>} fileIds - 选择的文件ID列表（为空则使用所有文件）
|
|||
|
|
* @param {string} userPrompt - 用户自定义提示词
|
|||
|
|
* @returns {Promise} - 思维导图数据
|
|||
|
|
*/
|
|||
|
|
generateMindmap: async (dbId, fileIds = [], userPrompt = '') => {
|
|||
|
|
return apiAdminPost('/api/mindmap/generate', {
|
|||
|
|
db_id: dbId,
|
|||
|
|
file_ids: fileIds,
|
|||
|
|
user_prompt: userPrompt
|
|||
|
|
})
|
|||
|
|
},
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取知识库的思维导图
|
|||
|
|
* @param {string} dbId - 知识库ID
|
|||
|
|
* @returns {Promise} - 思维导图数据
|
|||
|
|
*/
|
|||
|
|
getByDatabase: async (dbId) => {
|
|||
|
|
return apiAdminGet(`/api/mindmap/database/${dbId}`)
|
|||
|
|
}
|
|||
|
|
}
|