From 3d3fa2c75ac3ed9439b94ae2bdd034403839af34 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Thu, 30 Oct 2025 10:12:43 +0800
Subject: [PATCH] feat(ai): 添加上下文生成工作流功能
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java | 9 ++-------
1 files changed, 2 insertions(+), 7 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java
index 37f0506..38ac317 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java
@@ -67,6 +67,7 @@
this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.REFERENCE_ANSWER.getCode(),null,createTime);
this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.KEY_KNOWLEDGE.getCode(),null,createTime);
this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.REPORT.getCode(),null,createTime);
+ this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.CONTEXT_TALK.getCode(),null,createTime);
this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.ADMIN_QUESTION.getCode(),null,createTime);
String categoryId = "";
@@ -78,15 +79,9 @@
String knowledgeId = "";
try {
- Client client = KnowledgeBaseUtil.createClient();
- String workspaceId = KnowledgeBaseUtil.WORKSPACE_ID;
String name = entity.getName();
- String sourceType = "DATA_CENTER_FILE";
- String structureType = "unstructured";
- String sinkType = "DEFAULT";
String fileId = "file_b7d0a5b2df1745b9bfb7402f5fe1054f_12629554";
- CreateIndexResponse indexResponse = KnowledgeBaseUtil.createIndex(client, workspaceId, fileId, name, structureType, sourceType, sinkType);
- knowledgeId = indexResponse.getBody().getData().getId();
+ knowledgeId = KnowledgeBaseUtil.createKnowledgeBase(fileId, name);
} catch (Exception e) {
throw new FebsException("初始化知识库失败");
}
--
Gitblit v1.9.1