diff --git a/ui/src/api/knowledge/knowledge.ts b/ui/src/api/knowledge/knowledge.ts index dc7b6293e..2054d2d67 100644 --- a/ui/src/api/knowledge/knowledge.ts +++ b/ui/src/api/knowledge/knowledge.ts @@ -39,38 +39,6 @@ const getKnowledgeList: ( return get(`${prefix}/knowledge/${page.current_page}/${page.page_size}`, param, loading) } -/** - * 获取全部知识库 - * @param 参数 - */ -const getAllKnowledge: (loading?: Ref) => Promise> = (loading) => { - return get(`${prefix}`, undefined, loading) -} - -/** - * 同步知识库 - * @param 参数 knowledge_id - * @query 参数 sync_type // 同步类型->replace:替换同步,complete:完整同步 - */ -const putSyncWebKnowledge: ( - knowledge_id: string, - sync_type: string, - loading?: Ref, -) => Promise> = (knowledge_id, sync_type, loading) => { - return put(`${prefix}/knowledge/${knowledge_id}/sync`, undefined, { sync_type }, loading) -} - -/** - * 向量化知识库 - * @param 参数 knowledge_id - */ -const putReEmbeddingKnowledge: ( - knowledge_id: string, - loading?: Ref, -) => Promise> = (knowledge_id, loading) => { - return put(`${prefix}/knowledge/${knowledge_id}/embedding`, undefined, undefined, loading) -} - /** * 知识库详情 * @param 参数 knowledge_id @@ -82,41 +50,6 @@ const getKnowledgeDetail: (knowledge_id: string, loading?: Ref) => Prom return get(`${prefix}/knowledge/${knowledge_id}`, undefined, loading) } -/** - * 创建知识库 - * @param 参数 - * { - "name": "string", - "folder_id": "string", - "desc": "string", - "embedding": "string" - } - */ -const postKnowledge: (data: knowledgeData, loading?: Ref) => Promise> = ( - data, - loading, -) => { - return post(`${prefix}/knowledge/base`, data, undefined, loading, 1000 * 60 * 5) -} - -/** - * 创建Web知识库 - * @param 参数 - * { - "name": "string", - "folder_id": "string", - "desc": "string", - "embedding": "string", - "source_url": "string", - "selector": "string" - } - */ -const postWebKnowledge: (data: any, loading?: Ref) => Promise> = ( - data, - loading, -) => { - return post(`${prefix}/knowledge/web`, data, undefined, loading) -} /** * 修改知识库信息 * @param 参数 @@ -135,18 +68,25 @@ const putKnowledge: ( } /** - * 命中测试列表 - * @param knowledge_id - * @param loading - * @query { query_text: string, top_number: number, similarity: number } - * @returns + * 删除知识库 + * @param 参数 knowledge_id */ -const getKnowledgeHitTest: ( +const delKnowledge: (knowledge_id: String, loading?: Ref) => Promise> = ( + knowledge_id, + loading, +) => { + return del(`${prefix}/${knowledge_id}`, undefined, {}, loading) +} + +/** + * 向量化知识库 + * @param 参数 knowledge_id + */ +const putReEmbeddingKnowledge: ( knowledge_id: string, - data: any, loading?: Ref, -) => Promise>> = (knowledge_id, data, loading) => { - return get(`${prefix}/${knowledge_id}/hit_test`, data, loading) +) => Promise> = (knowledge_id, loading) => { + return put(`${prefix}/knowledge/${knowledge_id}/embedding`, undefined, undefined, loading) } /** @@ -160,7 +100,12 @@ const exportKnowledge: ( knowledge_id: string, loading?: Ref, ) => Promise = (knowledge_name, knowledge_id, loading) => { - return exportExcel(knowledge_name + '.xlsx', `${prefix}/${knowledge_id}/knowledge/${knowledge_id}/export`, undefined, loading) + return exportExcel( + knowledge_name + '.xlsx', + `${prefix}/${knowledge_id}/knowledge/${knowledge_id}/export`, + undefined, + loading, + ) } /** *导出Zip知识库 @@ -182,6 +127,80 @@ const exportZipKnowledge: ( ) } +/** + * 生成关联问题 + * @param knowledge_id 知识库id + * @param data + * @param loading + * @returns + */ +const putGenerateRelated: ( + knowledge_id: string, + data: any, + loading?: Ref, +) => Promise>> = (knowledge_id, data, loading) => { + return put(`${prefix}/${knowledge_id}/generate_related`, data, null, loading) +} + +/** + * 命中测试列表 + * @param knowledge_id + * @param loading + * @query { query_text: string, top_number: number, similarity: number } + * @returns + */ +const getKnowledgeHitTest: ( + knowledge_id: string, + data: any, + loading?: Ref, +) => Promise>> = (knowledge_id, data, loading) => { + return get(`${prefix}/${knowledge_id}/hit_test`, data, loading) +} + +/** + * 同步知识库 + * @param 参数 knowledge_id + * @query 参数 sync_type // 同步类型->replace:替换同步,complete:完整同步 + */ +const putSyncWebKnowledge: ( + knowledge_id: string, + sync_type: string, + loading?: Ref, +) => Promise> = (knowledge_id, sync_type, loading) => { + return put(`${prefix}/knowledge/${knowledge_id}/sync`, undefined, { sync_type }, loading) +} + +/** + * 创建知识库 + * @param 参数 + * { + "name": "string", + "folder_id": "string", + "desc": "string", + "embedding": "string" + } + */ +const postKnowledge: (data: knowledgeData, loading?: Ref) => Promise> = ( + data, + loading, +) => { + return post(`${prefix}/knowledge/base`, data, undefined, loading, 1000 * 60 * 5) +} + +/** + * 获取当前用户可使用的向量化模型列表 + * @param application_id + * @param loading + * @query { query_text: string, top_number: number, similarity: number } + * @returns + */ +const getKnowledgeEmdeddingModel: ( + knowledge_id: string, + loading?: Ref, +) => Promise>> = (knowledge_id, loading) => { + return get(`${prefix}/${knowledge_id}/emdedding_model`, loading) +} + /** * 获取当前用户可使用的模型列表 * @param application_id @@ -195,6 +214,34 @@ const getKnowledgeModel: ( ) => Promise>> = (knowledge_id, loading) => { return get(`${prefix}/${knowledge_id}/model`, loading) } + +/** + * 创建Web知识库 + * @param 参数 + * { + "name": "string", + "folder_id": "string", + "desc": "string", + "embedding": "string", + "source_url": "string", + "selector": "string" + } + */ +const postWebKnowledge: (data: any, loading?: Ref) => Promise> = ( + data, + loading, +) => { + return post(`${prefix}/knowledge/web`, data, undefined, loading) +} + +/** + * 获取全部知识库 + * @param 参数 + */ +const getAllKnowledge: (loading?: Ref) => Promise> = (loading) => { + return get(`${prefix}`, undefined, loading) +} + /** * 获取飞书文档列表 * @param knowledge_id @@ -218,48 +265,24 @@ const importLarkDocument: ( ) => Promise>> = (knowledge_id, data, loading) => { return post(`${prefix}/lark/${knowledge_id}/import`, data, null, loading) } -/** - * 生成关联问题 - * @param knowledge_id 知识库id - * @param data - * @param loading - * @returns - */ -const generateRelated: ( - knowledge_id: string, - data: any, - loading?: Ref, -) => Promise>> = (knowledge_id, data, loading) => { - return put(`${prefix}/${knowledge_id}/generate_related`, data, null, loading) -} - -/** - * 删除知识库 - * @param 参数 knowledge_id - */ -const delKnowledge: (knowledge_id: String, loading?: Ref) => Promise> = ( - knowledge_id, - loading -) => { - return del(`${prefix}/${knowledge_id}`, undefined, {}, loading) -} export default { getKnowledgeByFolder, getKnowledgeList, - putReEmbeddingKnowledge, - putSyncWebKnowledge, getKnowledgeDetail, - postKnowledge, - postWebKnowledge, putKnowledge, - getKnowledgeHitTest, + delKnowledge, + putReEmbeddingKnowledge, exportKnowledge, exportZipKnowledge, + putGenerateRelated, + getKnowledgeHitTest, + putSyncWebKnowledge, + postKnowledge, getKnowledgeModel, + postWebKnowledge, + getLarkDocumentList, importLarkDocument, - generateRelated, - delKnowledge, - getAllKnowledge + getAllKnowledge, } diff --git a/ui/src/components/generate-related-dialog/index.vue b/ui/src/components/generate-related-dialog/index.vue index a9bccccd9..0d3a41747 100644 --- a/ui/src/components/generate-related-dialog/index.vue +++ b/ui/src/components/generate-related-dialog/index.vue @@ -176,7 +176,7 @@ const submitHandle = async (formEl: FormInstance) => { ...form.value, state_list: stateMap[state.value] } - knowledgeApi.generateRelated(id ? id : knowledgeId.value, data, loading).then(() => { + knowledgeApi.putGenerateRelated(id ? id : knowledgeId.value, data, loading).then(() => { MsgSuccess(t('views.document.generateQuestion.successMessage')) dialogVisible.value = false })