mirror of
https://github.com/xinnan-tech/xiaozhi-esp32-server.git
synced 2026-07-22 07:03:53 +08:00
Merge pull request #2495 from xinnan-tech/dependabot/pip/main/xiaozhi-server/httpx-0.28.1
build(deps): bump httpx from 0.27.2 to 0.28.1 in /main/xiaozhi-server
This commit is contained in:
hrz
@@ -4,7 +4,9 @@ FROM python:3.10-slim
|
||||
|
||||
# 安装系统依赖
|
||||
RUN apt-get update && \
|
||||
apt-get install -y --no-install-recommends libopus0 ffmpeg && \
|
||||
apt-get install -y --no-install-recommends libopus0 ffmpeg locales && \
|
||||
sed -i '/zh_CN.UTF-8/s/^# //g' /etc/locale.gen && \
|
||||
locale-gen && \
|
||||
apt-get clean && \
|
||||
rm -rf /var/lib/apt/lists/*
|
||||
|
||||
@@ -14,6 +16,12 @@ RUN pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/ && \
|
||||
pip config set global.timeout 120 && \
|
||||
pip config set install.retries 5
|
||||
|
||||
# 设置环境变量以确保正确的字符编码
|
||||
ENV LANG=zh_CN.UTF-8 \
|
||||
LC_ALL=zh_CN.UTF-8 \
|
||||
LANGUAGE=zh_CN:zh \
|
||||
PYTHONIOENCODING=utf-8
|
||||
|
||||
WORKDIR /opt/xiaozhi-esp32-server
|
||||
|
||||
# 复制requirements.txt
|
||||
|
||||
+1
-1
@@ -80,7 +80,7 @@ xiaozhi-server
|
||||
打开命令行工具，使用`终端`或`命令行`工具 进入到你的`xiaozhi-server`，执行以下命令
|
||||
|
||||
```
|
||||
docker-compose up -d
|
||||
docker compose up -d
|
||||
```
|
||||
|
||||
执行完后，再执行以下命令，查看日志信息。
|
||||
|
||||
@@ -17,5 +17,5 @@ docker build -t xiaozhi-esp32-server:web_latest -f ./Dockerfile-web .
|
||||
# 编译完成后，可以使用docker-compose启动项目
|
||||
# docker-compose.yml你需要修改成自己编译的镜像版本
|
||||
cd main/xiaozhi-server
|
||||
docker-compose up -d
|
||||
docker compose up -d
|
||||
```
|
||||
|
||||
@@ -151,6 +151,11 @@ public interface Constant {
|
||||
*/
|
||||
String VOICE_CLONE_HUOSHAN_DOUBLE_STREAM = "huoshan_double_stream";
|
||||
|
||||
/**
|
||||
* RAG配置类型
|
||||
*/
|
||||
String RAG_CONFIG_TYPE = "RAG";
|
||||
|
||||
enum SysBaseParam {
|
||||
/**
|
||||
* ICP备案号
|
||||
@@ -294,7 +299,7 @@ public interface Constant {
|
||||
/**
|
||||
* 版本号
|
||||
*/
|
||||
public static final String VERSION = "0.8.6";
|
||||
public static final String VERSION = "0.8.7";
|
||||
|
||||
/**
|
||||
* 无效固件URL
|
||||
|
||||
@@ -119,10 +119,6 @@ public interface ErrorCode {
|
||||
int VOICEPRINT_UNREGISTER_PROCESS_ERROR = 10090; // 声纹注销处理失败
|
||||
int VOICEPRINT_IDENTIFY_REQUEST_ERROR = 10091; // 声纹识别请求失败
|
||||
|
||||
// 设备相关错误码
|
||||
int MAC_ADDRESS_ALREADY_EXISTS = 10161; // Mac地址已存在
|
||||
// 模型相关错误码
|
||||
int MODEL_PROVIDER_NOT_EXIST = 10162; // 供应器不存在
|
||||
int LLM_NOT_EXIST = 10092; // 设置的LLM不存在
|
||||
int MODEL_REFERENCED_BY_AGENT = 10093; // 该模型配置已被智能体引用，无法删除
|
||||
int LLM_REFERENCED_BY_INTENT = 10094; // 该LLM模型已被意图识别配置引用，无法删除
|
||||
@@ -198,4 +194,19 @@ public interface ErrorCode {
|
||||
int VOICE_CLONE_PREFIX = 10158; // 复刻音色前缀
|
||||
int VOICE_ID_ALREADY_EXISTS = 10159; // 音色ID已存在
|
||||
int VOICE_CLONE_HUOSHAN_VOICE_ID_ERROR = 10160; // 火山引擎音色ID格式错误
|
||||
|
||||
// 设备相关错误码
|
||||
int MAC_ADDRESS_ALREADY_EXISTS = 10161; // Mac地址已存在
|
||||
// 模型相关错误码
|
||||
int MODEL_PROVIDER_NOT_EXIST = 10162; // 供应器不存在
|
||||
|
||||
// 知识库相关错误码
|
||||
int Knowledge_Base_RECORD_NOT_EXISTS = 10163; // 知识库记录不存在
|
||||
int RAG_CONFIG_NOT_FOUND = 10164; // RAG配置未找到
|
||||
int RAG_CONFIG_TYPE_ERROR = 10165; // RAG配置类型错误
|
||||
int RAG_DEFAULT_CONFIG_NOT_FOUND = 10166; // 默认RAG配置未找到
|
||||
int RAG_API_ERROR = 10167; // RAG配置缺少必要参数
|
||||
int UPLOAD_FILE_ERROR = 10168; // 上传文件失败
|
||||
int NO_PERMISSION = 10169; // 没有权限
|
||||
int KNOWLEDGE_BASE_NAME_EXISTS = 10170; // 同名知识库已存在
|
||||
}
|
||||
|
||||
@@ -32,13 +32,23 @@ public class FieldMetaObjectHandler implements MetaObjectHandler {
|
||||
|
||||
// 创建者
|
||||
strictInsertFill(metaObject, CREATOR, Long.class, user.getId());
|
||||
// 创建时间
|
||||
strictInsertFill(metaObject, CREATE_DATE, Date.class, date);
|
||||
// 创建时间 - 支持createDate和createdAt两种字段名
|
||||
if (metaObject.hasSetter(CREATE_DATE)) {
|
||||
strictInsertFill(metaObject, CREATE_DATE, Date.class, date);
|
||||
}
|
||||
if (metaObject.hasSetter("createdAt")) {
|
||||
strictInsertFill(metaObject, "createdAt", Date.class, date);
|
||||
}
|
||||
|
||||
// 更新者
|
||||
strictInsertFill(metaObject, UPDATER, Long.class, user.getId());
|
||||
// 更新时间
|
||||
strictInsertFill(metaObject, UPDATE_DATE, Date.class, date);
|
||||
// 更新时间 - 支持updateDate和updatedAt两种字段名
|
||||
if (metaObject.hasSetter(UPDATE_DATE)) {
|
||||
strictInsertFill(metaObject, UPDATE_DATE, Date.class, date);
|
||||
}
|
||||
if (metaObject.hasSetter("updatedAt")) {
|
||||
strictInsertFill(metaObject, "updatedAt", Date.class, date);
|
||||
}
|
||||
|
||||
// 数据标识
|
||||
strictInsertFill(metaObject, DATA_OPERATION, String.class, Constant.DataOperation.INSERT.getValue());
|
||||
@@ -46,10 +56,17 @@ public class FieldMetaObjectHandler implements MetaObjectHandler {
|
||||
|
||||
@Override
|
||||
public void updateFill(MetaObject metaObject) {
|
||||
Date date = new Date();
|
||||
|
||||
// 更新者
|
||||
strictUpdateFill(metaObject, UPDATER, Long.class, SecurityUser.getUserId());
|
||||
// 更新时间
|
||||
strictUpdateFill(metaObject, UPDATE_DATE, Date.class, new Date());
|
||||
// 更新时间 - 支持updateDate和updatedAt两种字段名
|
||||
if (metaObject.hasSetter(UPDATE_DATE)) {
|
||||
strictUpdateFill(metaObject, UPDATE_DATE, Date.class, date);
|
||||
}
|
||||
if (metaObject.hasSetter("updatedAt")) {
|
||||
strictUpdateFill(metaObject, "updatedAt", Date.class, date);
|
||||
}
|
||||
|
||||
// 数据标识
|
||||
strictInsertFill(metaObject, DATA_OPERATION, String.class, Constant.DataOperation.UPDATE.getValue());
|
||||
|
||||
@@ -152,4 +152,11 @@ public class RedisKeys {
|
||||
public static String getVoiceCloneAudioIdKey(String uuid) {
|
||||
return "voiceClone:audio:id:" + uuid;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取知识库缓存key
|
||||
*/
|
||||
public static String getKnowledgeBaseCacheKey(String datasetId) {
|
||||
return "knowledge:base:" + datasetId;
|
||||
}
|
||||
}
|
||||
|
||||
+69
-1
@@ -1,16 +1,25 @@
|
||||
package xiaozhi.modules.agent.service.impl;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
import org.apache.commons.lang3.StringUtils;
|
||||
import org.springframework.stereotype.Service;
|
||||
|
||||
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
|
||||
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
||||
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import xiaozhi.common.utils.JsonUtils;
|
||||
import xiaozhi.modules.agent.dao.AgentPluginMappingMapper;
|
||||
import xiaozhi.modules.agent.entity.AgentPluginMapping;
|
||||
import xiaozhi.modules.agent.service.AgentPluginMappingService;
|
||||
import xiaozhi.modules.knowledge.entity.KnowledgeBaseEntity;
|
||||
import xiaozhi.modules.knowledge.service.KnowledgeBaseService;
|
||||
import xiaozhi.modules.model.entity.ModelConfigEntity;
|
||||
import xiaozhi.modules.model.service.ModelConfigService;
|
||||
|
||||
/**
|
||||
* @description 针对表【ai_agent_plugin_mapping(Agent与插件的唯一映射表)】的数据库操作Service实现
|
||||
@@ -21,10 +30,69 @@ import xiaozhi.modules.agent.service.AgentPluginMappingService;
|
||||
public class AgentPluginMappingServiceImpl extends ServiceImpl<AgentPluginMappingMapper, AgentPluginMapping>
|
||||
implements AgentPluginMappingService {
|
||||
private final AgentPluginMappingMapper agentPluginMappingMapper;
|
||||
private final KnowledgeBaseService knowledgeBaseService;
|
||||
private final ModelConfigService modelConfigService;
|
||||
|
||||
@Override
|
||||
public List<AgentPluginMapping> agentPluginParamsByAgentId(String agentId) {
|
||||
return agentPluginMappingMapper.selectPluginsByAgentId(agentId);
|
||||
List<AgentPluginMapping> list = agentPluginMappingMapper.selectPluginsByAgentId(agentId);
|
||||
Map<String, List<KnowledgeBaseEntity>> knowledgeBaseMap = new HashMap<>();
|
||||
Map<String, ModelConfigEntity> modelConfigMap = new HashMap<>();
|
||||
for (int i = list.size() - 1; i >= 0; i--) {
|
||||
AgentPluginMapping mapping = list.get(i);
|
||||
if (StringUtils.isBlank(mapping.getProviderCode())) {
|
||||
// 查询知识库插件参数
|
||||
KnowledgeBaseEntity knowledgeBaseEntity = knowledgeBaseService.selectById(mapping.getPluginId());
|
||||
if (knowledgeBaseEntity == null) {
|
||||
list.remove(i);
|
||||
continue;
|
||||
}
|
||||
ModelConfigEntity modelConfigEntity = modelConfigService
|
||||
.getModelByIdFromCache(knowledgeBaseEntity.getRagModelId());
|
||||
if (modelConfigEntity == null) {
|
||||
list.remove(i);
|
||||
continue;
|
||||
}
|
||||
List<KnowledgeBaseEntity> knowledgeBaseList = knowledgeBaseMap.get(modelConfigEntity.getModelCode());
|
||||
if (knowledgeBaseList == null) {
|
||||
knowledgeBaseList = new ArrayList<>();
|
||||
}
|
||||
modelConfigMap.put(modelConfigEntity.getModelCode(), modelConfigEntity);
|
||||
knowledgeBaseList.add(knowledgeBaseEntity);
|
||||
knowledgeBaseMap.put(modelConfigEntity.getModelCode(), knowledgeBaseList);
|
||||
list.remove(i);
|
||||
}
|
||||
}
|
||||
if (knowledgeBaseMap.size() > 0) {
|
||||
for (String pluginCode : knowledgeBaseMap.keySet()) {
|
||||
List<KnowledgeBaseEntity> knowledgeBaseList = knowledgeBaseMap.get(pluginCode);
|
||||
if (knowledgeBaseList == null || knowledgeBaseList.isEmpty()) {
|
||||
continue;
|
||||
}
|
||||
|
||||
AgentPluginMapping agentPluginMapping = new AgentPluginMapping();
|
||||
agentPluginMapping.setAgentId(agentId);
|
||||
agentPluginMapping.setPluginId(pluginCode);
|
||||
agentPluginMapping.setProviderCode("search_from_" + pluginCode);
|
||||
agentPluginMapping.setId(Long.valueOf(list.size() + 1));
|
||||
|
||||
Map<String, Object> paramInfo = new HashMap<>(4);
|
||||
ModelConfigEntity modelConfigEntity = modelConfigMap.get(pluginCode);
|
||||
paramInfo.put("base_url", modelConfigEntity.getConfigJson().getStr("base_url"));
|
||||
paramInfo.put("api_key", modelConfigEntity.getConfigJson().getStr("api_key"));
|
||||
paramInfo.put("dataset_ids",
|
||||
knowledgeBaseList.stream().map(KnowledgeBaseEntity::getDatasetId).toList());
|
||||
|
||||
String description = "如果用户询问与【"
|
||||
+ String.join(",", knowledgeBaseList.stream().map(KnowledgeBaseEntity::getName).toList())
|
||||
+ "】相关的问题应调用本方法，用于查询：" + String.join(",",
|
||||
knowledgeBaseList.stream().map(KnowledgeBaseEntity::getDescription).toList());
|
||||
paramInfo.put("description", description);
|
||||
agentPluginMapping.setParamInfo(JsonUtils.toJsonString(paramInfo));
|
||||
list.add(agentPluginMapping);
|
||||
}
|
||||
}
|
||||
return list;
|
||||
}
|
||||
|
||||
@Override
|
||||
|
||||
+4
@@ -45,6 +45,10 @@ public class AgentTemplateServiceImpl extends ServiceImpl<AgentTemplateDao, Agen
|
||||
@Override
|
||||
public void updateDefaultTemplateModelId(String modelType, String modelId) {
|
||||
modelType = modelType.toUpperCase();
|
||||
// 如果是rag模型，不需要更新
|
||||
if (modelType.equals("RAG")) {
|
||||
return;
|
||||
}
|
||||
|
||||
UpdateWrapper<AgentTemplateEntity> wrapper = new UpdateWrapper<>();
|
||||
switch (modelType) {
|
||||
|
||||
+6
-2
@@ -91,6 +91,7 @@ public class ConfigServiceImpl implements ConfigService {
|
||||
null,
|
||||
null,
|
||||
null,
|
||||
null,
|
||||
result,
|
||||
isCache);
|
||||
|
||||
@@ -195,6 +196,7 @@ public class ConfigServiceImpl implements ConfigService {
|
||||
agent.getTtsModelId(),
|
||||
agent.getMemModelId(),
|
||||
agent.getIntentModelId(),
|
||||
null,
|
||||
result,
|
||||
true);
|
||||
|
||||
@@ -371,12 +373,14 @@ public class ConfigServiceImpl implements ConfigService {
|
||||
String ttsModelId,
|
||||
String memModelId,
|
||||
String intentModelId,
|
||||
String ragModelId,
|
||||
Map<String, Object> result,
|
||||
boolean isCache) {
|
||||
Map<String, String> selectedModule = new HashMap<>();
|
||||
|
||||
String[] modelTypes = { "VAD", "ASR", "TTS", "Memory", "Intent", "LLM", "VLLM" };
|
||||
String[] modelIds = { vadModelId, asrModelId, ttsModelId, memModelId, intentModelId, llmModelId, vllmModelId };
|
||||
String[] modelTypes = { "VAD", "ASR", "TTS", "Memory", "Intent", "LLM", "VLLM", "RAG" };
|
||||
String[] modelIds = { vadModelId, asrModelId, ttsModelId, memModelId, intentModelId, llmModelId, vllmModelId,
|
||||
ragModelId };
|
||||
String intentLLMModelId = null;
|
||||
String memLocalShortLLMModelId = null;
|
||||
|
||||
|
||||
+159
@@ -0,0 +1,159 @@
|
||||
package xiaozhi.modules.knowledge.controller;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
import org.apache.commons.lang3.StringUtils;
|
||||
import org.apache.shiro.authz.annotation.RequiresPermissions;
|
||||
import org.springframework.validation.annotation.Validated;
|
||||
import org.springframework.web.bind.annotation.DeleteMapping;
|
||||
import org.springframework.web.bind.annotation.GetMapping;
|
||||
import org.springframework.web.bind.annotation.PathVariable;
|
||||
import org.springframework.web.bind.annotation.PostMapping;
|
||||
import org.springframework.web.bind.annotation.PutMapping;
|
||||
import org.springframework.web.bind.annotation.RequestBody;
|
||||
import org.springframework.web.bind.annotation.RequestMapping;
|
||||
import org.springframework.web.bind.annotation.RequestParam;
|
||||
import org.springframework.web.bind.annotation.RestController;
|
||||
|
||||
import io.swagger.v3.oas.annotations.Operation;
|
||||
import io.swagger.v3.oas.annotations.Parameter;
|
||||
import io.swagger.v3.oas.annotations.tags.Tag;
|
||||
import lombok.AllArgsConstructor;
|
||||
import xiaozhi.common.exception.ErrorCode;
|
||||
import xiaozhi.common.exception.RenException;
|
||||
import xiaozhi.common.page.PageData;
|
||||
import xiaozhi.common.utils.Result;
|
||||
import xiaozhi.modules.knowledge.dto.KnowledgeBaseDTO;
|
||||
import xiaozhi.modules.knowledge.service.KnowledgeBaseService;
|
||||
import xiaozhi.modules.security.user.SecurityUser;
|
||||
|
||||
@AllArgsConstructor
|
||||
@RestController
|
||||
@RequestMapping("/datasets")
|
||||
@Tag(name = "知识库管理")
|
||||
public class KnowledgeBaseController {
|
||||
|
||||
private final KnowledgeBaseService knowledgeBaseService;
|
||||
|
||||
@GetMapping
|
||||
@Operation(summary = "分页查询知识库列表")
|
||||
@RequiresPermissions("sys:role:normal")
|
||||
public Result<PageData<KnowledgeBaseDTO>> getPageList(
|
||||
@RequestParam(required = false) String name,
|
||||
@RequestParam(required = false, defaultValue = "1") Integer page,
|
||||
@RequestParam(required = false, defaultValue = "10") Integer page_size) {
|
||||
// 获取当前登录用户ID
|
||||
Long currentUserId = SecurityUser.getUserId();
|
||||
|
||||
KnowledgeBaseDTO knowledgeBaseDTO = new KnowledgeBaseDTO();
|
||||
knowledgeBaseDTO.setName(name);
|
||||
knowledgeBaseDTO.setCreator(currentUserId); // 设置创建者ID，用于权限过滤
|
||||
|
||||
PageData<KnowledgeBaseDTO> pageData = knowledgeBaseService.getPageList(knowledgeBaseDTO, page, page_size);
|
||||
return new Result<PageData<KnowledgeBaseDTO>>().ok(pageData);
|
||||
}
|
||||
|
||||
@GetMapping("/{dataset_id}")
|
||||
@Operation(summary = "根据知识库ID获取知识库详情")
|
||||
@RequiresPermissions("sys:role:normal")
|
||||
public Result<KnowledgeBaseDTO> getByDatasetId(@PathVariable("dataset_id") String datasetId) {
|
||||
// 获取当前登录用户ID
|
||||
Long currentUserId = SecurityUser.getUserId();
|
||||
|
||||
KnowledgeBaseDTO knowledgeBaseDTO = knowledgeBaseService.getByDatasetId(datasetId);
|
||||
|
||||
// 检查权限：用户只能查看自己创建的知识库
|
||||
if (knowledgeBaseDTO.getCreator() == null || !knowledgeBaseDTO.getCreator().equals(currentUserId)) {
|
||||
throw new RenException(ErrorCode.NO_PERMISSION);
|
||||
}
|
||||
|
||||
return new Result<KnowledgeBaseDTO>().ok(knowledgeBaseDTO);
|
||||
}
|
||||
|
||||
@PostMapping
|
||||
@Operation(summary = "创建知识库")
|
||||
@RequiresPermissions("sys:role:normal")
|
||||
public Result<KnowledgeBaseDTO> save(@RequestBody @Validated KnowledgeBaseDTO knowledgeBaseDTO) {
|
||||
KnowledgeBaseDTO resp = knowledgeBaseService.save(knowledgeBaseDTO);
|
||||
return new Result<KnowledgeBaseDTO>().ok(resp);
|
||||
}
|
||||
|
||||
@PutMapping("/{dataset_id}")
|
||||
@Operation(summary = "更新知识库")
|
||||
@RequiresPermissions("sys:role:normal")
|
||||
public Result<KnowledgeBaseDTO> update(@PathVariable("dataset_id") String datasetId,
|
||||
@RequestBody @Validated KnowledgeBaseDTO knowledgeBaseDTO) {
|
||||
// 获取当前登录用户ID
|
||||
Long currentUserId = SecurityUser.getUserId();
|
||||
|
||||
// 先获取现有知识库信息以检查权限
|
||||
KnowledgeBaseDTO existingKnowledgeBase = knowledgeBaseService.getByDatasetId(datasetId);
|
||||
|
||||
// 检查权限：用户只能更新自己创建的知识库
|
||||
if (existingKnowledgeBase.getCreator() == null || !existingKnowledgeBase.getCreator().equals(currentUserId)) {
|
||||
throw new RenException(ErrorCode.NO_PERMISSION);
|
||||
}
|
||||
|
||||
knowledgeBaseDTO.setDatasetId(datasetId);
|
||||
KnowledgeBaseDTO resp = knowledgeBaseService.update(knowledgeBaseDTO);
|
||||
return new Result<KnowledgeBaseDTO>().ok(resp);
|
||||
}
|
||||
|
||||
@DeleteMapping("/{dataset_id}")
|
||||
@Operation(summary = "删除单个知识库")
|
||||
@Parameter(name = "dataset_id", description = "知识库ID", required = true)
|
||||
@RequiresPermissions("sys:role:normal")
|
||||
public Result<Void> delete(@PathVariable("dataset_id") String datasetId) {
|
||||
// 获取当前登录用户ID
|
||||
Long currentUserId = SecurityUser.getUserId();
|
||||
|
||||
// 先获取现有知识库信息以检查权限
|
||||
KnowledgeBaseDTO existingKnowledgeBase = knowledgeBaseService.getByDatasetId(datasetId);
|
||||
|
||||
// 检查权限：用户只能删除自己创建的知识库
|
||||
if (existingKnowledgeBase.getCreator() == null || !existingKnowledgeBase.getCreator().equals(currentUserId)) {
|
||||
throw new RenException(ErrorCode.NO_PERMISSION);
|
||||
}
|
||||
|
||||
knowledgeBaseService.deleteByDatasetId(datasetId);
|
||||
return new Result<>();
|
||||
}
|
||||
|
||||
@DeleteMapping("/batch")
|
||||
@Operation(summary = "批量删除知识库")
|
||||
@Parameter(name = "ids", description = "知识库ID列表，用逗号分隔", required = true)
|
||||
@RequiresPermissions("sys:role:normal")
|
||||
public Result<Void> deleteBatch(@RequestParam("ids") String ids) {
|
||||
if (StringUtils.isBlank(ids)) {
|
||||
throw new RenException(ErrorCode.PARAMS_GET_ERROR);
|
||||
}
|
||||
|
||||
// 获取当前登录用户ID
|
||||
Long currentUserId = SecurityUser.getUserId();
|
||||
String[] idArray = ids.split(",");
|
||||
for (String datasetId : idArray) {
|
||||
if (StringUtils.isNotBlank(datasetId)) {
|
||||
// 先获取现有知识库信息以检查权限
|
||||
KnowledgeBaseDTO existingKnowledgeBase = knowledgeBaseService.getByDatasetId(datasetId.trim());
|
||||
|
||||
// 检查权限：用户只能删除自己创建的知识库
|
||||
if (existingKnowledgeBase.getCreator() == null
|
||||
|| !existingKnowledgeBase.getCreator().equals(currentUserId)) {
|
||||
throw new RenException(ErrorCode.NO_PERMISSION);
|
||||
}
|
||||
|
||||
knowledgeBaseService.deleteByDatasetId(datasetId.trim());
|
||||
}
|
||||
}
|
||||
return new Result<>();
|
||||
}
|
||||
|
||||
@GetMapping("/rag-models")
|
||||
@Operation(summary = "获取RAG模型列表")
|
||||
@RequiresPermissions("sys:role:normal")
|
||||
public Result<List<Map<String, Object>>> getRAGModels() {
|
||||
List<Map<String, Object>> result = knowledgeBaseService.getRAGModels();
|
||||
return new Result<List<Map<String, Object>>>().ok(result);
|
||||
}
|
||||
}
|
||||
+224
@@ -0,0 +1,224 @@
|
||||
package xiaozhi.modules.knowledge.controller;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
import org.apache.shiro.authz.annotation.RequiresPermissions;
|
||||
import org.springframework.web.bind.annotation.*;
|
||||
import org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
import com.fasterxml.jackson.core.type.TypeReference;
|
||||
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||
|
||||
import io.swagger.v3.oas.annotations.Operation;
|
||||
import io.swagger.v3.oas.annotations.Parameter;
|
||||
import io.swagger.v3.oas.annotations.tags.Tag;
|
||||
import lombok.AllArgsConstructor;
|
||||
import xiaozhi.common.exception.ErrorCode;
|
||||
import xiaozhi.common.exception.RenException;
|
||||
import xiaozhi.common.page.PageData;
|
||||
import xiaozhi.common.utils.Result;
|
||||
import xiaozhi.modules.knowledge.dto.KnowledgeBaseDTO;
|
||||
import xiaozhi.modules.knowledge.dto.KnowledgeFilesDTO;
|
||||
import xiaozhi.modules.knowledge.service.KnowledgeBaseService;
|
||||
import xiaozhi.modules.knowledge.service.KnowledgeFilesService;
|
||||
import xiaozhi.modules.security.user.SecurityUser;
|
||||
|
||||
@AllArgsConstructor
|
||||
@RestController
|
||||
@RequestMapping("/datasets/{dataset_id}")
|
||||
@Tag(name = "知识库文档管理")
|
||||
public class KnowledgeFilesController {
|
||||
|
||||
private final KnowledgeFilesService knowledgeFilesService;
|
||||
private final KnowledgeBaseService knowledgeBaseService;
|
||||
|
||||
/**
|
||||
* 验证当前用户是否有权限操作指定知识库
|
||||
*
|
||||
* @param datasetId 知识库ID
|
||||
*/
|
||||
private void validateKnowledgeBasePermission(String datasetId) {
|
||||
// 获取当前登录用户ID
|
||||
Long currentUserId = SecurityUser.getUserId();
|
||||
|
||||
// 获取知识库信息
|
||||
KnowledgeBaseDTO knowledgeBase = knowledgeBaseService.getByDatasetId(datasetId);
|
||||
|
||||
// 检查权限：用户只能操作自己创建的知识库
|
||||
if (knowledgeBase.getCreator() == null || !knowledgeBase.getCreator().equals(currentUserId)) {
|
||||
throw new RenException(ErrorCode.NO_PERMISSION);
|
||||
}
|
||||
}
|
||||
|
||||
@GetMapping("/documents")
|
||||
@Operation(summary = "分页查询文档列表")
|
||||
@RequiresPermissions("sys:role:normal")
|
||||
public Result<PageData<KnowledgeFilesDTO>> getPageList(
|
||||
@PathVariable("dataset_id") String datasetId,
|
||||
@RequestParam(required = false) String name,
|
||||
@RequestParam(required = false) Integer status,
|
||||
@RequestParam(required = false, defaultValue = "1") Integer page,
|
||||
@RequestParam(required = false, defaultValue = "10") Integer page_size) {
|
||||
// 验证知识库权限
|
||||
validateKnowledgeBasePermission(datasetId);
|
||||
|
||||
// 如果指定了状态参数，使用状态查询接口
|
||||
if (status != null) {
|
||||
PageData<KnowledgeFilesDTO> pageData = knowledgeFilesService.getPageListByStatus(datasetId, status, page, page_size);
|
||||
return new Result<PageData<KnowledgeFilesDTO>>().ok(pageData);
|
||||
}
|
||||
|
||||
// 否则使用通用查询接口
|
||||
KnowledgeFilesDTO knowledgeFilesDTO = new KnowledgeFilesDTO();
|
||||
knowledgeFilesDTO.setDatasetId(datasetId);
|
||||
knowledgeFilesDTO.setName(name);
|
||||
PageData<KnowledgeFilesDTO> pageData = knowledgeFilesService.getPageList(knowledgeFilesDTO, page, page_size);
|
||||
return new Result<PageData<KnowledgeFilesDTO>>().ok(pageData);
|
||||
}
|
||||
|
||||
@GetMapping("/documents/status/{status}")
|
||||
@Operation(summary = "根据状态分页查询文档列表")
|
||||
@RequiresPermissions("sys:role:normal")
|
||||
public Result<PageData<KnowledgeFilesDTO>> getPageListByStatus(
|
||||
@PathVariable("dataset_id") String datasetId,
|
||||
@PathVariable("status") Integer status,
|
||||
@RequestParam(required = false, defaultValue = "1") Integer page,
|
||||
@RequestParam(required = false, defaultValue = "10") Integer page_size) {
|
||||
// 验证知识库权限
|
||||
validateKnowledgeBasePermission(datasetId);
|
||||
|
||||
PageData<KnowledgeFilesDTO> pageData = knowledgeFilesService.getPageListByStatus(datasetId, status, page, page_size);
|
||||
return new Result<PageData<KnowledgeFilesDTO>>().ok(pageData);
|
||||
}
|
||||
|
||||
@PostMapping("/documents")
|
||||
@Operation(summary = "上传文档到知识库")
|
||||
@RequiresPermissions("sys:role:normal")
|
||||
public Result<KnowledgeFilesDTO> uploadDocument(
|
||||
@PathVariable("dataset_id") String datasetId,
|
||||
@RequestParam("file") MultipartFile file,
|
||||
@RequestParam(required = false) String name,
|
||||
@RequestParam(required = false) String chunkMethod,
|
||||
@RequestParam(required = false) String metaFields,
|
||||
@RequestParam(required = false) String parserConfig) {
|
||||
|
||||
// 验证知识库权限
|
||||
validateKnowledgeBasePermission(datasetId);
|
||||
|
||||
KnowledgeFilesDTO resp = knowledgeFilesService.uploadDocument(datasetId, file, name,
|
||||
metaFields != null ? parseJsonMap(metaFields) : null,
|
||||
chunkMethod,
|
||||
parserConfig != null ? parseJsonMap(parserConfig) : null);
|
||||
return new Result<KnowledgeFilesDTO>().ok(resp);
|
||||
}
|
||||
|
||||
@DeleteMapping("/documents/{document_id}")
|
||||
@Operation(summary = "删除单个文档")
|
||||
@Parameter(name = "document_id", description = "文档ID", required = true)
|
||||
@RequiresPermissions("sys:role:normal")
|
||||
public Result<Void> delete(@PathVariable("dataset_id") String datasetId,
|
||||
@PathVariable("document_id") String documentId) {
|
||||
// 验证知识库权限
|
||||
validateKnowledgeBasePermission(datasetId);
|
||||
|
||||
knowledgeFilesService.deleteByDocumentId(documentId, datasetId);
|
||||
return new Result<>();
|
||||
}
|
||||
|
||||
@PostMapping("/chunks")
|
||||
@Operation(summary = "解析文档（切块）")
|
||||
@RequiresPermissions("sys:role:normal")
|
||||
public Result<Void> parseDocuments(@PathVariable("dataset_id") String datasetId,
|
||||
@RequestBody Map<String, List<String>> requestBody) {
|
||||
// 验证知识库权限
|
||||
validateKnowledgeBasePermission(datasetId);
|
||||
|
||||
List<String> documentIds = requestBody.get("document_ids");
|
||||
if (documentIds == null || documentIds.isEmpty()) {
|
||||
return new Result<Void>().error("document_ids参数不能为空");
|
||||
}
|
||||
|
||||
boolean success = knowledgeFilesService.parseDocuments(datasetId, documentIds);
|
||||
if (success) {
|
||||
return new Result<Void>();
|
||||
} else {
|
||||
return new Result<Void>().error("文档解析失败，文档可能正在处理中");
|
||||
}
|
||||
}
|
||||
|
||||
@GetMapping("/documents/{document_id}/chunks")
|
||||
@Operation(summary = "列出指定文档的切片")
|
||||
@RequiresPermissions("sys:role:normal")
|
||||
public Result<Map<String, Object>> listChunks(@PathVariable("dataset_id") String datasetId,
|
||||
@PathVariable("document_id") String documentId,
|
||||
@RequestParam(required = false) String keywords,
|
||||
@RequestParam(required = false, defaultValue = "1") Integer page,
|
||||
@RequestParam(required = false, defaultValue = "1024") Integer page_size,
|
||||
@RequestParam(required = false) String id) {
|
||||
// 验证知识库权限
|
||||
validateKnowledgeBasePermission(datasetId);
|
||||
|
||||
Map<String, Object> result = knowledgeFilesService.listChunks(datasetId, documentId, keywords, page, page_size,
|
||||
id);
|
||||
return new Result<Map<String, Object>>().ok(result);
|
||||
}
|
||||
|
||||
/**
|
||||
* 召回测试
|
||||
*/
|
||||
@PostMapping("/retrieval-test")
|
||||
@Operation(summary = "召回测试")
|
||||
@RequiresPermissions("sys:role:normal")
|
||||
public Result<Map<String, Object>> retrievalTest(@PathVariable("dataset_id") String datasetId,
|
||||
@RequestBody Map<String, Object> params) {
|
||||
// 验证知识库权限
|
||||
validateKnowledgeBasePermission(datasetId);
|
||||
|
||||
try {
|
||||
// 提取参数
|
||||
String question = (String) params.get("question");
|
||||
if (question == null || question.trim().isEmpty()) {
|
||||
return new Result<Map<String, Object>>().error("问题不能为空");
|
||||
}
|
||||
|
||||
List<String> datasetIds = (List<String>) params.get("dataset_ids");
|
||||
List<String> documentIds = (List<String>) params.get("document_ids");
|
||||
Integer page = (Integer) params.get("page");
|
||||
Integer pageSize = (Integer) params.get("page_size");
|
||||
Float similarityThreshold = (Float) params.get("similarity_threshold");
|
||||
Float vectorSimilarityWeight = (Float) params.get("vector_similarity_weight");
|
||||
Integer topK = (Integer) params.get("top_k");
|
||||
String rerankId = (String) params.get("rerank_id");
|
||||
Boolean keyword = (Boolean) params.get("keyword");
|
||||
Boolean highlight = (Boolean) params.get("highlight");
|
||||
List<String> crossLanguages = (List<String>) params.get("cross_languages");
|
||||
Map<String, Object> metadataCondition = (Map<String, Object>) params.get("metadata_condition");
|
||||
|
||||
// 如果未指定数据集ID，使用当前数据集
|
||||
if (datasetIds == null || datasetIds.isEmpty()) {
|
||||
datasetIds = java.util.Arrays.asList(datasetId);
|
||||
}
|
||||
|
||||
Map<String, Object> result = knowledgeFilesService.retrievalTest(
|
||||
question, datasetIds, documentIds, page, pageSize, similarityThreshold,
|
||||
vectorSimilarityWeight, topK, rerankId, keyword, highlight, crossLanguages, metadataCondition);
|
||||
|
||||
return new Result<Map<String, Object>>().ok(result);
|
||||
} catch (Exception e) {
|
||||
return new Result<Map<String, Object>>().error("召回测试失败: " + e.getMessage());
|
||||
}
|
||||
}
|
||||
/**
|
||||
* 解析JSON字符串为Map对象
|
||||
*/
|
||||
private Map<String, Object> parseJsonMap(String jsonString) {
|
||||
try {
|
||||
ObjectMapper objectMapper = new ObjectMapper();
|
||||
return objectMapper.readValue(jsonString, new TypeReference<Map<String, Object>>() {
|
||||
});
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException("解析JSON字符串失败: " + jsonString, e);
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,14 @@
|
||||
package xiaozhi.modules.knowledge.dao;
|
||||
|
||||
import org.apache.ibatis.annotations.Mapper;
|
||||
|
||||
import xiaozhi.common.dao.BaseDao;
|
||||
import xiaozhi.modules.knowledge.entity.KnowledgeBaseEntity;
|
||||
|
||||
/**
|
||||
* 知识库知识库
|
||||
*/
|
||||
@Mapper
|
||||
public interface KnowledgeBaseDao extends BaseDao<KnowledgeBaseEntity> {
|
||||
|
||||
}
|
||||
@@ -0,0 +1,49 @@
|
||||
package xiaozhi.modules.knowledge.dto;
|
||||
|
||||
import java.io.Serial;
|
||||
import java.io.Serializable;
|
||||
import java.util.Date;
|
||||
|
||||
import io.swagger.v3.oas.annotations.media.Schema;
|
||||
import lombok.Data;
|
||||
|
||||
@Data
|
||||
@Schema(description = "知识库知识库")
|
||||
public class KnowledgeBaseDTO implements Serializable {
|
||||
|
||||
@Serial
|
||||
private static final long serialVersionUID = 1L;
|
||||
|
||||
@Schema(description = "唯一标识")
|
||||
private String id;
|
||||
|
||||
@Schema(description = "知识库ID")
|
||||
private String datasetId;
|
||||
|
||||
@Schema(description = "RAG模型配置ID")
|
||||
private String ragModelId;
|
||||
|
||||
@Schema(description = "知识库名称")
|
||||
private String name;
|
||||
|
||||
@Schema(description = "知识库描述")
|
||||
private String description;
|
||||
|
||||
@Schema(description = "状态(0:禁用 1:启用)")
|
||||
private Integer status;
|
||||
|
||||
@Schema(description = "创建者")
|
||||
private Long creator;
|
||||
|
||||
@Schema(description = "创建时间")
|
||||
private Date createdAt;
|
||||
|
||||
@Schema(description = "更新者")
|
||||
private Long updater;
|
||||
|
||||
@Schema(description = "更新时间")
|
||||
private Date updatedAt;
|
||||
|
||||
@Schema(description = "文档数量")
|
||||
private Integer documentCount;
|
||||
}
|
||||
@@ -0,0 +1,96 @@
|
||||
package xiaozhi.modules.knowledge.dto;
|
||||
|
||||
import java.io.Serial;
|
||||
import java.io.Serializable;
|
||||
import java.util.Date;
|
||||
import java.util.Map;
|
||||
|
||||
import io.swagger.v3.oas.annotations.media.Schema;
|
||||
import lombok.Data;
|
||||
|
||||
@Data
|
||||
@Schema(description = "知识库文档")
|
||||
public class KnowledgeFilesDTO implements Serializable {
|
||||
|
||||
@Serial
|
||||
private static final long serialVersionUID = 1L;
|
||||
@Schema(description = "唯一标识")
|
||||
private String id;
|
||||
|
||||
@Schema(description = "文档ID")
|
||||
private String documentId;
|
||||
|
||||
@Schema(description = "知识库ID")
|
||||
private String datasetId;
|
||||
|
||||
@Schema(description = "文档名称")
|
||||
private String name;
|
||||
|
||||
@Schema(description = "文档类型")
|
||||
private String fileType;
|
||||
|
||||
@Schema(description = "文件大小（字节）")
|
||||
private Long fileSize;
|
||||
|
||||
@Schema(description = "文件路径")
|
||||
private String filePath;
|
||||
|
||||
@Schema(description = "元数据字段")
|
||||
private Map<String, Object> metaFields;
|
||||
|
||||
@Schema(description = "分块方法")
|
||||
private String chunkMethod;
|
||||
|
||||
@Schema(description = "解析器配置")
|
||||
private Map<String, Object> parserConfig;
|
||||
|
||||
@Schema(description = "状态")
|
||||
private Integer status;
|
||||
|
||||
@Schema(description = "文档解析状态")
|
||||
private String run;
|
||||
|
||||
@Schema(description = "创建者")
|
||||
private Long creator;
|
||||
|
||||
@Schema(description = "创建时间")
|
||||
private Date createdAt;
|
||||
|
||||
@Schema(description = "更新者")
|
||||
private Long updater;
|
||||
|
||||
@Schema(description = "更新时间")
|
||||
private Date updatedAt;
|
||||
|
||||
// 文档解析状态常量定义
|
||||
private static final Integer STATUS_UNSTART = 0;
|
||||
private static final Integer STATUS_RUNNING = 1;
|
||||
private static final Integer STATUS_CANCEL = 2;
|
||||
private static final Integer STATUS_DONE = 3;
|
||||
private static final Integer STATUS_FAIL = 4;
|
||||
|
||||
/**
|
||||
* 获取文档解析状态码（基于run字段转换）
|
||||
*/
|
||||
public Integer getParseStatusCode() {
|
||||
if (run == null) {
|
||||
return STATUS_UNSTART;
|
||||
}
|
||||
|
||||
// 根据run字段的值直接映射到对应的状态码
|
||||
switch (run.toUpperCase()) {
|
||||
case "RUNNING":
|
||||
return STATUS_RUNNING;
|
||||
case "CANCEL":
|
||||
return STATUS_CANCEL;
|
||||
case "DONE":
|
||||
return STATUS_DONE;
|
||||
case "FAIL":
|
||||
return STATUS_FAIL;
|
||||
case "UNSTART":
|
||||
default:
|
||||
return STATUS_UNSTART;
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
+53
@@ -0,0 +1,53 @@
|
||||
package xiaozhi.modules.knowledge.entity;
|
||||
|
||||
import java.util.Date;
|
||||
|
||||
import com.baomidou.mybatisplus.annotation.FieldFill;
|
||||
import com.baomidou.mybatisplus.annotation.IdType;
|
||||
import com.baomidou.mybatisplus.annotation.TableField;
|
||||
import com.baomidou.mybatisplus.annotation.TableId;
|
||||
import com.baomidou.mybatisplus.annotation.TableName;
|
||||
|
||||
import io.swagger.v3.oas.annotations.media.Schema;
|
||||
import lombok.Data;
|
||||
|
||||
@Data
|
||||
@TableName(value = "ai_rag_dataset", autoResultMap = true)
|
||||
@Schema(description = "知识库知识库表")
|
||||
public class KnowledgeBaseEntity {
|
||||
|
||||
@TableId(type = IdType.ASSIGN_UUID)
|
||||
@Schema(description = "唯一标识")
|
||||
private String id;
|
||||
|
||||
@Schema(description = "知识库ID")
|
||||
private String datasetId;
|
||||
|
||||
@Schema(description = "RAG模型配置ID")
|
||||
private String ragModelId;
|
||||
|
||||
@Schema(description = "知识库名称")
|
||||
private String name;
|
||||
|
||||
@Schema(description = "知识库描述")
|
||||
private String description;
|
||||
|
||||
@Schema(description = "状态(0:禁用 1:启用)")
|
||||
private Integer status;
|
||||
|
||||
@Schema(description = "创建者")
|
||||
@TableField(fill = FieldFill.INSERT)
|
||||
private Long creator;
|
||||
|
||||
@Schema(description = "创建时间")
|
||||
@TableField(fill = FieldFill.INSERT)
|
||||
private Date createdAt;
|
||||
|
||||
@Schema(description = "更新者")
|
||||
@TableField(fill = FieldFill.UPDATE)
|
||||
private Long updater;
|
||||
|
||||
@Schema(description = "更新时间")
|
||||
@TableField(fill = FieldFill.UPDATE)
|
||||
private Date updatedAt;
|
||||
}
|
||||
+86
@@ -0,0 +1,86 @@
|
||||
package xiaozhi.modules.knowledge.service;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
import xiaozhi.common.page.PageData;
|
||||
import xiaozhi.common.service.BaseService;
|
||||
import xiaozhi.modules.knowledge.dto.KnowledgeBaseDTO;
|
||||
import xiaozhi.modules.knowledge.entity.KnowledgeBaseEntity;
|
||||
|
||||
/**
|
||||
* 知识库知识库服务接口
|
||||
*/
|
||||
public interface KnowledgeBaseService extends BaseService<KnowledgeBaseEntity> {
|
||||
|
||||
/**
|
||||
* 分页查询知识库列表
|
||||
*
|
||||
* @param knowledgeBaseDTO 查询条件
|
||||
* @param page 页码
|
||||
* @param limit 每页数量
|
||||
* @return 分页数据
|
||||
*/
|
||||
PageData<KnowledgeBaseDTO> getPageList(KnowledgeBaseDTO knowledgeBaseDTO, Integer page, Integer limit);
|
||||
|
||||
/**
|
||||
* 根据ID获取知识库详情
|
||||
*
|
||||
* @param id 知识库ID
|
||||
* @return 知识库详情
|
||||
*/
|
||||
KnowledgeBaseDTO getById(String id);
|
||||
|
||||
/**
|
||||
* 新增知识库
|
||||
*
|
||||
* @param knowledgeBaseDTO 知识库信息
|
||||
* @return 新增的知识库
|
||||
*/
|
||||
KnowledgeBaseDTO save(KnowledgeBaseDTO knowledgeBaseDTO);
|
||||
|
||||
/**
|
||||
* 更新知识库
|
||||
*
|
||||
* @param knowledgeBaseDTO 知识库信息
|
||||
* @return 更新的知识库
|
||||
*/
|
||||
KnowledgeBaseDTO update(KnowledgeBaseDTO knowledgeBaseDTO);
|
||||
|
||||
/**
|
||||
* 根据知识库ID查询知识库
|
||||
*
|
||||
* @param datasetId 知识库ID
|
||||
* @return 知识库详情
|
||||
*/
|
||||
KnowledgeBaseDTO getByDatasetId(String datasetId);
|
||||
|
||||
/**
|
||||
* 根据知识库ID删除知识库
|
||||
*
|
||||
* @param datasetId 知识库ID
|
||||
*/
|
||||
void deleteByDatasetId(String datasetId);
|
||||
|
||||
/**
|
||||
* 获取RAG配置信息
|
||||
*
|
||||
* @param ragModelId RAG模型配置ID
|
||||
* @return RAG配置信息
|
||||
*/
|
||||
Map<String, Object> getRAGConfig(String ragModelId);
|
||||
|
||||
/**
|
||||
* 获取默认RAG配置信息
|
||||
*
|
||||
* @return 默认RAG配置信息
|
||||
*/
|
||||
Map<String, Object> getDefaultRAGConfig();
|
||||
|
||||
/**
|
||||
* 获取RAG模型列表
|
||||
*
|
||||
* @return RAG模型列表
|
||||
*/
|
||||
List<Map<String, Object>> getRAGModels();
|
||||
}
|
||||
+130
@@ -0,0 +1,130 @@
|
||||
package xiaozhi.modules.knowledge.service;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
import org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
import xiaozhi.common.page.PageData;
|
||||
import xiaozhi.modules.knowledge.dto.KnowledgeFilesDTO;
|
||||
|
||||
/**
|
||||
* 知识库文档服务接口
|
||||
*/
|
||||
public interface KnowledgeFilesService {
|
||||
|
||||
/**
|
||||
* 分页查询文档列表
|
||||
*
|
||||
* @param knowledgeFilesDTO 查询条件
|
||||
* @param page 页码
|
||||
* @param limit 每页数量
|
||||
* @return 分页数据
|
||||
*/
|
||||
PageData<KnowledgeFilesDTO> getPageList(KnowledgeFilesDTO knowledgeFilesDTO, Integer page, Integer limit);
|
||||
|
||||
/**
|
||||
* 根据文档ID和知识库ID获取文档详情
|
||||
*
|
||||
* @param documentId 文档ID
|
||||
* @param datasetId 知识库ID
|
||||
* @return 文档详情
|
||||
*/
|
||||
KnowledgeFilesDTO getByDocumentId(String documentId, String datasetId);
|
||||
|
||||
/**
|
||||
* 上传文档到知识库
|
||||
*
|
||||
* @param datasetId 知识库ID
|
||||
* @param file 上传的文件
|
||||
* @param name 文档名称
|
||||
* @param metaFields 元数据字段
|
||||
* @param chunkMethod 分块方法
|
||||
* @param parserConfig 解析器配置
|
||||
* @return 上传的文档信息
|
||||
*/
|
||||
KnowledgeFilesDTO uploadDocument(String datasetId, MultipartFile file, String name,
|
||||
Map<String, Object> metaFields, String chunkMethod,
|
||||
Map<String, Object> parserConfig);
|
||||
|
||||
/**
|
||||
* 根据状态分页查询文档列表
|
||||
*
|
||||
* @param datasetId 知识库ID
|
||||
* @param status 文档解析状态（0-未开始，1-进行中，2-已取消，3-已完成，4-失败）
|
||||
* @param page 页码
|
||||
* @param limit 每页数量
|
||||
* @return 分页数据
|
||||
*/
|
||||
PageData<KnowledgeFilesDTO> getPageListByStatus(String datasetId, Integer status, Integer page, Integer limit);
|
||||
|
||||
/**
|
||||
* 根据文档ID和知识库ID删除文档
|
||||
*
|
||||
* @param documentId 文档ID
|
||||
* @param datasetId 知识库ID
|
||||
*/
|
||||
void deleteByDocumentId(String documentId, String datasetId);
|
||||
|
||||
/**
|
||||
* 获取RAG配置信息
|
||||
*
|
||||
* @param ragModelId RAG模型配置ID
|
||||
* @return RAG配置信息
|
||||
*/
|
||||
Map<String, Object> getRAGConfig(String ragModelId);
|
||||
|
||||
/**
|
||||
* 获取默认RAG配置信息
|
||||
*
|
||||
* @return 默认RAG配置信息
|
||||
*/
|
||||
Map<String, Object> getDefaultRAGConfig();
|
||||
|
||||
/**
|
||||
* 解析文档（切块）
|
||||
*
|
||||
* @param datasetId 知识库ID
|
||||
* @param documentIds 文档ID列表
|
||||
* @return 解析结果
|
||||
*/
|
||||
boolean parseDocuments(String datasetId, List<String> documentIds);
|
||||
|
||||
/**
|
||||
* 列出指定文档的切片
|
||||
*
|
||||
* @param datasetId 知识库ID
|
||||
* @param documentId 文档ID
|
||||
* @param keywords 关键词过滤
|
||||
* @param page 页码
|
||||
* @param pageSize 每页数量
|
||||
* @param chunkId 切片ID
|
||||
* @return 切片列表信息
|
||||
*/
|
||||
Map<String, Object> listChunks(String datasetId, String documentId, String keywords,
|
||||
Integer page, Integer pageSize, String chunkId);
|
||||
|
||||
/**
|
||||
* 召回测试 - 从指定数据集或文档中检索相关切片
|
||||
*
|
||||
* @param question 用户查询或查询关键词
|
||||
* @param datasetIds 数据集ID列表
|
||||
* @param documentIds 文档ID列表
|
||||
* @param page 页码
|
||||
* @param pageSize 每页数量
|
||||
* @param similarityThreshold 最小相似度阈值
|
||||
* @param vectorSimilarityWeight 向量相似度权重
|
||||
* @param topK 参与向量余弦计算的切片数量
|
||||
* @param rerankId 重排模型ID
|
||||
* @param keyword 是否启用关键词匹配
|
||||
* @param highlight 是否启用高亮显示
|
||||
* @param crossLanguages 跨语言翻译列表
|
||||
* @param metadataCondition 元数据过滤条件
|
||||
* @return 召回测试结果
|
||||
*/
|
||||
Map<String, Object> retrievalTest(String question, List<String> datasetIds, List<String> documentIds,
|
||||
Integer page, Integer pageSize, Float similarityThreshold,
|
||||
Float vectorSimilarityWeight, Integer topK, String rerankId,
|
||||
Boolean keyword, Boolean highlight, List<String> crossLanguages,
|
||||
Map<String, Object> metadataCondition);
|
||||
}
|
||||
+705
@@ -0,0 +1,705 @@
|
||||
package xiaozhi.modules.knowledge.service.impl;
|
||||
|
||||
import java.io.Serializable;
|
||||
import java.util.ArrayList;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
import org.apache.commons.lang3.StringUtils;
|
||||
import org.springframework.http.HttpEntity;
|
||||
import org.springframework.http.HttpHeaders;
|
||||
import org.springframework.http.HttpMethod;
|
||||
import org.springframework.http.MediaType;
|
||||
import org.springframework.http.ResponseEntity;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.springframework.web.client.RestTemplate;
|
||||
|
||||
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
||||
import com.baomidou.mybatisplus.core.metadata.IPage;
|
||||
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
||||
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||
|
||||
import lombok.AllArgsConstructor;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import xiaozhi.common.constant.Constant;
|
||||
import xiaozhi.common.exception.ErrorCode;
|
||||
import xiaozhi.common.exception.RenException;
|
||||
import xiaozhi.common.page.PageData;
|
||||
import xiaozhi.common.redis.RedisKeys;
|
||||
import xiaozhi.common.redis.RedisUtils;
|
||||
import xiaozhi.common.service.impl.BaseServiceImpl;
|
||||
import xiaozhi.common.utils.ConvertUtils;
|
||||
import xiaozhi.common.utils.MessageUtils;
|
||||
import xiaozhi.modules.knowledge.dao.KnowledgeBaseDao;
|
||||
import xiaozhi.modules.knowledge.dto.KnowledgeBaseDTO;
|
||||
import xiaozhi.modules.knowledge.entity.KnowledgeBaseEntity;
|
||||
import xiaozhi.modules.knowledge.service.KnowledgeBaseService;
|
||||
import xiaozhi.modules.model.dao.ModelConfigDao;
|
||||
import xiaozhi.modules.model.entity.ModelConfigEntity;
|
||||
import xiaozhi.modules.model.service.ModelConfigService;
|
||||
import xiaozhi.modules.security.user.SecurityUser;
|
||||
|
||||
@Service
|
||||
@AllArgsConstructor
|
||||
@Slf4j
|
||||
public class KnowledgeBaseServiceImpl extends BaseServiceImpl<KnowledgeBaseDao, KnowledgeBaseEntity>
|
||||
implements KnowledgeBaseService {
|
||||
|
||||
private final KnowledgeBaseDao knowledgeBaseDao;
|
||||
private final ModelConfigService modelConfigService;
|
||||
private final ModelConfigDao modelConfigDao;
|
||||
private final RedisUtils redisUtils;
|
||||
private RestTemplate restTemplate = new RestTemplate();
|
||||
|
||||
@Override
|
||||
public KnowledgeBaseEntity selectById(Serializable datasetId) {
|
||||
if (datasetId == null) {
|
||||
return null;
|
||||
}
|
||||
|
||||
// 先从Redis获取缓存
|
||||
String key = RedisKeys.getKnowledgeBaseCacheKey(datasetId.toString());
|
||||
KnowledgeBaseEntity cachedEntity = (KnowledgeBaseEntity) redisUtils.get(key);
|
||||
if (cachedEntity != null) {
|
||||
return cachedEntity;
|
||||
}
|
||||
|
||||
// 如果缓存中没有，则从数据库获取
|
||||
KnowledgeBaseEntity entity = knowledgeBaseDao.selectById(datasetId);
|
||||
if (entity == null) {
|
||||
return null;
|
||||
}
|
||||
|
||||
// 存入Redis缓存
|
||||
redisUtils.set(key, entity);
|
||||
|
||||
return entity;
|
||||
}
|
||||
|
||||
@Override
|
||||
public PageData<KnowledgeBaseDTO> getPageList(KnowledgeBaseDTO knowledgeBaseDTO, Integer page, Integer limit) {
|
||||
long curPage = page;
|
||||
long pageSize = limit;
|
||||
Page<KnowledgeBaseEntity> pageInfo = new Page<>(curPage, pageSize);
|
||||
|
||||
QueryWrapper<KnowledgeBaseEntity> queryWrapper = new QueryWrapper<>();
|
||||
|
||||
// 添加查询条件
|
||||
if (knowledgeBaseDTO != null) {
|
||||
queryWrapper.like(StringUtils.isNotBlank(knowledgeBaseDTO.getName()), "name", knowledgeBaseDTO.getName());
|
||||
queryWrapper.eq(knowledgeBaseDTO.getStatus() != null, "status", knowledgeBaseDTO.getStatus());
|
||||
queryWrapper.eq("creator", knowledgeBaseDTO.getCreator());
|
||||
}
|
||||
|
||||
// 添加排序规则：按创建时间降序
|
||||
queryWrapper.orderByDesc("created_at");
|
||||
|
||||
IPage<KnowledgeBaseEntity> knowledgeBaseEntityIPage = knowledgeBaseDao.selectPage(pageInfo, queryWrapper);
|
||||
|
||||
// 获取分页数据
|
||||
PageData<KnowledgeBaseDTO> pageData = getPageData(knowledgeBaseEntityIPage, KnowledgeBaseDTO.class);
|
||||
|
||||
// 为每个知识库获取文档数量
|
||||
if (pageData != null && pageData.getList() != null) {
|
||||
for (KnowledgeBaseDTO knowledgeBase : pageData.getList()) {
|
||||
try {
|
||||
Integer documentCount = getDocumentCountFromRAGFlow(knowledgeBase.getDatasetId(),
|
||||
knowledgeBase.getRagModelId());
|
||||
knowledgeBase.setDocumentCount(documentCount);
|
||||
} catch (Exception e) {
|
||||
log.warn("获取知识库 {} 的文档数量失败: {}", knowledgeBase.getDatasetId(), e.getMessage());
|
||||
knowledgeBase.setDocumentCount(0); // 设置默认值
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return pageData;
|
||||
}
|
||||
|
||||
@Override
|
||||
public KnowledgeBaseDTO getById(String id) {
|
||||
if (StringUtils.isBlank(id)) {
|
||||
throw new RenException(ErrorCode.IDENTIFIER_NOT_NULL);
|
||||
}
|
||||
|
||||
KnowledgeBaseEntity entity = knowledgeBaseDao.selectById(id);
|
||||
if (entity == null) {
|
||||
throw new RenException(ErrorCode.Knowledge_Base_RECORD_NOT_EXISTS);
|
||||
}
|
||||
|
||||
return ConvertUtils.sourceToTarget(entity, KnowledgeBaseDTO.class);
|
||||
}
|
||||
|
||||
@Override
|
||||
public KnowledgeBaseDTO save(KnowledgeBaseDTO knowledgeBaseDTO) {
|
||||
if (knowledgeBaseDTO == null) {
|
||||
throw new RenException(ErrorCode.PARAMS_GET_ERROR);
|
||||
}
|
||||
|
||||
// 检查是否存在同名知识库
|
||||
checkDuplicateKnowledgeBaseName(knowledgeBaseDTO, null);
|
||||
|
||||
String datasetId = null;
|
||||
// 调用RAGFlow API创建数据集
|
||||
try {
|
||||
Map<String, Object> ragConfig = getValidatedRAGConfig(knowledgeBaseDTO.getRagModelId());
|
||||
datasetId = createDatasetInRAGFlow(
|
||||
knowledgeBaseDTO.getName(),
|
||||
knowledgeBaseDTO.getDescription(),
|
||||
ragConfig);
|
||||
} catch (Exception e) {
|
||||
// 如果RAG API调用失败，直接抛出异常，无需回滚（因为还没有插入本地数据库）
|
||||
throw e;
|
||||
}
|
||||
|
||||
// 验证数据集ID是否已存在
|
||||
KnowledgeBaseEntity existingEntity = knowledgeBaseDao.selectOne(
|
||||
new QueryWrapper<KnowledgeBaseEntity>().eq("dataset_id", datasetId));
|
||||
if (existingEntity != null) {
|
||||
// 如果datasetId已存在，删除RAGFlow中的数据集并抛出异常
|
||||
try {
|
||||
Map<String, Object> ragConfig = getValidatedRAGConfig(knowledgeBaseDTO.getRagModelId());
|
||||
deleteDatasetInRAGFlow(datasetId, ragConfig);
|
||||
} catch (Exception deleteException) {
|
||||
log.warn("删除重复datasetId的RAGFlow数据集失败: {}", deleteException.getMessage());
|
||||
}
|
||||
throw new RenException(ErrorCode.DB_RECORD_EXISTS);
|
||||
}
|
||||
|
||||
// 创建本地实体并保存
|
||||
KnowledgeBaseEntity entity = ConvertUtils.sourceToTarget(knowledgeBaseDTO, KnowledgeBaseEntity.class);
|
||||
entity.setDatasetId(datasetId);
|
||||
knowledgeBaseDao.insert(entity);
|
||||
|
||||
return ConvertUtils.sourceToTarget(entity, KnowledgeBaseDTO.class);
|
||||
}
|
||||
|
||||
@Override
|
||||
public KnowledgeBaseDTO update(KnowledgeBaseDTO knowledgeBaseDTO) {
|
||||
if (knowledgeBaseDTO == null || StringUtils.isBlank(knowledgeBaseDTO.getId())) {
|
||||
throw new RenException(ErrorCode.IDENTIFIER_NOT_NULL);
|
||||
}
|
||||
|
||||
// 检查记录是否存在
|
||||
KnowledgeBaseEntity existingEntity = knowledgeBaseDao.selectById(knowledgeBaseDTO.getId());
|
||||
if (existingEntity == null) {
|
||||
throw new RenException(ErrorCode.Knowledge_Base_RECORD_NOT_EXISTS);
|
||||
}
|
||||
|
||||
// 检查是否存在同名知识库（排除当前记录）
|
||||
checkDuplicateKnowledgeBaseName(knowledgeBaseDTO, knowledgeBaseDTO.getId());
|
||||
|
||||
// 验证数据集ID是否与其他记录冲突
|
||||
if (StringUtils.isNotBlank(knowledgeBaseDTO.getDatasetId())) {
|
||||
KnowledgeBaseEntity conflictEntity = knowledgeBaseDao.selectOne(
|
||||
new QueryWrapper<KnowledgeBaseEntity>()
|
||||
.eq("dataset_id", knowledgeBaseDTO.getDatasetId())
|
||||
.ne("id", knowledgeBaseDTO.getId()));
|
||||
if (conflictEntity != null) {
|
||||
throw new RenException(ErrorCode.DB_RECORD_EXISTS);
|
||||
}
|
||||
}
|
||||
|
||||
KnowledgeBaseEntity entity = ConvertUtils.sourceToTarget(knowledgeBaseDTO, KnowledgeBaseEntity.class);
|
||||
knowledgeBaseDao.updateById(entity);
|
||||
|
||||
// 删除缓存
|
||||
if (entity.getDatasetId() != null) {
|
||||
redisUtils.delete(RedisKeys.getKnowledgeBaseCacheKey(entity.getId()));
|
||||
}
|
||||
|
||||
// 调用RAGFlow API更新数据集
|
||||
if (StringUtils.isNotBlank(knowledgeBaseDTO.getDatasetId())) {
|
||||
try {
|
||||
Map<String, Object> ragConfig = getValidatedRAGConfig(knowledgeBaseDTO.getRagModelId());
|
||||
updateDatasetInRAGFlow(
|
||||
knowledgeBaseDTO.getDatasetId(),
|
||||
knowledgeBaseDTO.getName(),
|
||||
knowledgeBaseDTO.getDescription(),
|
||||
ragConfig);
|
||||
} catch (Exception e) {
|
||||
// 如果RAG API调用失败，回滚本地数据库操作
|
||||
knowledgeBaseDao.updateById(existingEntity);
|
||||
throw e;
|
||||
}
|
||||
}
|
||||
|
||||
return ConvertUtils.sourceToTarget(entity, KnowledgeBaseDTO.class);
|
||||
}
|
||||
|
||||
@Override
|
||||
public KnowledgeBaseDTO getByDatasetId(String datasetId) {
|
||||
if (StringUtils.isBlank(datasetId)) {
|
||||
throw new RenException(ErrorCode.PARAMS_GET_ERROR);
|
||||
}
|
||||
|
||||
KnowledgeBaseEntity entity = knowledgeBaseDao.selectOne(
|
||||
new QueryWrapper<KnowledgeBaseEntity>().eq("dataset_id", datasetId));
|
||||
|
||||
if (entity == null) {
|
||||
throw new RenException(ErrorCode.Knowledge_Base_RECORD_NOT_EXISTS);
|
||||
}
|
||||
|
||||
return ConvertUtils.sourceToTarget(entity, KnowledgeBaseDTO.class);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void deleteByDatasetId(String datasetId) {
|
||||
if (StringUtils.isBlank(datasetId)) {
|
||||
throw new RenException(ErrorCode.PARAMS_GET_ERROR);
|
||||
}
|
||||
|
||||
log.info("=== 开始通过datasetId删除操作 ===");
|
||||
log.info("删除datasetId: {}", datasetId);
|
||||
|
||||
KnowledgeBaseEntity entity = knowledgeBaseDao.selectOne(
|
||||
new QueryWrapper<KnowledgeBaseEntity>().eq("dataset_id", datasetId));
|
||||
|
||||
if (entity == null) {
|
||||
log.warn("记录不存在，datasetId: {}", datasetId);
|
||||
throw new RenException(ErrorCode.Knowledge_Base_RECORD_NOT_EXISTS);
|
||||
}
|
||||
redisUtils.delete(RedisKeys.getKnowledgeBaseCacheKey(entity.getId()));
|
||||
|
||||
log.info("找到记录: ID={}, datasetId={}, ragModelId={}",
|
||||
entity.getId(), entity.getDatasetId(), entity.getRagModelId());
|
||||
|
||||
// 先删除本地数据库记录
|
||||
int deleteCount = knowledgeBaseDao.deleteById(entity.getId());
|
||||
log.info("本地数据库删除结果: {}", deleteCount > 0 ? "成功" : "失败");
|
||||
|
||||
// 调用RAGFlow API删除数据集
|
||||
if (StringUtils.isNotBlank(entity.getDatasetId()) && StringUtils.isNotBlank(entity.getRagModelId())) {
|
||||
try {
|
||||
log.info("开始调用RAGFlow API删除数据集");
|
||||
Map<String, Object> ragConfig = getValidatedRAGConfig(entity.getRagModelId());
|
||||
deleteDatasetInRAGFlow(entity.getDatasetId(), ragConfig);
|
||||
log.info("RAGFlow API删除调用完成");
|
||||
} catch (Exception e) {
|
||||
log.warn("删除RAGFlow数据集失败: {}", e.getMessage());
|
||||
}
|
||||
} else {
|
||||
log.warn("datasetId或ragModelId为空，跳过RAGFlow删除");
|
||||
}
|
||||
|
||||
log.info("=== 通过datasetId删除操作结束 ===");
|
||||
}
|
||||
|
||||
@Override
|
||||
public Map<String, Object> getRAGConfig(String ragModelId) {
|
||||
if (StringUtils.isBlank(ragModelId)) {
|
||||
throw new RenException(ErrorCode.PARAMS_GET_ERROR);
|
||||
}
|
||||
|
||||
// 从缓存获取模型配置
|
||||
ModelConfigEntity modelConfig = modelConfigService.getModelByIdFromCache(ragModelId);
|
||||
if (modelConfig == null || modelConfig.getConfigJson() == null) {
|
||||
throw new RenException(ErrorCode.RAG_CONFIG_NOT_FOUND);
|
||||
}
|
||||
|
||||
// 验证是否为RAG类型配置
|
||||
if (!Constant.RAG_CONFIG_TYPE.equals(modelConfig.getModelType().toUpperCase())) {
|
||||
throw new RenException(ErrorCode.RAG_CONFIG_TYPE_ERROR);
|
||||
}
|
||||
|
||||
Map<String, Object> config = modelConfig.getConfigJson();
|
||||
|
||||
// 验证必要的配置参数
|
||||
validateRagConfig(config);
|
||||
|
||||
// 返回配置信息
|
||||
return config;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Map<String, Object> getDefaultRAGConfig() {
|
||||
// 获取默认RAG模型配置
|
||||
QueryWrapper<ModelConfigEntity> queryWrapper = new QueryWrapper<>();
|
||||
queryWrapper.eq("model_type", Constant.RAG_CONFIG_TYPE)
|
||||
.eq("is_default", 1)
|
||||
.eq("is_enabled", 1);
|
||||
|
||||
List<ModelConfigEntity> modelConfigs = modelConfigDao.selectList(queryWrapper);
|
||||
if (modelConfigs == null || modelConfigs.isEmpty()) {
|
||||
throw new RenException(ErrorCode.RAG_DEFAULT_CONFIG_NOT_FOUND);
|
||||
}
|
||||
|
||||
ModelConfigEntity defaultConfig = modelConfigs.get(0);
|
||||
if (defaultConfig.getConfigJson() == null) {
|
||||
throw new RenException(ErrorCode.RAG_CONFIG_NOT_FOUND);
|
||||
}
|
||||
|
||||
Map<String, Object> config = defaultConfig.getConfigJson();
|
||||
|
||||
// 验证必要的配置参数
|
||||
validateRagConfig(config);
|
||||
|
||||
return config;
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<Map<String, Object>> getRAGModels() {
|
||||
// 查询RAG类型的模型配置
|
||||
QueryWrapper<ModelConfigEntity> queryWrapper = new QueryWrapper<>();
|
||||
queryWrapper.eq("model_type", Constant.RAG_CONFIG_TYPE)
|
||||
.eq("is_enabled", 1)
|
||||
.orderByDesc("is_default")
|
||||
.orderByDesc("create_date");
|
||||
|
||||
List<ModelConfigEntity> modelConfigs = modelConfigDao.selectList(queryWrapper);
|
||||
|
||||
List<Map<String, Object>> modelList = new ArrayList<>();
|
||||
for (ModelConfigEntity modelConfig : modelConfigs) {
|
||||
Map<String, Object> modelInfo = new HashMap<>();
|
||||
modelInfo.put("id", modelConfig.getId());
|
||||
modelInfo.put("modelName", modelConfig.getModelName());
|
||||
modelList.add(modelInfo);
|
||||
}
|
||||
return modelList;
|
||||
}
|
||||
|
||||
/**
|
||||
* 验证RAG配置中是否包含必要的参数
|
||||
*/
|
||||
private void validateRagConfig(Map<String, Object> config) {
|
||||
if (config == null) {
|
||||
throw new RenException(ErrorCode.RAG_CONFIG_NOT_FOUND);
|
||||
}
|
||||
|
||||
// 从配置中提取必要的参数
|
||||
String baseUrl = (String) config.get("base_url");
|
||||
String apiKey = (String) config.get("api_key");
|
||||
|
||||
// 验证base_url是否存在且非空
|
||||
if (StringUtils.isBlank(baseUrl)) {
|
||||
throw new RenException(ErrorCode.RAG_API_ERROR);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 调用RAGFlow API创建数据集
|
||||
*/
|
||||
private String createDatasetInRAGFlow(String name, String description, Map<String, Object> ragConfig) {
|
||||
String datasetId = null;
|
||||
String baseUrl = (String) ragConfig.get("base_url");
|
||||
String apiKey = (String) ragConfig.get("api_key");
|
||||
|
||||
log.info("开始调用RAGFlow API创建数据集, name: {}", name);
|
||||
log.debug("RAGFlow配置 - baseUrl: {}, apiKey: {}", baseUrl, StringUtils.isBlank(apiKey) ? "未配置" : "已配置");
|
||||
|
||||
// 构建请求URL
|
||||
String url = baseUrl + "/api/v1/datasets";
|
||||
log.debug("请求URL: {}", url);
|
||||
|
||||
// 构建请求体
|
||||
Map<String, Object> requestBody = new HashMap<>();
|
||||
String username = SecurityUser.getUser().getUsername();
|
||||
requestBody.put("name", username + "_" + name);
|
||||
if (StringUtils.isNotBlank(description)) {
|
||||
requestBody.put("description", description);
|
||||
}
|
||||
log.debug("请求体: {}", requestBody);
|
||||
|
||||
// 设置请求头
|
||||
HttpHeaders headers = new HttpHeaders();
|
||||
headers.setContentType(MediaType.APPLICATION_JSON);
|
||||
headers.set("Authorization", "Bearer " + apiKey);
|
||||
|
||||
HttpEntity<Map<String, Object>> requestEntity = new HttpEntity<>(requestBody, headers);
|
||||
|
||||
// 发送POST请求
|
||||
log.info("发送POST请求到RAGFlow API...");
|
||||
ResponseEntity<String> response = restTemplate.exchange(url, HttpMethod.POST, requestEntity, String.class);
|
||||
|
||||
log.info("RAGFlow API响应状态码: {}", response.getStatusCode());
|
||||
log.debug("RAGFlow API响应内容: {}", response.getBody());
|
||||
|
||||
if (!response.getStatusCode().is2xxSuccessful()) {
|
||||
log.error("RAGFlow API调用失败，状态码: {}, 响应内容: {}", response.getStatusCode(), response.getBody());
|
||||
throw new RenException(ErrorCode.RAG_API_ERROR, response.getStatusCode().toString());
|
||||
}
|
||||
|
||||
// 解析响应体，提取datasetId
|
||||
String responseBody = response.getBody();
|
||||
if (StringUtils.isNotBlank(responseBody)) {
|
||||
try {
|
||||
// 解析RAGFlow API响应，支持多种可能的字段名
|
||||
ObjectMapper objectMapper = new ObjectMapper();
|
||||
Map<String, Object> responseMap = objectMapper.readValue(responseBody, Map.class);
|
||||
|
||||
log.debug("RAGFlow API响应解析结果: {}", responseMap);
|
||||
|
||||
// 首先检查响应码
|
||||
Integer code = (Integer) responseMap.get("code");
|
||||
String message = (String) responseMap.get("message");
|
||||
if (code != null && code == 0) {
|
||||
// 响应码为0表示成功，从data字段中获取datasetId
|
||||
Object dataObj = responseMap.get("data");
|
||||
if (dataObj instanceof Map) {
|
||||
Map<String, Object> dataMap = (Map<String, Object>) dataObj;
|
||||
datasetId = (String) dataMap.get("id");
|
||||
|
||||
if (StringUtils.isBlank(datasetId)) {
|
||||
// 如果id字段为空，尝试其他可能的字段名
|
||||
datasetId = (String) dataMap.get("dataset_id");
|
||||
datasetId = (String) dataMap.get("datasetId");
|
||||
}
|
||||
}
|
||||
} else {
|
||||
// 如果响应码不为0，说明API调用失败
|
||||
log.error("RAGFlow API调用失败，响应码: {}, 响应内容: {}", code, message);
|
||||
throw new RenException(ErrorCode.RAG_API_ERROR,
|
||||
"RAGFlow API调用失败，响应码: " + code + ", 消息: " + message);
|
||||
}
|
||||
|
||||
log.info("从RAGFlow API响应中解析出datasetId: {}", datasetId);
|
||||
log.debug("完整响应内容: {}", responseBody);
|
||||
} catch (Exception e) {
|
||||
log.error("解析RAGFlow API响应失败: {}", e.getMessage(), e);
|
||||
throw new RenException(ErrorCode.RAG_API_ERROR, "解析响应失败: " + e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
if (StringUtils.isBlank(datasetId)) {
|
||||
log.error("无法从RAGFlow API响应中获取datasetId，响应内容: {}", responseBody);
|
||||
throw new RenException(ErrorCode.RAG_API_ERROR, "RAGFlow API响应中未包含datasetId");
|
||||
}
|
||||
log.info("RAGFlow数据集创建成功，datasetId: {}", datasetId);
|
||||
|
||||
return datasetId;
|
||||
}
|
||||
|
||||
/**
|
||||
* 调用RAGFlow API更新数据集
|
||||
*/
|
||||
private void updateDatasetInRAGFlow(String datasetId, String name, String description,
|
||||
Map<String, Object> ragConfig) {
|
||||
String baseUrl = (String) ragConfig.get("base_url");
|
||||
String apiKey = (String) ragConfig.get("api_key");
|
||||
|
||||
log.info("开始调用RAGFlow API更新数据集，datasetId: {}, name: {}", datasetId, name);
|
||||
log.debug("RAGFlow配置 - baseUrl: {}, apiKey: {}", baseUrl, StringUtils.isBlank(apiKey) ? "未配置" : "已配置");
|
||||
|
||||
// 构建请求URL
|
||||
String url = baseUrl + "/api/v1/datasets/" + datasetId;
|
||||
log.debug("请求URL: {}", url);
|
||||
// 构建请求体
|
||||
Map<String, Object> requestBody = new HashMap<>();
|
||||
requestBody.put("dataset_id", datasetId);
|
||||
String username = SecurityUser.getUser().getUsername();
|
||||
requestBody.put("name", username + "_" + name);
|
||||
if (StringUtils.isNotBlank(description)) {
|
||||
requestBody.put("description", description);
|
||||
}
|
||||
log.debug("请求体: {}", requestBody);
|
||||
|
||||
// 设置请求头
|
||||
HttpHeaders headers = new HttpHeaders();
|
||||
headers.setContentType(MediaType.APPLICATION_JSON);
|
||||
headers.set("Authorization", "Bearer " + apiKey);
|
||||
|
||||
HttpEntity<Map<String, Object>> requestEntity = new HttpEntity<>(requestBody, headers);
|
||||
|
||||
// 发送PUT请求
|
||||
log.info("发送PUT请求到RAGFlow API...");
|
||||
ResponseEntity<String> response = restTemplate.exchange(url, HttpMethod.PUT, requestEntity, String.class);
|
||||
|
||||
log.info("RAGFlow API响应状态码: {}", response.getStatusCode());
|
||||
log.debug("RAGFlow API响应内容: {}", response.getBody());
|
||||
|
||||
if (!response.getStatusCode().is2xxSuccessful()) {
|
||||
log.error("RAGFlow API调用失败，状态码: {}, 响应内容: {}", response.getStatusCode(), response.getBody());
|
||||
throw new RenException(ErrorCode.RAG_API_ERROR);
|
||||
}
|
||||
|
||||
log.info("RAGFlow数据集更新成功，datasetId: {}", datasetId);
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* 调用RAGFlow API删除数据集
|
||||
*/
|
||||
private void deleteDatasetInRAGFlow(String datasetId, Map<String, Object> ragConfig) {
|
||||
String baseUrl = (String) ragConfig.get("base_url");
|
||||
String apiKey = (String) ragConfig.get("api_key");
|
||||
|
||||
log.info("开始调用RAGFlow API删除数据集，datasetId: {}", datasetId);
|
||||
log.debug("RAGFlow配置 - baseUrl: {}, apiKey: {}", baseUrl, StringUtils.isBlank(apiKey) ? "未配置" : "已配置");
|
||||
|
||||
// 构建请求URL
|
||||
String url = baseUrl + "/api/v1/datasets";
|
||||
log.debug("请求URL: {}", url);
|
||||
|
||||
// 构建请求体
|
||||
Map<String, Object> requestBody = new HashMap<>();
|
||||
requestBody.put("ids", List.of(datasetId));
|
||||
log.debug("请求体: {}", requestBody);
|
||||
|
||||
// 设置请求头
|
||||
HttpHeaders headers = new HttpHeaders();
|
||||
headers.setContentType(MediaType.APPLICATION_JSON);
|
||||
headers.set("Authorization", "Bearer " + apiKey);
|
||||
|
||||
HttpEntity<Map<String, Object>> requestEntity = new HttpEntity<>(requestBody, headers);
|
||||
|
||||
// 发送DELETE请求
|
||||
log.info("发送DELETE请求到RAGFlow API...");
|
||||
ResponseEntity<String> response = restTemplate.exchange(url, HttpMethod.DELETE, requestEntity,
|
||||
String.class);
|
||||
|
||||
log.info("RAGFlow API响应状态码: {}", response.getStatusCode());
|
||||
log.debug("RAGFlow API响应内容: {}", response.getBody());
|
||||
|
||||
if (!response.getStatusCode().is2xxSuccessful()) {
|
||||
log.error("RAGFlow API调用失败，状态码: {}, 响应内容: {}", response.getStatusCode(), response.getBody());
|
||||
throw new RenException(ErrorCode.RAG_API_ERROR);
|
||||
}
|
||||
log.info("RAGFlow数据集删除成功，datasetId: {}", datasetId);
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取RAG配置并验证
|
||||
*/
|
||||
private Map<String, Object> getValidatedRAGConfig(String ragModelId) {
|
||||
Map<String, Object> ragConfig;
|
||||
if (StringUtils.isNotBlank(ragModelId)) {
|
||||
ragConfig = getRAGConfig(ragModelId);
|
||||
} else {
|
||||
ragConfig = getDefaultRAGConfig();
|
||||
}
|
||||
|
||||
// 验证baseUrl和apiKey
|
||||
validateRAGConfigParameters(ragConfig);
|
||||
|
||||
return ragConfig;
|
||||
}
|
||||
|
||||
/**
|
||||
* 验证RAG配置参数
|
||||
*/
|
||||
private void validateRAGConfigParameters(Map<String, Object> ragConfig) {
|
||||
if (ragConfig == null) {
|
||||
throw new RenException(ErrorCode.RAG_CONFIG_NOT_FOUND, "RAG配置为空，请检查配置");
|
||||
}
|
||||
|
||||
// 从配置中提取必要的参数
|
||||
String baseUrl = (String) ragConfig.get("base_url");
|
||||
String apiKey = (String) ragConfig.get("api_key");
|
||||
|
||||
if (StringUtils.isBlank(baseUrl)) {
|
||||
throw new RenException(ErrorCode.RAG_API_ERROR, "RAG配置中base_url为空，请完善配置");
|
||||
}
|
||||
|
||||
if (StringUtils.isBlank(apiKey)) {
|
||||
throw new RenException(ErrorCode.RAG_API_ERROR, "RAG配置中api_key为空，请完善配置");
|
||||
}
|
||||
|
||||
if (apiKey.contains("你")) {
|
||||
throw new RenException(ErrorCode.RAG_API_ERROR, "RAG配置中api_key包含占位符'你'，请替换为实际的API密钥");
|
||||
}
|
||||
|
||||
if (!baseUrl.startsWith("http://") && !baseUrl.startsWith("https://")) {
|
||||
throw new RenException(ErrorCode.RAG_API_ERROR, "RAG配置中base_url格式不正确，必须以http://或https://开头");
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查是否存在同名知识库
|
||||
*
|
||||
* @param knowledgeBaseDTO 知识库DTO
|
||||
* @param excludeId 排除的ID（更新时使用）
|
||||
*/
|
||||
private void checkDuplicateKnowledgeBaseName(KnowledgeBaseDTO knowledgeBaseDTO, String excludeId) {
|
||||
if (StringUtils.isNotBlank(knowledgeBaseDTO.getName())) {
|
||||
Long currentUserId = SecurityUser.getUserId();
|
||||
QueryWrapper<KnowledgeBaseEntity> queryWrapper = new QueryWrapper<KnowledgeBaseEntity>()
|
||||
.eq("name", knowledgeBaseDTO.getName())
|
||||
.eq("creator", currentUserId);
|
||||
|
||||
// 如果提供了排除ID，则排除该记录
|
||||
if (StringUtils.isNotBlank(excludeId)) {
|
||||
queryWrapper.ne("id", excludeId);
|
||||
}
|
||||
|
||||
long count = knowledgeBaseDao.selectCount(queryWrapper);
|
||||
if (count > 0) {
|
||||
throw new RenException(ErrorCode.KNOWLEDGE_BASE_NAME_EXISTS,
|
||||
MessageUtils.getMessage(ErrorCode.KNOWLEDGE_BASE_NAME_EXISTS));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 从RAGFlow API获取知识库的文档数量
|
||||
*/
|
||||
private Integer getDocumentCountFromRAGFlow(String datasetId, String ragModelId) {
|
||||
if (StringUtils.isBlank(datasetId) || StringUtils.isBlank(ragModelId)) {
|
||||
log.warn("datasetId或ragModelId为空，无法获取文档数量");
|
||||
return 0;
|
||||
}
|
||||
|
||||
log.info("开始获取知识库 {} 的文档数量", datasetId);
|
||||
|
||||
// 获取RAG配置
|
||||
Map<String, Object> ragConfig = getValidatedRAGConfig(ragModelId);
|
||||
String baseUrl = (String) ragConfig.get("base_url");
|
||||
String apiKey = (String) ragConfig.get("api_key");
|
||||
|
||||
// 构建请求URL - 调用RAGFlow API获取文档列表，但不返回文档详情，只获取总数
|
||||
String url = baseUrl + "/api/v1/datasets/" + datasetId + "/documents?page=1&size=1";
|
||||
log.debug("请求URL: {}", url);
|
||||
|
||||
// 设置请求头
|
||||
HttpHeaders headers = new HttpHeaders();
|
||||
headers.setContentType(MediaType.APPLICATION_JSON);
|
||||
headers.set("Authorization", "Bearer " + apiKey);
|
||||
|
||||
HttpEntity<String> requestEntity = new HttpEntity<>(headers);
|
||||
|
||||
// 发送GET请求
|
||||
log.info("发送GET请求到RAGFlow API获取文档数量...");
|
||||
ResponseEntity<String> response = restTemplate.exchange(url, HttpMethod.GET, requestEntity, String.class);
|
||||
|
||||
log.info("RAGFlow API响应状态码: {}", response.getStatusCode());
|
||||
|
||||
if (!response.getStatusCode().is2xxSuccessful()) {
|
||||
log.error("RAGFlow API调用失败，状态码: {}, 响应内容: {}", response.getStatusCode(), response.getBody());
|
||||
return 0;
|
||||
}
|
||||
|
||||
String responseBody = response.getBody();
|
||||
log.debug("RAGFlow API响应内容: {}", responseBody);
|
||||
|
||||
// 解析响应
|
||||
try {
|
||||
ObjectMapper objectMapper = new ObjectMapper();
|
||||
Map<String, Object> responseMap = objectMapper.readValue(responseBody, Map.class);
|
||||
Integer code = (Integer) responseMap.get("code");
|
||||
|
||||
if (code != null && code == 0) {
|
||||
Object dataObj = responseMap.get("data");
|
||||
if (dataObj instanceof Map) {
|
||||
Map<String, Object> dataMap = (Map<String, Object>) dataObj;
|
||||
Object totalObj = dataMap.get("total");
|
||||
if (totalObj instanceof Integer) {
|
||||
Integer documentCount = (Integer) totalObj;
|
||||
log.info("获取知识库 {} 的文档数量成功: {}", datasetId, documentCount);
|
||||
return documentCount;
|
||||
} else if (totalObj instanceof Long) {
|
||||
Long documentCount = (Long) totalObj;
|
||||
log.info("获取知识库 {} 的文档数量成功: {}", datasetId, documentCount);
|
||||
return documentCount.intValue();
|
||||
}
|
||||
}
|
||||
} else {
|
||||
log.error("RAGFlow API调用失败，响应码: {}, 响应内容: {}", code, responseBody);
|
||||
}
|
||||
} catch (Exception e) {
|
||||
log.error("解析RAGFlow API响应失败: {}", e.getMessage(), e);
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
}
|
||||
+1788
File diff suppressed because it is too large
Load Diff
+33
-1
@@ -22,6 +22,8 @@ import xiaozhi.common.page.PageData;
|
||||
import xiaozhi.common.service.impl.BaseServiceImpl;
|
||||
import xiaozhi.common.user.UserDetail;
|
||||
import xiaozhi.common.utils.ConvertUtils;
|
||||
import xiaozhi.modules.knowledge.dao.KnowledgeBaseDao;
|
||||
import xiaozhi.modules.knowledge.entity.KnowledgeBaseEntity;
|
||||
import xiaozhi.modules.model.dao.ModelProviderDao;
|
||||
import xiaozhi.modules.model.dto.ModelProviderDTO;
|
||||
import xiaozhi.modules.model.entity.ModelProviderEntity;
|
||||
@@ -34,13 +36,43 @@ public class ModelProviderServiceImpl extends BaseServiceImpl<ModelProviderDao,
|
||||
implements ModelProviderService {
|
||||
|
||||
private final ModelProviderDao modelProviderDao;
|
||||
private final KnowledgeBaseDao knowledgeBaseDao;
|
||||
|
||||
@Override
|
||||
public List<ModelProviderDTO> getPluginList() {
|
||||
// 1. 获取插件列表
|
||||
LambdaQueryWrapper<ModelProviderEntity> queryWrapper = new LambdaQueryWrapper<>();
|
||||
queryWrapper.eq(ModelProviderEntity::getModelType, "Plugin");
|
||||
List<ModelProviderEntity> providerEntities = modelProviderDao.selectList(queryWrapper);
|
||||
return ConvertUtils.sourceToTarget(providerEntities, ModelProviderDTO.class);
|
||||
List<ModelProviderDTO> resultList = ConvertUtils.sourceToTarget(providerEntities, ModelProviderDTO.class);
|
||||
|
||||
// 2. 获取当前用户的知识库列表并追加到结果中
|
||||
UserDetail userDetail = SecurityUser.getUser();
|
||||
if (userDetail != null && userDetail.getId() != null) {
|
||||
// 查询当前用户的知识库
|
||||
LambdaQueryWrapper<KnowledgeBaseEntity> kbQueryWrapper = new LambdaQueryWrapper<>();
|
||||
kbQueryWrapper.eq(KnowledgeBaseEntity::getCreator, userDetail.getId());
|
||||
kbQueryWrapper.eq(KnowledgeBaseEntity::getStatus, 1); // 只获取启用状态的知识库
|
||||
List<KnowledgeBaseEntity> knowledgeBases = knowledgeBaseDao.selectList(kbQueryWrapper);
|
||||
|
||||
// 将知识库转换为ModelProviderDTO格式并添加到结果列表
|
||||
for (KnowledgeBaseEntity kb : knowledgeBases) {
|
||||
ModelProviderDTO dto = new ModelProviderDTO();
|
||||
dto.setId(kb.getId());
|
||||
dto.setModelType("Rag");
|
||||
dto.setName("[知识库]" + kb.getName());
|
||||
dto.setProviderCode("ragflow"); // 假设所有RAG都使用ragflow
|
||||
dto.setFields("[]");
|
||||
dto.setSort(0);
|
||||
dto.setCreateDate(kb.getCreatedAt());
|
||||
dto.setUpdateDate(kb.getUpdatedAt());
|
||||
dto.setCreator(0L);
|
||||
dto.setUpdater(0L);
|
||||
resultList.add(dto);
|
||||
}
|
||||
}
|
||||
|
||||
return resultList;
|
||||
}
|
||||
|
||||
@Override
|
||||
|
||||
@@ -0,0 +1,24 @@
|
||||
-- 添加RAG模型供应器和配置
|
||||
-- -------------------------------------------------------
|
||||
|
||||
-- 添加RAG模型供应器
|
||||
delete from `ai_model_provider` where id = 'SYSTEM_RAG_ragflow';
|
||||
INSERT INTO `ai_model_provider` (`id`, `model_type`, `provider_code`, `name`, `fields`, `sort`, `creator`, `create_date`, `updater`, `update_date`) VALUES
|
||||
('SYSTEM_RAG_ragflow', 'RAG', 'ragflow', 'RAGFlow', '[{"key": "base_url", "type": "string", "label": "服务地址"}, {"key": "api_key", "type": "string", "label": "API密钥"}]', 1, 1, NOW(), 1, NOW());
|
||||
|
||||
-- 添加RAG模型配置
|
||||
delete from `ai_model_config` where id = 'RAG_RAGFlow';
|
||||
INSERT INTO `ai_model_config` VALUES ('RAG_RAGFlow', 'RAG', 'ragflow', 'RAGFlow', 1, 1, '{"type": "ragflow", "base_url": "http://localhost", "api_key": "你的RAG密钥"}', 'https://github.com/infiniflow/ragflow/blob/main/README_zh.md', 'RAGFlow配置说明：
|
||||
一、快速部署教程（docker部署）
|
||||
1.$ sysctl vm.max_map_count
|
||||
2.$ sysctl -w vm.max_map_count=262144
|
||||
3.$ git clone https://github.com/infiniflow/ragflow.git
|
||||
4.docker compose -f docker-compose.yml up -d
|
||||
5.$ docker logs -f docker-ragflow-cpu-1
|
||||
6.注冊登录后，点击右上角头像，获得RAGFlow的API KEY和API服务器地址。使用RAGFlow前请在Model Provider中添加模型和设置默认模型。
|
||||
二、如果您希望关掉注册功能
|
||||
1.停止服务 docker compose down
|
||||
2. sed -i ''s/REGISTER_ENABLED=1/REGISTER_ENABLED=0/g'' .env
|
||||
3.cat .env | grep -i register
|
||||
4.看到REGISTER_ENABLED=0 重启服务即可。', 1, NULL, NULL, NULL, NULL);
|
||||
|
||||
@@ -0,0 +1,19 @@
|
||||
-- 知识库表
|
||||
DROP TABLE IF EXISTS `ai_rag_dataset`;
|
||||
CREATE TABLE `ai_rag_dataset` (
|
||||
`id` VARCHAR(32) NOT NULL COMMENT '唯一标识',
|
||||
`dataset_id` VARCHAR(64) NOT NULL COMMENT '知识库ID',
|
||||
`rag_model_id` VARCHAR(64) COMMENT 'RAG模型配置ID',
|
||||
`name` VARCHAR(100) NOT NULL COMMENT '知识库名称',
|
||||
`description` TEXT COMMENT '知识库描述',
|
||||
`status` TINYINT(1) DEFAULT 1 COMMENT '状态：0停用 1启用',
|
||||
`creator` BIGINT COMMENT '创建者',
|
||||
`created_at` DATETIME COMMENT '创建时间',
|
||||
`updater` BIGINT COMMENT '更新者',
|
||||
`updated_at` DATETIME COMMENT '更新时间',
|
||||
PRIMARY KEY (`id`),
|
||||
UNIQUE KEY `uk_dataset_id` (`dataset_id`),
|
||||
INDEX `idx_ai_rag_dataset_status` (`status`),
|
||||
INDEX `idx_ai_rag_dataset_creator` (`creator`),
|
||||
INDEX `idx_ai_rag_dataset_created_at` (`created_at`)
|
||||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='知识库表';
|
||||
@@ -402,3 +402,17 @@ databaseChangeLog:
|
||||
- sqlFile:
|
||||
encoding: utf8
|
||||
path: classpath:db/changelog/202510191042.sql
|
||||
- changeSet:
|
||||
id: 202510250956
|
||||
author: rainv123
|
||||
changes:
|
||||
- sqlFile:
|
||||
encoding: utf8
|
||||
path: classpath:db/changelog/202510250956.sql
|
||||
- changeSet:
|
||||
id: 202510251150
|
||||
author: rainv123
|
||||
changes:
|
||||
- sqlFile:
|
||||
encoding: utf8
|
||||
path: classpath:db/changelog/202510251150.sql
|
||||
@@ -168,4 +168,12 @@
|
||||
10159=\u97F3\u8272ID\u5DF2\u5B58\u5728
|
||||
10160=\u706B\u5C71\u5F15\u64CE\u97F3\u8272ID\u683C\u5F0F\u9519\u8BEF\uFF0C\u5FC5\u987B\u4EE5S_\u5F00\u5934
|
||||
10161=Mac\u5730\u5740\u5DF2\u5B58\u5728
|
||||
10162=\u6A21\u578B\u4F9B\u5E94\u5668\u4E0D\u5B58\u5728
|
||||
10162=\u6A21\u578B\u4F9B\u5E94\u5668\u4E0D\u5B58\u5728
|
||||
10163=\u77E5\u8BC6\u5E93\u8BB0\u5F55\u4E0D\u5B58\u5728
|
||||
10164=RAG\u914D\u7F6E\u672A\u627E\u5230
|
||||
10165=RAG\u914D\u7F6E\u7C7B\u578B\u9519\u8BEF
|
||||
10166=\u9ED8\u8BA4RAG\u914D\u7F6E\u672A\u627E\u5230
|
||||
10167=RAG\u63A5\u53E3\u9519\u8BEF\uFF0C{0}
|
||||
10168=\u4E0A\u4F20\u6587\u4EF6\u5931\u8D25
|
||||
10169=\u60A8\u6CA1\u6709\u6743\u9650\u64CD\u4F5C\u8BE5\u8BB0\u5F55
|
||||
10170=\u77E5\u8BC6\u5E93\u540D\u79F0\u91CD\u590D
|
||||
@@ -168,4 +168,12 @@
|
||||
10159=Voice ID already exists
|
||||
10160=Huoshan Engine voice ID format error, must start with S_
|
||||
10161=Mac address already exists
|
||||
10162=Model provider does not exist
|
||||
10162=Model provider does not exist
|
||||
10163=Knowledge base record does not exist
|
||||
10164=RAG configuration not found
|
||||
10165=RAG configuration type error
|
||||
10166=Default RAG configuration not found
|
||||
10167=RAG API interface error: {0}
|
||||
10168=Upload file failed
|
||||
10169=No permission to operate this knowledge base
|
||||
10170=Knowledge base name already exists
|
||||
@@ -168,4 +168,12 @@
|
||||
10159=\u97F3\u8272ID\u5DF2\u5B58\u5728
|
||||
10160=\u706B\u5C71\u5F15\u64CE\u97F3\u8272ID\u683C\u5F0F\u9519\u8BEF\uFF0C\u5FC5\u987B\u4EE5S_\u5F00\u5934
|
||||
10161=Mac\u5730\u5740\u5DF2\u5B58\u5728
|
||||
10162=\u6A21\u578B\u4F9B\u5E94\u5668\u4E0D\u5B58\u5728
|
||||
10162=\u6A21\u578B\u4F9B\u5E94\u5668\u4E0D\u5B58\u5728
|
||||
10163=\u77E5\u8BC6\u5E93\u8BB0\u5F55\u4E0D\u5B58\u5728
|
||||
10164=RAG\u914D\u7F6E\u672A\u627E\u5230
|
||||
10165=RAG\u914D\u7F6E\u7C7B\u578B\u9519\u8BEF
|
||||
10166=\u9ED8\u8BA4RAG\u914D\u7F6E\u672A\u627E\u5230
|
||||
10167=RAG\u63A5\u53E3\u9519\u8BEF\uFF0C{0}
|
||||
10168=\u4E0A\u4F20\u6587\u4EF6\u5931\u8D25
|
||||
10169=\u60A8\u6CA1\u6709\u6743\u9650\u64CD\u4F5C\u8BE5\u8BB0\u5F55
|
||||
10170=\u77E5\u8BC6\u5E93\u540D\u79F0\u91CD\u590D
|
||||
@@ -169,4 +169,11 @@
|
||||
10160=\u706B\u5C71\u5F15\u64CE\u97F3\u8272ID\u683C\u5F0F\u932F\u8AA4\uFF0C\u5FC5\u9808\u4EE5S_\u958B\u982D
|
||||
10161=Mac\u5730\u5740\u5DF2\u5B58\u5728
|
||||
10162=\u6A21\u578B\u63D0\u4F9B\u5546\u4E0D\u5B58\u5728
|
||||
|
||||
10163=\u77E5\u8B58\u5EAB\u8A18\u9304\u4E0D\u5B58\u5728
|
||||
10164=RAG\u914D\u7F6E\u672A\u627E\u5230
|
||||
10165=RAG\u914D\u7F6E\u985E\u578B\u932F\u8AA4
|
||||
10166=\u9810\u8A2DRAG\u914D\u7F6E\u672A\u627E\u5230
|
||||
10167=RAG\u63A5\u53E3\u932F\u8AA4\uFF0C{0}
|
||||
10168=\u4E0A\u50B3\u6587\u4EF6\u5931\u6557
|
||||
10169=\u60A8\u6C92\u6709\u6B0A\u9650\u64CD\u4F5C\u8A72\u8A18\u9304
|
||||
10170=\u77E5\u8B58\u5EAB\u540D\u7A31\u91CD\u8907
|
||||
|
||||
@@ -235,7 +235,7 @@ function showAbout() {
|
||||
title: t('settings.aboutApp', { appName: import.meta.env.VITE_APP_TITLE }),
|
||||
content: t('settings.aboutContent', {
|
||||
appName: import.meta.env.VITE_APP_TITLE,
|
||||
version: '0.8.6'
|
||||
version: '0.8.7'
|
||||
}),
|
||||
showCancel: false,
|
||||
confirmText: t('common.confirm'),
|
||||
|
||||
@@ -9,6 +9,7 @@ import timbre from "./module/timbre.js"
|
||||
import user from './module/user.js'
|
||||
import voiceClone from './module/voiceClone.js'
|
||||
import voiceResource from './module/voiceResource.js'
|
||||
import knowledgeBase from './module/knowledgeBase.js'
|
||||
|
||||
|
||||
|
||||
@@ -39,5 +40,6 @@ export default {
|
||||
ota,
|
||||
dict,
|
||||
voiceResource,
|
||||
voiceClone
|
||||
}
|
||||
voiceClone,
|
||||
knowledgeBase
|
||||
}
|
||||
|
||||
@@ -0,0 +1,318 @@
|
||||
import { getServiceUrl } from '../api';
|
||||
import RequestService from '../httpRequest';
|
||||
|
||||
/**
|
||||
* 获取认证token
|
||||
*/
|
||||
function getAuthToken() {
|
||||
return localStorage.getItem('token') || '';
|
||||
}
|
||||
|
||||
/**
|
||||
* 通用API请求包装器
|
||||
* @param {Object} config - 请求配置
|
||||
* @param {string} config.url - 请求URL
|
||||
* @param {string} config.method - 请求方法
|
||||
* @param {Object} [config.data] - 请求数据
|
||||
* @param {Object} [config.headers] - 额外请求头
|
||||
* @param {Function} config.callback - 成功回调
|
||||
* @param {Function} [config.errorCallback] - 错误回调
|
||||
* @param {string} [config.errorMessage] - 错误消息
|
||||
* @param {Function} [config.retryFunction] - 重试函数
|
||||
*/
|
||||
function makeApiRequest(config) {
|
||||
const token = getAuthToken();
|
||||
const { url, method, data, headers, callback, errorCallback, errorMessage, retryFunction } = config;
|
||||
|
||||
const requestBuilder = RequestService.sendRequest()
|
||||
.url(url)
|
||||
.method(method)
|
||||
.header({
|
||||
'Authorization': `Bearer ${token}`,
|
||||
...headers
|
||||
});
|
||||
|
||||
if (data) {
|
||||
requestBuilder.data(data);
|
||||
}
|
||||
|
||||
requestBuilder
|
||||
.success((res) => {
|
||||
RequestService.clearRequestTime();
|
||||
callback(res);
|
||||
})
|
||||
.fail((err) => {
|
||||
console.error(errorMessage || '操作失败', err);
|
||||
if (errorCallback) {
|
||||
errorCallback(err);
|
||||
}
|
||||
})
|
||||
.networkFail(() => {
|
||||
if (retryFunction) {
|
||||
RequestService.reAjaxFun(() => {
|
||||
retryFunction();
|
||||
});
|
||||
}
|
||||
}).send();
|
||||
}
|
||||
|
||||
/**
|
||||
* 知识库管理相关API
|
||||
*/
|
||||
export default {
|
||||
/**
|
||||
* 获取知识库列表
|
||||
* @param {Object} params - 查询参数
|
||||
* @param {Function} callback - 回调函数
|
||||
* @param {Function} errorCallback - 错误回调
|
||||
*/
|
||||
getKnowledgeBaseList(params, callback, errorCallback) {
|
||||
const queryParams = new URLSearchParams({
|
||||
page: params.page,
|
||||
page_size: params.page_size,
|
||||
name: params.name || ''
|
||||
}).toString();
|
||||
|
||||
makeApiRequest({
|
||||
url: `${getServiceUrl()}/datasets?${queryParams}`,
|
||||
method: 'GET',
|
||||
callback: callback,
|
||||
errorCallback: errorCallback,
|
||||
errorMessage: '获取知识库列表失败',
|
||||
retryFunction: () => this.getKnowledgeBaseList(params, callback, errorCallback)
|
||||
});
|
||||
},
|
||||
|
||||
/**
|
||||
* 创建知识库
|
||||
* @param {Object} data - 知识库数据
|
||||
* @param {Function} callback - 回调函数
|
||||
* @param {Function} errorCallback - 错误回调
|
||||
*/
|
||||
createKnowledgeBase(data, callback, errorCallback) {
|
||||
console.log('createKnowledgeBase called with data:', data);
|
||||
console.log('API URL:', `${getServiceUrl()}/datasets`);
|
||||
|
||||
makeApiRequest({
|
||||
url: `${getServiceUrl()}/datasets`,
|
||||
method: 'POST',
|
||||
data: data,
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
callback: (res) => {
|
||||
console.log('createKnowledgeBase success response:', res);
|
||||
callback(res);
|
||||
},
|
||||
errorCallback: (err) => {
|
||||
console.error('创建知识库失败:', err);
|
||||
if (err.response) {
|
||||
console.error('Error response data:', err.response.data);
|
||||
console.error('Error response status:', err.response.status);
|
||||
}
|
||||
if (errorCallback) {
|
||||
errorCallback(err);
|
||||
}
|
||||
},
|
||||
errorMessage: '创建知识库失败',
|
||||
retryFunction: () => this.createKnowledgeBase(data, callback, errorCallback)
|
||||
});
|
||||
},
|
||||
|
||||
/**
|
||||
* 更新知识库
|
||||
* @param {string} datasetId - 知识库ID
|
||||
* @param {Object} data - 更新数据
|
||||
* @param {Function} callback - 回调函数
|
||||
* @param {Function} errorCallback - 错误回调
|
||||
*/
|
||||
updateKnowledgeBase(datasetId, data, callback, errorCallback) {
|
||||
console.log('updateKnowledgeBase called with datasetId:', datasetId, 'data:', data);
|
||||
console.log('API URL:', `${getServiceUrl()}/datasets/${datasetId}`);
|
||||
|
||||
makeApiRequest({
|
||||
url: `${getServiceUrl()}/datasets/${datasetId}`,
|
||||
method: 'PUT',
|
||||
data: data,
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
callback: callback,
|
||||
errorCallback: errorCallback,
|
||||
errorMessage: '更新知识库失败',
|
||||
retryFunction: () => this.updateKnowledgeBase(datasetId, data, callback, errorCallback)
|
||||
});
|
||||
},
|
||||
|
||||
/**
|
||||
* 删除单个知识库
|
||||
* @param {string} datasetId - 知识库ID
|
||||
* @param {Function} callback - 回调函数
|
||||
* @param {Function} errorCallback - 错误回调
|
||||
*/
|
||||
deleteKnowledgeBase(datasetId, callback, errorCallback) {
|
||||
console.log('deleteKnowledgeBase called with datasetId:', datasetId);
|
||||
console.log('API URL:', `${getServiceUrl()}/datasets/${datasetId}`);
|
||||
|
||||
makeApiRequest({
|
||||
url: `${getServiceUrl()}/datasets/${datasetId}`,
|
||||
method: 'DELETE',
|
||||
callback: callback,
|
||||
errorCallback: errorCallback,
|
||||
errorMessage: '删除知识库失败',
|
||||
retryFunction: () => this.deleteKnowledgeBase(datasetId, callback, errorCallback)
|
||||
});
|
||||
},
|
||||
|
||||
/**
|
||||
* 批量删除知识库
|
||||
* @param {string|Array} ids - 知识库ID字符串或数组
|
||||
* @param {Function} callback - 回调函数
|
||||
* @param {Function} errorCallback - 错误回调
|
||||
*/
|
||||
deleteKnowledgeBases(ids, callback, errorCallback) {
|
||||
// 确保ids是正确格式的字符串
|
||||
const idsStr = Array.isArray(ids) ? ids.join(',') : ids;
|
||||
|
||||
makeApiRequest({
|
||||
url: `${getServiceUrl()}/datasets/batch?ids=${idsStr}`,
|
||||
method: 'DELETE',
|
||||
callback: callback,
|
||||
errorCallback: errorCallback,
|
||||
errorMessage: '批量删除知识库失败',
|
||||
retryFunction: () => this.deleteKnowledgeBases(ids, callback, errorCallback)
|
||||
});
|
||||
},
|
||||
|
||||
/**
|
||||
* 获取文档列表
|
||||
* @param {string} datasetId - 知识库ID
|
||||
* @param {Object} params - 查询参数
|
||||
* @param {Function} callback - 回调函数
|
||||
* @param {Function} errorCallback - 错误回调
|
||||
*/
|
||||
getDocumentList(datasetId, params, callback, errorCallback) {
|
||||
const queryParams = new URLSearchParams({
|
||||
page: params.page,
|
||||
page_size: params.page_size,
|
||||
name: params.name || ''
|
||||
}).toString();
|
||||
|
||||
makeApiRequest({
|
||||
url: `${getServiceUrl()}/datasets/${datasetId}/documents?${queryParams}`,
|
||||
method: 'GET',
|
||||
callback: callback,
|
||||
errorCallback: errorCallback,
|
||||
errorMessage: '获取文档列表失败',
|
||||
retryFunction: () => this.getDocumentList(datasetId, params, callback, errorCallback)
|
||||
});
|
||||
},
|
||||
|
||||
/**
|
||||
* 上传文档
|
||||
* @param {string} datasetId - 知识库ID
|
||||
* @param {Object} formData - 表单数据
|
||||
* @param {Function} callback - 回调函数
|
||||
* @param {Function} errorCallback - 错误回调
|
||||
*/
|
||||
uploadDocument(datasetId, formData, callback, errorCallback) {
|
||||
makeApiRequest({
|
||||
url: `${getServiceUrl()}/datasets/${datasetId}/documents`,
|
||||
method: 'POST',
|
||||
data: formData,
|
||||
headers: { 'Content-Type': 'multipart/form-data' },
|
||||
callback: callback,
|
||||
errorCallback: errorCallback,
|
||||
errorMessage: '上传文档失败',
|
||||
retryFunction: () => this.uploadDocument(datasetId, formData, callback, errorCallback)
|
||||
});
|
||||
},
|
||||
|
||||
/**
|
||||
* 解析文档
|
||||
* @param {string} datasetId - 知识库ID
|
||||
* @param {string} documentId - 文档ID
|
||||
* @param {Function} callback - 回调函数
|
||||
* @param {Function} errorCallback - 错误回调
|
||||
*/
|
||||
parseDocument(datasetId, documentId, callback, errorCallback) {
|
||||
const requestBody = {
|
||||
document_ids: [documentId]
|
||||
};
|
||||
|
||||
makeApiRequest({
|
||||
url: `${getServiceUrl()}/datasets/${datasetId}/chunks`,
|
||||
method: 'POST',
|
||||
data: requestBody,
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
callback: callback,
|
||||
errorCallback: errorCallback,
|
||||
errorMessage: '解析文档失败',
|
||||
retryFunction: () => this.parseDocument(datasetId, documentId, callback, errorCallback)
|
||||
});
|
||||
},
|
||||
|
||||
/**
|
||||
* 删除文档
|
||||
* @param {string} datasetId - 知识库ID
|
||||
* @param {string} documentId - 文档ID
|
||||
* @param {Function} callback - 回调函数
|
||||
* @param {Function} errorCallback - 错误回调
|
||||
*/
|
||||
deleteDocument(datasetId, documentId, callback, errorCallback) {
|
||||
makeApiRequest({
|
||||
url: `${getServiceUrl()}/datasets/${datasetId}/documents/${documentId}`,
|
||||
method: 'DELETE',
|
||||
callback: callback,
|
||||
errorCallback: errorCallback,
|
||||
errorMessage: '删除文档失败',
|
||||
retryFunction: () => this.deleteDocument(datasetId, documentId, callback, errorCallback)
|
||||
});
|
||||
},
|
||||
|
||||
/**
|
||||
* 获取文档切片列表
|
||||
* @param {string} datasetId - 知识库ID
|
||||
* @param {string} documentId - 文档ID
|
||||
* @param {Object} params - 查询参数
|
||||
* @param {Function} callback - 回调函数
|
||||
* @param {Function} errorCallback - 错误回调
|
||||
*/
|
||||
listChunks(datasetId, documentId, params, callback, errorCallback) {
|
||||
let queryParams = new URLSearchParams({
|
||||
page: params.page || 1,
|
||||
page_size: params.page_size || 10
|
||||
}).toString();
|
||||
|
||||
// 添加关键词搜索参数
|
||||
if (params.keywords) {
|
||||
queryParams += `&keywords=${encodeURIComponent(params.keywords)}`;
|
||||
}
|
||||
|
||||
makeApiRequest({
|
||||
url: `${getServiceUrl()}/datasets/${datasetId}/documents/${documentId}/chunks?${queryParams}`,
|
||||
method: 'GET',
|
||||
callback: callback,
|
||||
errorCallback: errorCallback,
|
||||
errorMessage: '获取切片列表失败',
|
||||
retryFunction: () => this.listChunks(datasetId, documentId, params, callback, errorCallback)
|
||||
});
|
||||
},
|
||||
|
||||
/**
|
||||
* 召回测试
|
||||
* @param {string} datasetId - 知识库ID
|
||||
* @param {Object} data - 召回测试参数
|
||||
* @param {Function} callback - 回调函数
|
||||
* @param {Function} errorCallback - 错误回调
|
||||
*/
|
||||
retrievalTest(datasetId, data, callback, errorCallback) {
|
||||
makeApiRequest({
|
||||
url: `${getServiceUrl()}/datasets/${datasetId}/retrieval-test`,
|
||||
method: 'POST',
|
||||
data: data,
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
callback: callback,
|
||||
errorCallback: errorCallback,
|
||||
errorMessage: '召回测试失败',
|
||||
retryFunction: () => this.retrievalTest(datasetId, data, callback, errorCallback)
|
||||
});
|
||||
}
|
||||
|
||||
};
|
||||
@@ -337,5 +337,23 @@ export default {
|
||||
this.getPluginFunctionList(params, callback)
|
||||
})
|
||||
}).send()
|
||||
},
|
||||
|
||||
// 获取RAG模型列表
|
||||
getRAGModels(callback) {
|
||||
RequestService.sendRequest()
|
||||
.url(`${getServiceUrl()}/datasets/rag-models`)
|
||||
.method('GET')
|
||||
.success((res) => {
|
||||
RequestService.clearRequestTime()
|
||||
callback(res)
|
||||
})
|
||||
.networkFail((err) => {
|
||||
console.error('获取RAG模型列表失败:', err)
|
||||
this.$message.error(err.msg || '获取RAG模型列表失败')
|
||||
RequestService.reAjaxFun(() => {
|
||||
this.getRAGModels(callback)
|
||||
})
|
||||
}).send()
|
||||
}
|
||||
}
|
||||
|
||||
Binary file not shown.
|
After Width: | Height: | Size: 2.8 KiB |
@@ -9,82 +9,50 @@
|
||||
|
||||
<!-- 中间导航菜单 -->
|
||||
<div class="header-center">
|
||||
<div
|
||||
class="equipment-management"
|
||||
:class="{
|
||||
'active-tab':
|
||||
<div class="equipment-management" :class="{
|
||||
'active-tab':
|
||||
$route.path === '/home' ||
|
||||
$route.path === '/role-config' ||
|
||||
$route.path === '/device-management',
|
||||
}" @click="goHome">
|
||||
<img loading="lazy" alt="" src="@/assets/header/robot.png" :style="{
|
||||
filter:
|
||||
$route.path === '/home' ||
|
||||
$route.path === '/role-config' ||
|
||||
$route.path === '/device-management',
|
||||
}"
|
||||
@click="goHome"
|
||||
>
|
||||
<img
|
||||
loading="lazy"
|
||||
alt=""
|
||||
src="@/assets/header/robot.png"
|
||||
:style="{
|
||||
filter:
|
||||
$route.path === '/home' ||
|
||||
$route.path === '/role-config' ||
|
||||
$route.path === '/device-management'
|
||||
? 'brightness(0) invert(1)'
|
||||
: 'None',
|
||||
}"
|
||||
/>
|
||||
? 'brightness(0) invert(1)'
|
||||
: 'None',
|
||||
}" />
|
||||
<span class="nav-text">{{ $t("header.smartManagement") }}</span>
|
||||
</div>
|
||||
<!-- 普通用户显示音色克隆 -->
|
||||
<div
|
||||
v-if="!isSuperAdmin"
|
||||
class="equipment-management"
|
||||
:class="{ 'active-tab': $route.path === '/voice-clone-management' }"
|
||||
@click="goVoiceCloneManagement"
|
||||
>
|
||||
<img
|
||||
loading="lazy"
|
||||
alt=""
|
||||
src="@/assets/header/voice.png"
|
||||
:style="{
|
||||
filter:
|
||||
$route.path === '/voice-clone-management'
|
||||
? 'brightness(0) invert(1)'
|
||||
: 'None',
|
||||
}"
|
||||
/>
|
||||
<div v-if="!isSuperAdmin" class="equipment-management"
|
||||
:class="{ 'active-tab': $route.path === '/voice-clone-management' }" @click="goVoiceCloneManagement">
|
||||
<img loading="lazy" alt="" src="@/assets/header/voice.png" :style="{
|
||||
filter:
|
||||
$route.path === '/voice-clone-management'
|
||||
? 'brightness(0) invert(1)'
|
||||
: 'None',
|
||||
}" />
|
||||
<span class="nav-text">{{ $t("header.voiceCloneManagement") }}</span>
|
||||
</div>
|
||||
|
||||
<!-- 超级管理员显示音色克隆下拉菜单 -->
|
||||
<el-dropdown
|
||||
v-if="isSuperAdmin"
|
||||
trigger="click"
|
||||
class="equipment-management more-dropdown"
|
||||
:class="{
|
||||
'active-tab':
|
||||
$route.path === '/voice-clone-management' ||
|
||||
$route.path === '/voice-resource-management',
|
||||
}"
|
||||
@visible-change="handleVoiceCloneDropdownVisibleChange"
|
||||
>
|
||||
<el-dropdown v-if="isSuperAdmin" trigger="click" class="equipment-management more-dropdown" :class="{
|
||||
'active-tab':
|
||||
$route.path === '/voice-clone-management' ||
|
||||
$route.path === '/voice-resource-management',
|
||||
}" @visible-change="handleVoiceCloneDropdownVisibleChange">
|
||||
<span class="el-dropdown-link">
|
||||
<img
|
||||
loading="lazy"
|
||||
alt=""
|
||||
src="@/assets/header/voice.png"
|
||||
:style="{
|
||||
filter:
|
||||
$route.path === '/voice-clone-management' ||
|
||||
<img loading="lazy" alt="" src="@/assets/header/voice.png" :style="{
|
||||
filter:
|
||||
$route.path === '/voice-clone-management' ||
|
||||
$route.path === '/voice-resource-management'
|
||||
? 'brightness(0) invert(1)'
|
||||
: 'None',
|
||||
}"
|
||||
/>
|
||||
? 'brightness(0) invert(1)'
|
||||
: 'None',
|
||||
}" />
|
||||
<span class="nav-text">{{ $t("header.voiceCloneManagement") }}</span>
|
||||
<i
|
||||
class="el-icon-arrow-down el-icon--right"
|
||||
:class="{ 'rotate-down': voiceCloneDropdownVisible }"
|
||||
></i>
|
||||
<i class="el-icon-arrow-down el-icon--right" :class="{ 'rotate-down': voiceCloneDropdownVisible }"></i>
|
||||
</span>
|
||||
<el-dropdown-menu slot="dropdown">
|
||||
<el-dropdown-item @click.native="goVoiceCloneManagement">
|
||||
@@ -96,82 +64,56 @@
|
||||
</el-dropdown-menu>
|
||||
</el-dropdown>
|
||||
|
||||
<div
|
||||
v-if="isSuperAdmin"
|
||||
class="equipment-management"
|
||||
:class="{ 'active-tab': $route.path === '/model-config' }"
|
||||
@click="goModelConfig"
|
||||
>
|
||||
<img
|
||||
loading="lazy"
|
||||
alt=""
|
||||
src="@/assets/header/model_config.png"
|
||||
:style="{
|
||||
filter:
|
||||
$route.path === '/model-config' ? 'brightness(0) invert(1)' : 'None',
|
||||
}"
|
||||
/>
|
||||
<div v-if="isSuperAdmin" class="equipment-management" :class="{ 'active-tab': $route.path === '/model-config' }"
|
||||
@click="goModelConfig">
|
||||
<img loading="lazy" alt="" src="@/assets/header/model_config.png" :style="{
|
||||
filter:
|
||||
$route.path === '/model-config' ? 'brightness(0) invert(1)' : 'None',
|
||||
}" />
|
||||
<span class="nav-text">{{ $t("header.modelConfig") }}</span>
|
||||
</div>
|
||||
<div
|
||||
v-if="isSuperAdmin"
|
||||
class="equipment-management"
|
||||
:class="{ 'active-tab': $route.path === '/user-management' }"
|
||||
@click="goUserManagement"
|
||||
>
|
||||
<img
|
||||
loading="lazy"
|
||||
alt=""
|
||||
src="@/assets/header/user_management.png"
|
||||
:style="{
|
||||
filter:
|
||||
$route.path === '/user-management' ? 'brightness(0) invert(1)' : 'None',
|
||||
}"
|
||||
/>
|
||||
<span class="nav-text">{{ $t("header.userManagement") }}</span>
|
||||
<div class="equipment-management"
|
||||
:class="{ 'active-tab': $route.path === '/knowledge-base-management' || $route.path === '/knowledge-file-upload' }"
|
||||
@click="goKnowledgeBaseManagement">
|
||||
<img loading="lazy" alt="" src="@/assets/header/knowledge_base.png" :style="{
|
||||
filter:
|
||||
$route.path === '/knowledge-base-management' || $route.path === '/knowledge-file-upload' ? 'brightness(0) invert(1)' : 'None',
|
||||
}" />
|
||||
<span class="nav-text">{{ $t("header.knowledgeBase") }}</span>
|
||||
</div>
|
||||
<el-dropdown
|
||||
v-if="isSuperAdmin"
|
||||
trigger="click"
|
||||
class="equipment-management more-dropdown"
|
||||
:class="{
|
||||
'active-tab':
|
||||
$route.path === '/dict-management' ||
|
||||
$route.path === '/params-management' ||
|
||||
$route.path === '/provider-management' ||
|
||||
$route.path === '/server-side-management' ||
|
||||
$route.path === '/agent-template-management' ||
|
||||
$route.path === '/ota-management',
|
||||
}"
|
||||
@visible-change="handleParamDropdownVisibleChange"
|
||||
>
|
||||
<el-dropdown v-if="isSuperAdmin" trigger="click" class="equipment-management more-dropdown" :class="{
|
||||
'active-tab':
|
||||
$route.path === '/dict-management' ||
|
||||
$route.path === '/params-management' ||
|
||||
$route.path === '/provider-management' ||
|
||||
$route.path === '/server-side-management' ||
|
||||
$route.path === '/agent-template-management' ||
|
||||
$route.path === '/ota-management' ||
|
||||
$route.path === '/user-management',
|
||||
}" @visible-change="handleParamDropdownVisibleChange">
|
||||
<span class="el-dropdown-link">
|
||||
<img
|
||||
loading="lazy"
|
||||
alt=""
|
||||
src="@/assets/header/param_management.png"
|
||||
:style="{
|
||||
filter:
|
||||
$route.path === '/dict-management' ||
|
||||
<img loading="lazy" alt="" src="@/assets/header/param_management.png" :style="{
|
||||
filter:
|
||||
$route.path === '/dict-management' ||
|
||||
$route.path === '/params-management' ||
|
||||
$route.path === '/provider-management' ||
|
||||
$route.path === '/server-side-management' ||
|
||||
$route.path === '/agent-template-management' ||
|
||||
$route.path === '/ota-management'
|
||||
? 'brightness(0) invert(1)'
|
||||
: 'None',
|
||||
}"
|
||||
/>
|
||||
$route.path === '/ota-management' ||
|
||||
$route.path === '/user-management'
|
||||
? 'brightness(0) invert(1)'
|
||||
: 'None',
|
||||
}" />
|
||||
<span class="nav-text">{{ $t("header.paramDictionary") }}</span>
|
||||
<i
|
||||
class="el-icon-arrow-down el-icon--right"
|
||||
:class="{ 'rotate-down': paramDropdownVisible }"
|
||||
></i>
|
||||
<i class="el-icon-arrow-down el-icon--right" :class="{ 'rotate-down': paramDropdownVisible }"></i>
|
||||
</span>
|
||||
<el-dropdown-menu slot="dropdown">
|
||||
<el-dropdown-item @click.native="goParamManagement">
|
||||
{{ $t("header.paramManagement") }}
|
||||
</el-dropdown-item>
|
||||
<el-dropdown-item @click.native="goUserManagement">
|
||||
{{ $t("header.userManagement") }}
|
||||
</el-dropdown-item>
|
||||
<el-dropdown-item @click.native="goOtaManagement">
|
||||
{{ $t("header.otaManagement") }}
|
||||
</el-dropdown-item>
|
||||
@@ -193,85 +135,40 @@
|
||||
|
||||
<!-- 右侧元素 -->
|
||||
<div class="header-right">
|
||||
<div
|
||||
class="search-container"
|
||||
v-if="$route.path === '/home' && !(isSuperAdmin && isSmallScreen)"
|
||||
>
|
||||
<div class="search-container" v-if="$route.path === '/home' && !(isSuperAdmin && isSmallScreen)">
|
||||
<div class="search-wrapper">
|
||||
<el-input
|
||||
v-model="search"
|
||||
:placeholder="$t('header.searchPlaceholder')"
|
||||
class="custom-search-input"
|
||||
@keyup.enter.native="handleSearch"
|
||||
@focus="showSearchHistory"
|
||||
@blur="hideSearchHistory"
|
||||
clearable
|
||||
ref="searchInput"
|
||||
>
|
||||
<i
|
||||
slot="suffix"
|
||||
class="el-icon-search search-icon"
|
||||
@click="handleSearch"
|
||||
></i>
|
||||
<el-input v-model="search" :placeholder="$t('header.searchPlaceholder')" class="custom-search-input"
|
||||
@keyup.enter.native="handleSearch" @focus="showSearchHistory" @blur="hideSearchHistory" clearable
|
||||
ref="searchInput">
|
||||
<i slot="suffix" class="el-icon-search search-icon" @click="handleSearch"></i>
|
||||
</el-input>
|
||||
<!-- 搜索历史下拉框 -->
|
||||
<div
|
||||
v-if="showHistory && searchHistory.length > 0"
|
||||
class="search-history-dropdown"
|
||||
>
|
||||
<div v-if="showHistory && searchHistory.length > 0" class="search-history-dropdown">
|
||||
<div class="search-history-header">
|
||||
<span>{{ $t("header.searchHistory") }}</span>
|
||||
<el-button
|
||||
type="text"
|
||||
size="small"
|
||||
class="clear-history-btn"
|
||||
@click="clearSearchHistory"
|
||||
>
|
||||
<el-button type="text" size="small" class="clear-history-btn" @click="clearSearchHistory">
|
||||
{{ $t("header.clearHistory") }}
|
||||
</el-button>
|
||||
</div>
|
||||
<div class="search-history-list">
|
||||
<div
|
||||
v-for="(item, index) in searchHistory"
|
||||
:key="index"
|
||||
class="search-history-item"
|
||||
@click.stop="selectSearchHistory(item)"
|
||||
>
|
||||
<div v-for="(item, index) in searchHistory" :key="index" class="search-history-item"
|
||||
@click.stop="selectSearchHistory(item)">
|
||||
<span class="history-text">{{ item }}</span>
|
||||
<i
|
||||
class="el-icon-close clear-item-icon"
|
||||
@click.stop="removeSearchHistory(index)"
|
||||
></i>
|
||||
<i class="el-icon-close clear-item-icon" @click.stop="removeSearchHistory(index)"></i>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<img
|
||||
loading="lazy"
|
||||
alt=""
|
||||
src="@/assets/home/avatar.png"
|
||||
class="avatar-img"
|
||||
@click="handleAvatarClick"
|
||||
/>
|
||||
<img loading="lazy" alt="" src="@/assets/home/avatar.png" class="avatar-img" @click="handleAvatarClick" />
|
||||
<span class="el-dropdown-link" @click="handleAvatarClick">
|
||||
{{ userInfo.username || "加载中..." }}
|
||||
<i
|
||||
class="el-icon-arrow-down el-icon--right"
|
||||
:class="{ 'rotate-down': userMenuVisible }"
|
||||
></i>
|
||||
<i class="el-icon-arrow-down el-icon--right" :class="{ 'rotate-down': userMenuVisible }"></i>
|
||||
</span>
|
||||
<el-cascader
|
||||
:options="userMenuOptions"
|
||||
trigger="click"
|
||||
:props="cascaderProps"
|
||||
style="width: 0px; overflow: hidden"
|
||||
:show-all-levels="false"
|
||||
@change="handleCascaderChange"
|
||||
@visible-change="handleUserMenuVisibleChange"
|
||||
ref="userCascader"
|
||||
>
|
||||
<el-cascader :options="userMenuOptions" trigger="click" :props="cascaderProps"
|
||||
style="width: 0px; overflow: hidden" :show-all-levels="false" @change="handleCascaderChange"
|
||||
@visible-change="handleUserMenuVisibleChange" ref="userCascader">
|
||||
<template slot-scope="{ data }">
|
||||
<span>{{ data.label }}</span>
|
||||
</template>
|
||||
@@ -399,6 +296,9 @@ export default {
|
||||
goModelConfig() {
|
||||
this.$router.push("/model-config");
|
||||
},
|
||||
goKnowledgeBaseManagement() {
|
||||
this.$router.push("/knowledge-base-management");
|
||||
},
|
||||
goVoiceCloneManagement() {
|
||||
this.$router.push("/voice-clone-management");
|
||||
},
|
||||
@@ -863,7 +763,7 @@ export default {
|
||||
color: #ff4949;
|
||||
}
|
||||
|
||||
.custom-search-input >>> .el-input__inner {
|
||||
.custom-search-input>>>.el-input__inner {
|
||||
height: 18px;
|
||||
border-radius: 9px;
|
||||
background-color: #fff;
|
||||
@@ -933,6 +833,7 @@ export default {
|
||||
color: #606266;
|
||||
white-space: nowrap;
|
||||
}
|
||||
|
||||
/* 添加倒三角旋转样式 */
|
||||
.rotate-down {
|
||||
transform: rotate(180deg);
|
||||
|
||||
@@ -0,0 +1,238 @@
|
||||
<template>
|
||||
<el-dialog :title="title" :visible="dialogVisible" width="600px" class="knowledge-base-dialog" @close="handleClose">
|
||||
<el-form ref="knowledgeBaseForm" :model="form" :rules="rules" label-width="100px" size="medium">
|
||||
<el-form-item :label="$t('knowledgeBaseDialog.name')" prop="name">
|
||||
<el-input v-model="form.name" :placeholder="$t('knowledgeBaseDialog.namePlaceholder')" clearable></el-input>
|
||||
</el-form-item>
|
||||
<el-form-item :label="$t('knowledgeBaseDialog.description')" prop="description">
|
||||
<el-input v-model="form.description" :placeholder="$t('knowledgeBaseDialog.descriptionPlaceholder')"
|
||||
type="textarea" :rows="4" maxlength="200" show-word-limit></el-input>
|
||||
</el-form-item>
|
||||
<el-form-item :label="$t('knowledgeBaseDialog.ragModel')" prop="ragModelId">
|
||||
<el-select v-model="form.ragModelId" :placeholder="$t('knowledgeBaseDialog.ragModelPlaceholder')" clearable
|
||||
filterable style="width: 100%" @focus="loadRAGModels">
|
||||
<el-option v-for="model in ragModels" :key="model.id" :label="model.modelName" :value="model.id">
|
||||
</el-option>
|
||||
</el-select>
|
||||
</el-form-item>
|
||||
</el-form>
|
||||
<div slot="footer" class="dialog-footer">
|
||||
<el-button @click="handleClose">{{ $t('knowledgeBaseDialog.cancel') }}</el-button>
|
||||
<el-button type="primary" @click="handleSubmit">{{ $t('knowledgeBaseDialog.confirm') }}</el-button>
|
||||
</div>
|
||||
</el-dialog>
|
||||
</template>
|
||||
|
||||
<script>
|
||||
import Api from "@/apis/api";
|
||||
|
||||
export default {
|
||||
name: "KnowledgeBaseDialog",
|
||||
props: {
|
||||
title: {
|
||||
type: String,
|
||||
default: ""
|
||||
},
|
||||
visible: {
|
||||
type: Boolean,
|
||||
default: false
|
||||
},
|
||||
form: {
|
||||
type: Object,
|
||||
default: () => ({
|
||||
id: null,
|
||||
datasetId: null,
|
||||
ragModelId: null,
|
||||
name: "",
|
||||
description: ""
|
||||
})
|
||||
}
|
||||
},
|
||||
data() {
|
||||
return {
|
||||
dialogVisible: this.visible,
|
||||
ragModels: [],
|
||||
rules: {
|
||||
name: [
|
||||
{
|
||||
required: true,
|
||||
message: this.$t("knowledgeBaseDialog.nameRequired"),
|
||||
trigger: "blur"
|
||||
},
|
||||
{
|
||||
min: 1,
|
||||
max: 50,
|
||||
message: this.$t("knowledgeBaseDialog.nameLength"),
|
||||
trigger: "blur"
|
||||
},
|
||||
{
|
||||
pattern: /^[\u4e00-\u9fa5a-zA-Z0-9\s-_]+$/,
|
||||
message: this.$t("knowledgeBaseDialog.namePattern"),
|
||||
trigger: "blur"
|
||||
}
|
||||
],
|
||||
description: [
|
||||
{
|
||||
max: 200,
|
||||
message: this.$t("knowledgeBaseDialog.descriptionLength"),
|
||||
trigger: "blur"
|
||||
}
|
||||
],
|
||||
ragModelId: [
|
||||
{
|
||||
required: true,
|
||||
message: this.$t("knowledgeBaseDialog.ragModelRequired"),
|
||||
trigger: "change"
|
||||
}
|
||||
]
|
||||
}
|
||||
};
|
||||
},
|
||||
watch: {
|
||||
visible(val) {
|
||||
this.dialogVisible = val;
|
||||
if (val) {
|
||||
// 对话框显示时加载RAG模型列表
|
||||
this.loadRAGModels();
|
||||
|
||||
// 如果是新增知识库且没有设置ragModelId，则默认选择第一个RAG模型
|
||||
if (!this.form.id && !this.form.ragModelId && this.ragModels.length > 0) {
|
||||
this.$set(this.form, 'ragModelId', this.ragModels[0].id);
|
||||
}
|
||||
|
||||
if (this.$refs.knowledgeBaseForm) {
|
||||
this.$refs.knowledgeBaseForm.clearValidate();
|
||||
}
|
||||
}
|
||||
},
|
||||
// 监听RAG模型列表变化，确保新增时能正确设置默认值
|
||||
ragModels(newModels) {
|
||||
if (newModels.length > 0) {
|
||||
// 如果是新增知识库且没有设置ragModelId，则默认选择第一个RAG模型
|
||||
if (!this.form.id && !this.form.ragModelId) {
|
||||
this.$set(this.form, 'ragModelId', newModels[0].id);
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
methods: {
|
||||
handleClose() {
|
||||
// 不重置表单字段，以便在编辑时能保留之前的选择
|
||||
// 只在对话框关闭时重置验证状态
|
||||
if (this.$refs.knowledgeBaseForm) {
|
||||
this.$refs.knowledgeBaseForm.clearValidate();
|
||||
}
|
||||
this.dialogVisible = false;
|
||||
this.$emit("update:visible", false);
|
||||
},
|
||||
handleSubmit() {
|
||||
console.log('KnowledgeBaseDialog handleSubmit called');
|
||||
this.$refs.knowledgeBaseForm.validate(valid => {
|
||||
console.log('Form validation result:', valid);
|
||||
if (valid) {
|
||||
console.log('Emitting submit event with form:', this.form);
|
||||
this.$emit("submit", {
|
||||
...this.form
|
||||
});
|
||||
}
|
||||
});
|
||||
},
|
||||
loadRAGModels() {
|
||||
if (this.ragModels.length > 0) {
|
||||
return; // 已经加载过，避免重复加载
|
||||
}
|
||||
|
||||
console.log('开始加载RAG模型列表');
|
||||
Api.model.getRAGModels((res) => {
|
||||
console.log('RAG模型列表响应:', res);
|
||||
if (res.data && res.data.code === 0) {
|
||||
this.ragModels = res.data.data || [];
|
||||
console.log('RAG模型列表加载成功，共', this.ragModels.length, '个模型');
|
||||
|
||||
// 如果是新增知识库且没有设置ragModelId，则默认选择第一个RAG模型
|
||||
if (!this.form.id && !this.form.ragModelId && this.ragModels.length > 0) {
|
||||
this.$set(this.form, 'ragModelId', this.ragModels[0].id);
|
||||
console.log('已设置默认RAG模型:', this.ragModels[0].id);
|
||||
}
|
||||
} else {
|
||||
console.error('获取RAG模型列表失败:', res.data?.msg);
|
||||
this.$message.error(this.$t('knowledgeBaseDialog.loadRAGModelsFailed'));
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
};
|
||||
</script>
|
||||
|
||||
<style lang="scss" scoped>
|
||||
.knowledge-base-dialog {
|
||||
::v-deep .el-dialog {
|
||||
border-radius: 20px;
|
||||
overflow: hidden;
|
||||
}
|
||||
|
||||
::v-deep .el-dialog__body {
|
||||
padding: 20px 30px;
|
||||
}
|
||||
|
||||
::v-deep .el-form-item {
|
||||
margin-bottom: 20px;
|
||||
}
|
||||
|
||||
::v-deep .el-form-item__label {
|
||||
font-weight: 500;
|
||||
color: #34495e;
|
||||
font-size: 14px;
|
||||
}
|
||||
|
||||
::v-deep .el-input {
|
||||
.el-input__inner {
|
||||
height: 36px;
|
||||
font-size: 14px;
|
||||
border-radius: 4px;
|
||||
border: 1px solid #ddd;
|
||||
transition: all 0.3s ease;
|
||||
|
||||
&:focus {
|
||||
border-color: #5f70f3;
|
||||
box-shadow: 0 0 0 2px rgba(95, 112, 243, 0.2);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
::v-deep .el-textarea {
|
||||
.el-textarea__inner {
|
||||
font-size: 14px;
|
||||
border-radius: 4px;
|
||||
border: 1px solid #ddd;
|
||||
transition: all 0.3s ease;
|
||||
|
||||
&:focus {
|
||||
border-color: #5f70f3;
|
||||
box-shadow: 0 0 0 2px rgba(95, 112, 243, 0.2);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
::v-deep .el-select {
|
||||
.el-input__inner {
|
||||
height: 36px;
|
||||
font-size: 14px;
|
||||
border-radius: 4px;
|
||||
border: 1px solid #ddd;
|
||||
transition: all 0.3s ease;
|
||||
|
||||
&:focus {
|
||||
border-color: #5f70f3;
|
||||
box-shadow: 0 0 0 2px rgba(95, 112, 243, 0.2);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
.dialog-footer {
|
||||
display: flex;
|
||||
justify-content: flex-end;
|
||||
gap: 10px;
|
||||
}
|
||||
}
|
||||
</style>
|
||||
@@ -88,6 +88,7 @@
|
||||
<el-option :label="$t('providerDialog.booleanType')" value="boolean"></el-option>
|
||||
<el-option :label="$t('providerDialog.dictType')" value="dict"></el-option>
|
||||
<el-option :label="$t('providerDialog.arrayType')" value="array"></el-option>
|
||||
<el-option :label="$t('providerDialog.ragType')" value="RAG"></el-option>
|
||||
</el-select>
|
||||
</template>
|
||||
<template v-else>
|
||||
@@ -165,7 +166,8 @@ export default {
|
||||
'number': this.$t('providerDialog.numberType'),
|
||||
'boolean': this.$t('providerDialog.booleanType'),
|
||||
'dict': this.$t('providerDialog.dictType'),
|
||||
'array': this.$t('providerDialog.arrayType')
|
||||
'array': this.$t('providerDialog.arrayType'),
|
||||
'RAG': this.$t('providerDialog.ragType')
|
||||
};
|
||||
return typeMap[type];
|
||||
},
|
||||
|
||||
@@ -9,6 +9,7 @@ export default {
|
||||
// HeaderBar组件文本
|
||||
'header.smartManagement': 'Agents',
|
||||
'header.modelConfig': 'Models',
|
||||
'header.knowledgeBase': 'Knowledge',
|
||||
'header.voiceCloneManagement': 'Voice Clone',
|
||||
'header.voiceResourceManagement': 'Voice Resource',
|
||||
'header.userManagement': 'Users',
|
||||
@@ -632,6 +633,7 @@ export default {
|
||||
'common.confirm': 'Confirm',
|
||||
'common.cancel': 'Cancel',
|
||||
'common.sensitive': 'Sensitive',
|
||||
'common.loading': 'Loading',
|
||||
|
||||
// Language switch
|
||||
'language.zhCN': '中文简体',
|
||||
@@ -793,6 +795,7 @@ export default {
|
||||
'modelConfig.intent': 'Intent Recognition',
|
||||
'modelConfig.tts': 'Text-to-Speech',
|
||||
'modelConfig.memory': 'Memory',
|
||||
'modelConfig.rag': 'RAG',
|
||||
'modelConfig.modelId': 'Model ID',
|
||||
'modelConfig.modelName': 'Model Name',
|
||||
'modelConfig.provider': 'Provider',
|
||||
@@ -815,6 +818,7 @@ export default {
|
||||
'modelConfig.partialDeleteFailed': 'Partial deletion failed',
|
||||
'modelConfig.deleteSuccess': 'Deletion successful',
|
||||
'modelConfig.deleteFailed': 'Deletion failed',
|
||||
'modelConfig.deleteCancelled': 'Deletion cancelled',
|
||||
'modelConfig.duplicateSuccess': 'Duplication successful',
|
||||
'modelConfig.duplicateFailed': 'Duplication failed',
|
||||
'modelConfig.saveSuccess': 'Save successful',
|
||||
@@ -942,6 +946,7 @@ export default {
|
||||
'providerManagement.modelType.Memory': 'Memory Module',
|
||||
'providerManagement.modelType.VAD': 'Voice Activity Detection',
|
||||
'providerManagement.modelType.Plugin': 'Plugin Tool',
|
||||
'providerManagement.modelType.RAG': 'RAG',
|
||||
|
||||
// Provider Dialog translations
|
||||
'providerDialog.category': 'Category',
|
||||
@@ -965,6 +970,7 @@ export default {
|
||||
'providerDialog.booleanType': 'Boolean',
|
||||
'providerDialog.dictType': 'Dictionary',
|
||||
'providerDialog.arrayType': 'Semicolon-separated List',
|
||||
'providerDialog.ragType': 'RAG',
|
||||
'providerDialog.defaultValue': 'Default Value',
|
||||
'providerDialog.inputDefaultValue': 'Please input default value',
|
||||
'providerDialog.operation': 'Operation',
|
||||
@@ -1118,4 +1124,139 @@ export default {
|
||||
'voiceClone.updateNameFailed': 'Failed to update name',
|
||||
'voiceClone.playFailed': 'Play failed',
|
||||
'voiceClone.Details': 'Error Details',
|
||||
|
||||
// Knowledge Base Management page text
|
||||
'knowledgeBaseManagement.title': 'Knowledge Base',
|
||||
'knowledgeBaseManagement.searchPlaceholder': 'Please enter knowledge base name to search',
|
||||
'knowledgeBaseManagement.search': 'Search',
|
||||
'knowledgeBaseManagement.name': 'Knowledge Base Name',
|
||||
'knowledgeBaseManagement.description': 'Knowledge Base Description',
|
||||
'knowledgeBaseManagement.documentCount': 'Document Count',
|
||||
'knowledgeBaseManagement.status': 'Enabled',
|
||||
'knowledgeBaseManagement.createdAt': 'Created At',
|
||||
'knowledgeBaseManagement.operation': 'Operation',
|
||||
'knowledgeBaseManagement.add': 'Add',
|
||||
'knowledgeBaseManagement.delete': 'Delete',
|
||||
'knowledgeBaseManagement.edit': 'Edit',
|
||||
'knowledgeBaseManagement.itemsPerPage': 'items/page',
|
||||
'knowledgeBaseManagement.firstPage': 'First Page',
|
||||
'knowledgeBaseManagement.prevPage': 'Previous Page',
|
||||
'knowledgeBaseManagement.nextPage': 'Next Page',
|
||||
'knowledgeBaseManagement.totalRecords': 'Total {total} records',
|
||||
'knowledgeBaseManagement.addKnowledgeBase': 'Add Knowledge Base',
|
||||
'knowledgeBaseManagement.editKnowledgeBase': 'Edit Knowledge Base',
|
||||
'knowledgeBaseManagement.getKnowledgeBaseListFailed': 'Failed to get knowledge base list',
|
||||
'knowledgeBaseManagement.selectKnowledgeBaseFirst': 'Please select knowledge bases to delete first',
|
||||
'knowledgeBaseManagement.confirmBatchDelete': 'Are you sure you want to delete the selected {count} knowledge bases?',
|
||||
'knowledgeBaseManagement.batchDeleteSuccess': 'Successfully deleted {count} knowledge bases',
|
||||
'knowledgeBaseManagement.deleteFailed': 'Delete failed, please try again',
|
||||
'knowledgeBaseManagement.operationCancelled': 'Delete operation has been cancelled',
|
||||
'knowledgeBaseManagement.updateSuccess': 'Update successful',
|
||||
'knowledgeBaseManagement.addSuccess': 'Add successful',
|
||||
'knowledgeBaseManagement.updateFailed': 'Update failed',
|
||||
'knowledgeBaseManagement.addFailed': 'Add failed',
|
||||
'knowledgeBaseManagement.selectAll': 'Select All',
|
||||
'knowledgeBaseManagement.cancelSelectAll': 'Deselect All',
|
||||
|
||||
// Knowledge Base Dialog text
|
||||
'knowledgeBaseDialog.title': 'Knowledge Base',
|
||||
'knowledgeBaseDialog.name': 'Knowledge Base Name',
|
||||
'knowledgeBaseDialog.namePlaceholder': 'Please enter knowledge base name accurately to improve the accuracy of knowledge base call',
|
||||
'knowledgeBaseDialog.description': 'Knowledge Base Description',
|
||||
'knowledgeBaseDialog.descriptionPlaceholder': 'Please enter knowledge base description accurately to help the model better understand the knowledge base content',
|
||||
'knowledgeBaseDialog.ragModel': 'RAG Model',
|
||||
'knowledgeBaseDialog.ragModelPlaceholder': 'Please select RAG model',
|
||||
'knowledgeBaseDialog.ragModelRequired': 'Please select RAG model',
|
||||
'knowledgeBaseDialog.loadRAGModelsFailed': 'Failed to load RAG models list',
|
||||
'knowledgeBaseDialog.status': 'Status',
|
||||
'knowledgeBaseDialog.statusEnabled': 'Enabled',
|
||||
'knowledgeBaseDialog.statusDisabled': 'Disabled',
|
||||
'knowledgeBaseDialog.save': 'Save',
|
||||
'knowledgeBaseDialog.cancel': 'Cancel',
|
||||
'knowledgeBaseDialog.confirm': 'Confirm',
|
||||
'knowledgeBaseDialog.requiredName': 'Please enter knowledge base name',
|
||||
'knowledgeBaseDialog.nameRequired': 'Please enter knowledge base name',
|
||||
'knowledgeBaseDialog.nameLength': 'Knowledge base name length should be between 1 and 50 characters',
|
||||
'knowledgeBaseDialog.namePattern': 'Knowledge base name can only contain Chinese, English, numbers, spaces, underscores and hyphens',
|
||||
'knowledgeBaseDialog.descriptionLength': 'Knowledge base description cannot exceed 200 characters',
|
||||
'knowledgeBaseDialog.nameLengthLimit': 'Knowledge base name cannot exceed 50 characters',
|
||||
'knowledgeBaseDialog.descriptionLengthLimit': 'Knowledge base description cannot exceed 200 characters',
|
||||
|
||||
// Knowledge Base Management page new view button text
|
||||
'knowledgeBaseManagement.view': 'View',
|
||||
|
||||
// Knowledge File Upload page text
|
||||
'knowledgeFileUpload.back': 'Back',
|
||||
'knowledgeFileUpload.searchPlaceholder': 'Please enter document name to search',
|
||||
'knowledgeFileUpload.search': 'Search',
|
||||
'knowledgeFileUpload.addDocument': 'Add Document',
|
||||
'knowledgeFileUpload.documentName': 'Document Name',
|
||||
'knowledgeFileUpload.uploadTime': 'Upload Time',
|
||||
'knowledgeFileUpload.status': 'Status',
|
||||
'knowledgeFileUpload.operation': 'Operation',
|
||||
'knowledgeFileUpload.parse': 'Parse',
|
||||
'knowledgeFileUpload.viewSlices': 'View Slices',
|
||||
'knowledgeFileUpload.delete': 'Delete',
|
||||
'knowledgeFileUpload.itemsPerPage': 'items/page',
|
||||
'knowledgeFileUpload.firstPage': 'First Page',
|
||||
'knowledgeFileUpload.prevPage': 'Previous Page',
|
||||
'knowledgeFileUpload.nextPage': 'Next Page',
|
||||
'knowledgeFileUpload.totalRecords': 'Total {total} records',
|
||||
'knowledgeFileUpload.uploadDocument': 'Upload Document',
|
||||
'knowledgeFileUpload.documentNamePlaceholder': 'Please enter document name',
|
||||
'knowledgeFileUpload.file': 'File',
|
||||
'knowledgeFileUpload.clickToUpload': 'Click to upload',
|
||||
'knowledgeFileUpload.uploadTip': 'Supported file types: PDF, DOC, DOCX, TXT, MD, CSV, XLS, XLSX, PPT, PPTX. Maximum 32 files per upload, each file size up to 10MB',
|
||||
'knowledgeFileUpload.dragOrClick': 'Drag file here, or click to upload',
|
||||
'knowledgeFileUpload.cancel': 'Cancel',
|
||||
'knowledgeFileUpload.confirm': 'Confirm',
|
||||
'knowledgeFileUpload.knowledgeBaseName': 'Knowledge Base Name',
|
||||
'knowledgeFileUpload.statusNotStarted': 'Not Started',
|
||||
'knowledgeFileUpload.statusProcessing': 'Processing',
|
||||
'knowledgeFileUpload.statusCancelled': 'Cancelled',
|
||||
'knowledgeFileUpload.statusCompleted': 'Completed',
|
||||
'knowledgeFileUpload.statusFailed': 'Parse Failed',
|
||||
'knowledgeFileUpload.uploadSuccess': 'Document upload successful',
|
||||
'knowledgeFileUpload.uploadFailed': 'Document upload failed',
|
||||
'knowledgeFileUpload.parseSuccess': 'Document parse successful',
|
||||
'knowledgeFileUpload.parseFailed': 'Document parse failed',
|
||||
'knowledgeFileUpload.deleteSuccess': 'Document delete successful',
|
||||
'knowledgeFileUpload.deleteFailed': 'Document delete failed',
|
||||
'knowledgeFileUpload.confirmDelete': 'Are you sure you want to delete this document?',
|
||||
'knowledgeFileUpload.confirmParse': 'Are you sure you want to parse this document?',
|
||||
'knowledgeFileUpload.nameRequired': 'Please enter document name',
|
||||
'knowledgeFileUpload.fileRequired': 'Please select a file to upload',
|
||||
'knowledgeFileUpload.getListFailed': 'Failed to get document list',
|
||||
'knowledgeFileUpload.parseCancelled': 'Parse cancelled',
|
||||
'knowledgeFileUpload.deleteCancelled': 'Delete cancelled',
|
||||
'knowledgeFileUpload.selectFilesFirst': 'Please select files to delete first',
|
||||
'knowledgeFileUpload.selectAll': 'Select All',
|
||||
'knowledgeFileUpload.deselectAll': 'Deselect All',
|
||||
'knowledgeFileUpload.batchDelete': 'Delete',
|
||||
'knowledgeFileUpload.confirmBatchDelete': 'Are you sure you want to delete the selected {count} files?',
|
||||
'knowledgeFileUpload.batchDeleteSuccess': 'Successfully deleted {count} files',
|
||||
'knowledgeFileUpload.batchDeleteFailed': 'Batch delete failed',
|
||||
'knowledgeFileUpload.sliceCount': 'Slice Count',
|
||||
'knowledgeFileUpload.add': 'Add',
|
||||
'knowledgeFileUpload.retrievalTest': 'Retrieval Test',
|
||||
'knowledgeFileUpload.testQuestion': 'Test Question',
|
||||
'knowledgeFileUpload.testQuestionPlaceholder': 'Please enter the question to test',
|
||||
'knowledgeFileUpload.executeTest': 'Execute Test',
|
||||
'knowledgeFileUpload.testResult': 'Test Result:',
|
||||
'knowledgeFileUpload.selectedFiles': 'Selected Files',
|
||||
'knowledgeFileUpload.totalSlices': 'Total {total} records',
|
||||
'knowledgeFileUpload.slice': 'Slice',
|
||||
'knowledgeFileUpload.noSliceData': 'No slice data available',
|
||||
'knowledgeFileUpload.firstPage': 'First',
|
||||
'knowledgeFileUpload.prevPage': 'Previous',
|
||||
'knowledgeFileUpload.nextPage': 'Next',
|
||||
'knowledgeFileUpload.totalRecords': 'Total {total} records',
|
||||
'knowledgeFileUpload.testQuestion': 'Test Question',
|
||||
'knowledgeFileUpload.testQuestionPlaceholder': 'Please enter the question to test',
|
||||
'knowledgeFileUpload.runTest': 'Run Test',
|
||||
'knowledgeFileUpload.testResult': 'Test Result:',
|
||||
'knowledgeFileUpload.noRelatedSlices': 'No related slices found',
|
||||
'knowledgeFileUpload.comprehensiveSimilarity': 'Comprehensive Similarity',
|
||||
'knowledgeFileUpload.content': 'Content:',
|
||||
'knowledgeFileUpload.testQuestionRequired': 'Please enter test question',
|
||||
}
|
||||
@@ -11,6 +11,7 @@ export default {
|
||||
'header.voiceCloneManagement': '音色克隆',
|
||||
'header.voiceResourceManagement': '音色资源',
|
||||
'header.modelConfig': '模型配置',
|
||||
'header.knowledgeBase': '知识库',
|
||||
'header.userManagement': '用户管理',
|
||||
'header.otaManagement': 'OTA管理',
|
||||
'header.paramDictionary': '参数字典',
|
||||
@@ -632,6 +633,7 @@ export default {
|
||||
'common.confirm': '确定',
|
||||
'common.cancel': '取消',
|
||||
'common.sensitive': '敏感',
|
||||
'common.loading': '加载中',
|
||||
|
||||
// 语言切换
|
||||
'language.zhCN': '中文简体',
|
||||
@@ -793,6 +795,7 @@ export default {
|
||||
'modelConfig.intent': '意图识别',
|
||||
'modelConfig.tts': '语音合成',
|
||||
'modelConfig.memory': '记忆',
|
||||
'modelConfig.rag': '知识库',
|
||||
'modelConfig.modelId': '模型ID',
|
||||
'modelConfig.modelName': '模型名称',
|
||||
'modelConfig.provider': '提供商',
|
||||
@@ -815,6 +818,7 @@ export default {
|
||||
'modelConfig.partialDeleteFailed': '部分删除失败',
|
||||
'modelConfig.deleteSuccess': '删除成功',
|
||||
'modelConfig.deleteFailed': '删除失败',
|
||||
'modelConfig.deleteCancelled': '删除已取消',
|
||||
'modelConfig.duplicateSuccess': '创建副本成功',
|
||||
'modelConfig.duplicateFailed': '创建副本失败',
|
||||
'modelConfig.saveSuccess': '保存成功',
|
||||
@@ -942,6 +946,7 @@ export default {
|
||||
'providerManagement.modelType.Memory': '记忆模块',
|
||||
'providerManagement.modelType.VAD': '语音活动检测',
|
||||
'providerManagement.modelType.Plugin': '插件工具',
|
||||
'providerManagement.modelType.RAG': '知识库',
|
||||
|
||||
// Provider Dialog 翻译
|
||||
'providerDialog.category': '类别',
|
||||
@@ -965,6 +970,7 @@ export default {
|
||||
'providerDialog.booleanType': '布尔值',
|
||||
'providerDialog.dictType': '字典',
|
||||
'providerDialog.arrayType': '分号分割的列表',
|
||||
'providerDialog.ragType': '知识库',
|
||||
'providerDialog.defaultValue': '默认值',
|
||||
'providerDialog.inputDefaultValue': '请输入默认值',
|
||||
'providerDialog.operation': '操作',
|
||||
@@ -1118,4 +1124,139 @@ export default {
|
||||
'voiceClone.updateNameFailed': '名称更新失败',
|
||||
'voiceClone.playFailed': '播放失败',
|
||||
'voiceClone.Details': '错误详情',
|
||||
|
||||
// 知识库管理页面文本
|
||||
'knowledgeBaseManagement.title': '知识库',
|
||||
'knowledgeBaseManagement.searchPlaceholder': '请输入知识库名称搜索',
|
||||
'knowledgeBaseManagement.search': '搜索',
|
||||
'knowledgeBaseManagement.name': '知识库名称',
|
||||
'knowledgeBaseManagement.description': '知识库描述',
|
||||
'knowledgeBaseManagement.documentCount': '文档数量',
|
||||
'knowledgeBaseManagement.status': '启用',
|
||||
'knowledgeBaseManagement.createdAt': '创建时间',
|
||||
'knowledgeBaseManagement.operation': '操作',
|
||||
'knowledgeBaseManagement.add': '新增',
|
||||
'knowledgeBaseManagement.delete': '删除',
|
||||
'knowledgeBaseManagement.edit': '编辑',
|
||||
'knowledgeBaseManagement.itemsPerPage': '条/页',
|
||||
'knowledgeBaseManagement.firstPage': '首页',
|
||||
'knowledgeBaseManagement.prevPage': '上一页',
|
||||
'knowledgeBaseManagement.nextPage': '下一页',
|
||||
'knowledgeBaseManagement.totalRecords': '共{total}条记录',
|
||||
'knowledgeBaseManagement.addKnowledgeBase': '新增知识库',
|
||||
'knowledgeBaseManagement.editKnowledgeBase': '编辑知识库',
|
||||
'knowledgeBaseManagement.getKnowledgeBaseListFailed': '获取知识库列表失败',
|
||||
'knowledgeBaseManagement.selectKnowledgeBaseFirst': '请先选择需要删除的知识库',
|
||||
'knowledgeBaseManagement.confirmBatchDelete': '确定要删除选中的{count}个知识库吗？',
|
||||
'knowledgeBaseManagement.batchDeleteSuccess': '成功删除{count}个知识库',
|
||||
'knowledgeBaseManagement.deleteFailed': '删除失败，请重试',
|
||||
'knowledgeBaseManagement.operationCancelled': '已取消删除操作',
|
||||
'knowledgeBaseManagement.updateSuccess': '修改成功',
|
||||
'knowledgeBaseManagement.addSuccess': '新增成功',
|
||||
'knowledgeBaseManagement.updateFailed': '更新失败',
|
||||
'knowledgeBaseManagement.addFailed': '新增失败',
|
||||
'knowledgeBaseManagement.selectAll': '全选',
|
||||
'knowledgeBaseManagement.cancelSelectAll': '取消全选',
|
||||
|
||||
// 知识库对话框文本
|
||||
'knowledgeBaseDialog.title': '知识库',
|
||||
'knowledgeBaseDialog.name': '知识库名称',
|
||||
'knowledgeBaseDialog.namePlaceholder': '请精确输入知识库名称，才能提高调用知识库的精准性',
|
||||
'knowledgeBaseDialog.description': '知识库描述',
|
||||
'knowledgeBaseDialog.descriptionPlaceholder': '请详细输入知识库描述，以便大模型更好地理解这个知识库的总体内容',
|
||||
'knowledgeBaseDialog.ragModel': 'RAG模型',
|
||||
'knowledgeBaseDialog.ragModelPlaceholder': '请选择RAG模型',
|
||||
'knowledgeBaseDialog.ragModelRequired': '请选择RAG模型',
|
||||
'knowledgeBaseDialog.loadRAGModelsFailed': '加载RAG模型列表失败',
|
||||
'knowledgeBaseDialog.status': '状态',
|
||||
'knowledgeBaseDialog.statusEnabled': '启用',
|
||||
'knowledgeBaseDialog.statusDisabled': '禁用',
|
||||
'knowledgeBaseDialog.save': '保存',
|
||||
'knowledgeBaseDialog.cancel': '取消',
|
||||
'knowledgeBaseDialog.confirm': '确认',
|
||||
'knowledgeBaseDialog.requiredName': '请输入知识库名称',
|
||||
'knowledgeBaseDialog.nameRequired': '请输入知识库名称',
|
||||
'knowledgeBaseDialog.nameLength': '知识库名称长度在1到50个字符之间',
|
||||
'knowledgeBaseDialog.namePattern': '知识库名称只能包含中文、英文、数字、空格、下划线和连字符',
|
||||
'knowledgeBaseDialog.descriptionLength': '知识库描述不能超过200个字符',
|
||||
'knowledgeBaseDialog.nameLengthLimit': '知识库名称不能超过50个字符',
|
||||
'knowledgeBaseDialog.descriptionLengthLimit': '知识库描述不能超过200个字符',
|
||||
|
||||
// 知识库管理页面新增查看按钮文本
|
||||
'knowledgeBaseManagement.view': '查看',
|
||||
|
||||
// 上传文件页面文本
|
||||
'knowledgeFileUpload.back': '返回',
|
||||
'knowledgeFileUpload.searchPlaceholder': '请输入文档名称搜索',
|
||||
'knowledgeFileUpload.search': '搜索',
|
||||
'knowledgeFileUpload.addDocument': '新增文档',
|
||||
'knowledgeFileUpload.documentName': '文档名称',
|
||||
'knowledgeFileUpload.uploadTime': '上传时间',
|
||||
'knowledgeFileUpload.status': '状态',
|
||||
'knowledgeFileUpload.operation': '操作',
|
||||
'knowledgeFileUpload.parse': '解析',
|
||||
'knowledgeFileUpload.viewSlices': '查看切片',
|
||||
'knowledgeFileUpload.delete': '删除',
|
||||
'knowledgeFileUpload.itemsPerPage': '条/页',
|
||||
'knowledgeFileUpload.firstPage': '首页',
|
||||
'knowledgeFileUpload.prevPage': '上一页',
|
||||
'knowledgeFileUpload.nextPage': '下一页',
|
||||
'knowledgeFileUpload.totalRecords': '共{total}条记录',
|
||||
'knowledgeFileUpload.uploadDocument': '上传文档',
|
||||
'knowledgeFileUpload.documentNamePlaceholder': '请输入文档名称',
|
||||
'knowledgeFileUpload.file': '文件',
|
||||
'knowledgeFileUpload.clickToUpload': '点击上传',
|
||||
'knowledgeFileUpload.uploadTip': '支持的文档类型：PDF、DOC、DOCX、TXT、MD、CSV、XLS、XLSX、PPT、PPTX，单次批量上传文件数不超过 32 个',
|
||||
'knowledgeFileUpload.dragOrClick': '将文件拖到此处，或点击上传',
|
||||
'knowledgeFileUpload.cancel': '取消',
|
||||
'knowledgeFileUpload.confirm': '确定',
|
||||
'knowledgeFileUpload.knowledgeBaseName': '知识库名称',
|
||||
'knowledgeFileUpload.statusNotStarted': '未开始',
|
||||
'knowledgeFileUpload.statusProcessing': '处理中',
|
||||
'knowledgeFileUpload.statusCancelled': '已取消',
|
||||
'knowledgeFileUpload.statusCompleted': '已完成',
|
||||
'knowledgeFileUpload.statusFailed': '解析失败',
|
||||
'knowledgeFileUpload.uploadSuccess': '文档上传成功',
|
||||
'knowledgeFileUpload.uploadFailed': '文档上传失败',
|
||||
'knowledgeFileUpload.parseSuccess': '文档解析成功',
|
||||
'knowledgeFileUpload.parseFailed': '文档解析失败',
|
||||
'knowledgeFileUpload.deleteSuccess': '文档删除成功',
|
||||
'knowledgeFileUpload.deleteFailed': '文档删除失败',
|
||||
'knowledgeFileUpload.confirmDelete': '确定要删除此文檔吗？',
|
||||
'knowledgeFileUpload.confirmParse': '确定要解析该文档吗？',
|
||||
'knowledgeFileUpload.nameRequired': '请输入文档名称',
|
||||
'knowledgeFileUpload.fileRequired': '请选择要上传的文件',
|
||||
'knowledgeFileUpload.getListFailed': '获取文档列表失败',
|
||||
'knowledgeFileUpload.parseCancelled': '已取消解析',
|
||||
'knowledgeFileUpload.deleteCancelled': '已取消删除',
|
||||
'knowledgeFileUpload.selectFilesFirst': '请先选择要删除的文件',
|
||||
'knowledgeFileUpload.selectAll': '全选',
|
||||
'knowledgeFileUpload.deselectAll': '取消全选',
|
||||
'knowledgeFileUpload.batchDelete': '删除',
|
||||
'knowledgeFileUpload.confirmBatchDelete': '确定要删除选中的{count}个文件吗？',
|
||||
'knowledgeFileUpload.batchDeleteSuccess': '成功删除{count}个文件',
|
||||
'knowledgeFileUpload.batchDeleteFailed': '批量删除失败',
|
||||
'knowledgeFileUpload.sliceCount': '切片数量',
|
||||
'knowledgeFileUpload.add': '新增',
|
||||
'knowledgeFileUpload.retrievalTest': '召回测试',
|
||||
'knowledgeFileUpload.testQuestion': '测试问题',
|
||||
'knowledgeFileUpload.testQuestionPlaceholder': '请输入要测试的问题',
|
||||
'knowledgeFileUpload.executeTest': '执行测试',
|
||||
'knowledgeFileUpload.testResult': '测试结果：',
|
||||
'knowledgeFileUpload.selectedFiles': '已选择文件',
|
||||
'knowledgeFileUpload.totalSlices': '共{total}条记录',
|
||||
'knowledgeFileUpload.slice': '切片',
|
||||
'knowledgeFileUpload.noSliceData': '暂无切片数据',
|
||||
'knowledgeFileUpload.firstPage': '首页',
|
||||
'knowledgeFileUpload.prevPage': '上一页',
|
||||
'knowledgeFileUpload.nextPage': '下一页',
|
||||
'knowledgeFileUpload.totalRecords': '共{total}条记录',
|
||||
'knowledgeFileUpload.testQuestion': '测试问题',
|
||||
'knowledgeFileUpload.testQuestionPlaceholder': '请输入要测试的问题',
|
||||
'knowledgeFileUpload.runTest': '执行测试',
|
||||
'knowledgeFileUpload.testResult': '测试结果：',
|
||||
'knowledgeFileUpload.noRelatedSlices': '未找到相关切片',
|
||||
'knowledgeFileUpload.comprehensiveSimilarity': '综合相似度',
|
||||
'knowledgeFileUpload.content': '内容:',
|
||||
'knowledgeFileUpload.testQuestionRequired': '请输入测试问题',
|
||||
}
|
||||
@@ -9,6 +9,7 @@ export default {
|
||||
// HeaderBar组件文本
|
||||
'header.smartManagement': '智能體管理',
|
||||
'header.modelConfig': '模型配置',
|
||||
'header.knowledgeBase': '知識庫',
|
||||
'header.userManagement': '用戶管理',
|
||||
'header.voiceCloneManagement': '音色克隆',
|
||||
'header.voiceResourceManagement': '音色資源',
|
||||
@@ -632,6 +633,7 @@ export default {
|
||||
'common.confirm': '確定',
|
||||
'common.cancel': '取消',
|
||||
'common.sensitive': '敏感',
|
||||
'common.loading': '載入中',
|
||||
|
||||
// 語言切換
|
||||
'language.zhCN': '中文简体',
|
||||
@@ -793,6 +795,7 @@ export default {
|
||||
'modelConfig.intent': '意圖識別',
|
||||
'modelConfig.tts': '語音合成',
|
||||
'modelConfig.memory': '記憶',
|
||||
'modelConfig.rag': '知識庫',
|
||||
'modelConfig.modelId': '模型ID',
|
||||
'modelConfig.modelName': '模型名稱',
|
||||
'modelConfig.provider': '提供商',
|
||||
@@ -815,6 +818,7 @@ export default {
|
||||
'modelConfig.partialDeleteFailed': '部分刪除失敗',
|
||||
'modelConfig.deleteSuccess': '刪除成功',
|
||||
'modelConfig.deleteFailed': '刪除失敗',
|
||||
'modelConfig.deleteCancelled': '刪除已取消',
|
||||
'modelConfig.duplicateSuccess': '創建副本成功',
|
||||
'modelConfig.duplicateFailed': '創建副本失敗',
|
||||
'modelConfig.saveSuccess': '保存成功',
|
||||
@@ -942,6 +946,7 @@ export default {
|
||||
'providerManagement.modelType.Memory': '記憶模組',
|
||||
'providerManagement.modelType.VAD': '語音活動檢測',
|
||||
'providerManagement.modelType.Plugin': '插件工具',
|
||||
'providerManagement.modelType.RAG': '知識庫',
|
||||
|
||||
// Provider Dialog 翻譯
|
||||
'providerDialog.category': '類別',
|
||||
@@ -965,6 +970,7 @@ export default {
|
||||
'providerDialog.booleanType': '布林值',
|
||||
'providerDialog.dictType': '字典',
|
||||
'providerDialog.arrayType': '分號分隔的列表',
|
||||
'providerDialog.ragType': '知識庫',
|
||||
'providerDialog.defaultValue': '預設值',
|
||||
'providerDialog.inputDefaultValue': '請輸入預設值',
|
||||
'providerDialog.operation': '操作',
|
||||
@@ -1118,4 +1124,138 @@ export default {
|
||||
'voiceClone.updateNameFailed': '名稱更新失敗',
|
||||
'voiceClone.playFailed': '播放失敗',
|
||||
'voiceClone.Details': '錯誤詳情',
|
||||
|
||||
// 知識庫管理頁面文本
|
||||
'knowledgeBaseManagement.title': '知識庫',
|
||||
'knowledgeBaseManagement.searchPlaceholder': '請輸入知識庫名稱搜尋',
|
||||
'knowledgeBaseManagement.search': '搜尋',
|
||||
'knowledgeBaseManagement.name': '知識庫名稱',
|
||||
'knowledgeBaseManagement.description': '知識庫描述',
|
||||
'knowledgeBaseManagement.documentCount': '文檔數量',
|
||||
'knowledgeBaseManagement.status': '啟用',
|
||||
'knowledgeBaseManagement.createdAt': '建立時間',
|
||||
'knowledgeBaseManagement.operation': '操作',
|
||||
'knowledgeBaseManagement.add': '新增',
|
||||
'knowledgeBaseManagement.delete': '刪除',
|
||||
'knowledgeBaseManagement.edit': '編輯',
|
||||
'knowledgeBaseManagement.itemsPerPage': '條/頁',
|
||||
'knowledgeBaseManagement.firstPage': '首頁',
|
||||
'knowledgeBaseManagement.prevPage': '上一頁',
|
||||
'knowledgeBaseManagement.nextPage': '下一頁',
|
||||
'knowledgeBaseManagement.totalRecords': '共{total}條記錄',
|
||||
'knowledgeBaseManagement.addKnowledgeBase': '新增知識庫',
|
||||
'knowledgeBaseManagement.editKnowledgeBase': '編輯知識庫',
|
||||
'knowledgeBaseManagement.getKnowledgeBaseListFailed': '獲取知識庫列表失敗',
|
||||
'knowledgeBaseManagement.selectKnowledgeBaseFirst': '請先選擇需要刪除的知識庫',
|
||||
'knowledgeBaseManagement.confirmBatchDelete': '確定要刪除選中的{count}個知識庫嗎？',
|
||||
'knowledgeBaseManagement.batchDeleteSuccess': '成功刪除{count}個知識庫',
|
||||
'knowledgeBaseManagement.deleteFailed': '刪除失敗，請重試',
|
||||
'knowledgeBaseManagement.operationCancelled': '已取消刪除操作',
|
||||
'knowledgeBaseManagement.updateSuccess': '修改成功',
|
||||
'knowledgeBaseManagement.addSuccess': '新增成功',
|
||||
'knowledgeBaseManagement.updateFailed': '更新失敗',
|
||||
'knowledgeBaseManagement.addFailed': '新增失敗',
|
||||
'knowledgeBaseManagement.selectAll': '全選',
|
||||
'knowledgeBaseManagement.cancelSelectAll': '取消全選',
|
||||
|
||||
// 知識庫對話框文本
|
||||
'knowledgeBaseDialog.title': '知識庫',
|
||||
'knowledgeBaseDialog.name': '知識庫名稱',
|
||||
'knowledgeBaseDialog.namePlaceholder': '請精確輸入知識庫名稱，才能提高知識庫呼叫的準確性',
|
||||
'knowledgeBaseDialog.description': '知識庫描述',
|
||||
'knowledgeBaseDialog.descriptionPlaceholder': '請詳細輸入知識庫描述，以便大模型更好地理解這個知識庫的总体內容',
|
||||
'knowledgeBaseDialog.ragModel': 'RAG模型',
|
||||
'knowledgeBaseDialog.ragModelPlaceholder': '請選擇RAG模型',
|
||||
'knowledgeBaseDialog.ragModelRequired': '請選擇RAG模型',
|
||||
'knowledgeBaseDialog.loadRAGModelsFailed': '載入RAG模型列表失敗',
|
||||
'knowledgeBaseDialog.status': '狀態',
|
||||
'knowledgeBaseDialog.statusEnabled': '啟用',
|
||||
'knowledgeBaseDialog.statusDisabled': '禁用',
|
||||
'knowledgeBaseDialog.save': '儲存',
|
||||
'knowledgeBaseDialog.cancel': '取消',
|
||||
'knowledgeBaseDialog.confirm': '確認',
|
||||
'knowledgeBaseDialog.requiredName': '請輸入知識庫名稱',
|
||||
'knowledgeBaseDialog.nameRequired': '請輸入知識庫名稱',
|
||||
'knowledgeBaseDialog.nameLength': '知識庫名稱長度在1到50個字符之間',
|
||||
'knowledgeBaseDialog.namePattern': '知識庫名稱只能包含中文、英文、數字、空格、下劃線和連字符',
|
||||
'knowledgeBaseDialog.descriptionLength': '知識庫描述不能超過200個字符',
|
||||
'knowledgeBaseDialog.nameLengthLimit': '知識庫名稱不能超過50個字符',
|
||||
'knowledgeBaseDialog.descriptionLengthLimit': '知識庫描述不能超過200個字符',
|
||||
|
||||
// 知識庫管理頁面查看按鈕文本
|
||||
'knowledgeBaseManagement.view': '查看',
|
||||
|
||||
// 知識庫文件上傳頁面文本
|
||||
'knowledgeFileUpload.back': '返回',
|
||||
'knowledgeFileUpload.searchPlaceholder': '請輸入文檔名稱搜尋',
|
||||
'knowledgeFileUpload.search': '搜尋',
|
||||
'knowledgeFileUpload.addDocument': '新增文檔',
|
||||
'knowledgeFileUpload.documentName': '文檔名稱',
|
||||
'knowledgeFileUpload.uploadTime': '上傳時間',
|
||||
'knowledgeFileUpload.status': '狀態',
|
||||
'knowledgeFileUpload.operation': '操作',
|
||||
'knowledgeFileUpload.parse': '解析',
|
||||
'knowledgeFileUpload.viewSlices': '查看切片',
|
||||
'knowledgeFileUpload.delete': '刪除',
|
||||
'knowledgeFileUpload.itemsPerPage': '條/頁',
|
||||
'knowledgeFileUpload.firstPage': '首頁',
|
||||
'knowledgeFileUpload.prevPage': '上一頁',
|
||||
'knowledgeFileUpload.nextPage': '下一頁',
|
||||
'knowledgeFileUpload.totalRecords': '共{total}條記錄',
|
||||
'knowledgeFileUpload.uploadDocument': '上傳文檔',
|
||||
'knowledgeFileUpload.documentNamePlaceholder': '請輸入文檔名稱',
|
||||
'knowledgeFileUpload.file': '文件',
|
||||
'knowledgeFileUpload.clickToUpload': '點擊上傳',
|
||||
'knowledgeFileUpload.uploadTip': '支持的文件類型：PDF、DOC、DOCX、TXT、MD、CSV、XLS、XLSX、PPT、PPTX，单次批量上傳文件數不超過 32 個，每個文件大小不超過 10MB',
|
||||
'knowledgeFileUpload.dragOrClick': '將文件拖到此處，或點擊上傳',
|
||||
'knowledgeFileUpload.cancel': '取消',
|
||||
'knowledgeFileUpload.confirm': '確定',
|
||||
'knowledgeFileUpload.knowledgeBaseName': '知識庫名稱',
|
||||
'knowledgeFileUpload.statusNotStarted': '未開始',
|
||||
'knowledgeFileUpload.statusProcessing': '處理中',
|
||||
'knowledgeFileUpload.statusCancelled': '已取消',
|
||||
'knowledgeFileUpload.statusCompleted': '已完成',
|
||||
'knowledgeFileUpload.uploadSuccess': '文檔上傳成功',
|
||||
'knowledgeFileUpload.uploadFailed': '文檔上傳失敗',
|
||||
'knowledgeFileUpload.parseSuccess': '文檔解析成功',
|
||||
'knowledgeFileUpload.parseFailed': '文檔解析失敗',
|
||||
'knowledgeFileUpload.deleteSuccess': '文檔刪除成功',
|
||||
'knowledgeFileUpload.deleteFailed': '文檔刪除失敗',
|
||||
'knowledgeFileUpload.confirmDelete': '確定要刪除此文檔嗎？',
|
||||
'knowledgeFileUpload.confirmParse': '確定要解析該文檔嗎？',
|
||||
'knowledgeFileUpload.nameRequired': '請輸入文檔名稱',
|
||||
'knowledgeFileUpload.fileRequired': '請選擇要上傳的文件',
|
||||
'knowledgeFileUpload.getListFailed': '獲取文檔列表失敗',
|
||||
'knowledgeFileUpload.parseCancelled': '已取消解析',
|
||||
'knowledgeFileUpload.deleteCancelled': '已取消删除',
|
||||
'knowledgeFileUpload.selectFilesFirst': '請先選擇要刪除的文件',
|
||||
'knowledgeFileUpload.selectAll': '全選',
|
||||
'knowledgeFileUpload.deselectAll': '取消全選',
|
||||
'knowledgeFileUpload.batchDelete': '刪除',
|
||||
'knowledgeFileUpload.confirmBatchDelete': '確定要刪除選中的{count}個文件嗎？',
|
||||
'knowledgeFileUpload.batchDeleteSuccess': '成功刪除{count}個文件',
|
||||
'knowledgeFileUpload.batchDeleteFailed': '批量刪除失敗',
|
||||
'knowledgeFileUpload.sliceCount': '切片數量',
|
||||
'knowledgeFileUpload.add': '新增',
|
||||
'knowledgeFileUpload.retrievalTest': '召回測試',
|
||||
'knowledgeFileUpload.testQuestion': '測試問題',
|
||||
'knowledgeFileUpload.testQuestionPlaceholder': '請輸入要測試的問題',
|
||||
'knowledgeFileUpload.executeTest': '執行測試',
|
||||
'knowledgeFileUpload.testResult': '測試結果：',
|
||||
'knowledgeFileUpload.selectedFiles': '已選擇文件',
|
||||
'knowledgeFileUpload.totalSlices': '共{total}條記錄',
|
||||
'knowledgeFileUpload.slice': '切片',
|
||||
'knowledgeFileUpload.noSliceData': '暫無切片數據',
|
||||
'knowledgeFileUpload.firstPage': '首頁',
|
||||
'knowledgeFileUpload.prevPage': '上一頁',
|
||||
'knowledgeFileUpload.nextPage': '下一頁',
|
||||
'knowledgeFileUpload.totalRecords': '共{total}條記錄',
|
||||
'knowledgeFileUpload.testQuestion': '測試問題',
|
||||
'knowledgeFileUpload.testQuestionPlaceholder': '請輸入要測試的問題',
|
||||
'knowledgeFileUpload.runTest': '執行測試',
|
||||
'knowledgeFileUpload.testResult': '測試結果：',
|
||||
'knowledgeFileUpload.noRelatedSlices': '未找到相關切片',
|
||||
'knowledgeFileUpload.comprehensiveSimilarity': '綜合相似度',
|
||||
'knowledgeFileUpload.content': '內容:',
|
||||
'knowledgeFileUpload.testQuestionRequired': '請輸入測試問題',
|
||||
}
|
||||
@@ -76,8 +76,7 @@ const routes = [
|
||||
return import('../views/ModelConfig.vue')
|
||||
}
|
||||
},
|
||||
{
|
||||
path: '/params-management',
|
||||
{ path: '/params-management',
|
||||
name: 'ParamsManagement',
|
||||
component: function () {
|
||||
return import('../views/ParamsManagement.vue')
|
||||
@@ -87,6 +86,26 @@ const routes = [
|
||||
title: '参数管理'
|
||||
}
|
||||
},
|
||||
{ path: '/knowledge-base-management',
|
||||
name: 'KnowledgeBaseManagement',
|
||||
component: function () {
|
||||
return import('../views/KnowledgeBaseManagement.vue')
|
||||
},
|
||||
meta: {
|
||||
requiresAuth: true,
|
||||
title: '知识库管理'
|
||||
}
|
||||
},
|
||||
{ path: '/knowledge-file-upload',
|
||||
name: 'KnowledgeFileUpload',
|
||||
component: function () {
|
||||
return import('../views/KnowledgeFileUpload.vue')
|
||||
},
|
||||
meta: {
|
||||
requiresAuth: true,
|
||||
title: '文档上传管理'
|
||||
}
|
||||
},
|
||||
|
||||
{
|
||||
path: '/server-side-management',
|
||||
@@ -183,7 +202,7 @@ VueRouter.prototype.push = function push(location) {
|
||||
}
|
||||
|
||||
// 需要登录才能访问的路由
|
||||
const protectedRoutes = ['home', 'RoleConfig', 'DeviceManagement', 'UserManagement', 'ModelConfig']
|
||||
const protectedRoutes = ['home', 'RoleConfig', 'DeviceManagement', 'UserManagement', 'ModelConfig', 'KnowledgeBaseManagement', 'KnowledgeFileUpload']
|
||||
|
||||
// 路由守卫
|
||||
router.beforeEach((to, from, next) => {
|
||||
|
||||
@@ -0,0 +1,806 @@
|
||||
<template>
|
||||
<div class="welcome">
|
||||
<HeaderBar />
|
||||
|
||||
<div class="operation-bar">
|
||||
<h2 class="page-title">{{ $t('knowledgeBaseManagement.title') }}</h2>
|
||||
<div class="right-operations">
|
||||
<el-input :placeholder="$t('knowledgeBaseManagement.searchPlaceholder')" v-model="searchName" class="search-input"
|
||||
@keyup.enter.native="handleSearch" clearable />
|
||||
<el-button class="btn-search" @click="handleSearch">{{ $t('knowledgeBaseManagement.search') }}</el-button>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div class="main-wrapper">
|
||||
<div class="content-panel">
|
||||
<div class="content-area">
|
||||
<el-card class="params-card" shadow="never">
|
||||
<div>
|
||||
<el-table ref="paramsTable" :data="knowledgeBaseList" class="transparent-table" v-loading="loading"
|
||||
:element-loading-text="$t('common.loading')" element-loading-spinner="el-icon-loading"
|
||||
element-loading-background="rgba(255, 255, 255, 0.7)"
|
||||
:header-cell-class-name="headerCellClassName" @selection-change="handleSelectionChange">
|
||||
<el-table-column type="selection" width="55" align="center" />
|
||||
<el-table-column :label="$t('knowledgeBaseManagement.name')" prop="name" align="center">
|
||||
<template slot-scope="scope">
|
||||
<span class="knowledge-base-name">{{ scope.row.name }}</span>
|
||||
</template>
|
||||
</el-table-column>
|
||||
<el-table-column :label="$t('knowledgeBaseManagement.description')" prop="description" align="center" show-overflow-tooltip>
|
||||
<template slot-scope="scope">
|
||||
<span>{{ scope.row.description || '-' }}</span>
|
||||
</template>
|
||||
</el-table-column>
|
||||
<el-table-column :label="$t('knowledgeBaseManagement.documentCount')" align="center">
|
||||
<template slot-scope="scope">
|
||||
<span>{{ scope.row.documentCount || 0 }}</span>
|
||||
</template>
|
||||
</el-table-column>
|
||||
<el-table-column :label="$t('knowledgeBaseManagement.status')" align="center">
|
||||
<template slot-scope="scope">
|
||||
<el-switch
|
||||
v-model="scope.row.status"
|
||||
:active-value="1"
|
||||
:inactive-value="0"
|
||||
active-color="#13ce66"
|
||||
inactive-color="#909399"
|
||||
@change="handleStatusChange(scope.row)"
|
||||
></el-switch>
|
||||
</template>
|
||||
</el-table-column>
|
||||
<el-table-column :label="$t('knowledgeBaseManagement.createdAt')" prop="createdAt" align="center">
|
||||
<template slot-scope="scope">
|
||||
<span>{{ formatDate(scope.row.createdAt) }}</span>
|
||||
</template>
|
||||
</el-table-column>
|
||||
<el-table-column :label="$t('knowledgeBaseManagement.operation')" align="center">
|
||||
<template slot-scope="scope">
|
||||
<el-button size="mini" type="text" @click="showViewDialog(scope.row)">
|
||||
{{ $t('knowledgeBaseManagement.view') }}
|
||||
</el-button>
|
||||
<el-button size="mini" type="text" @click="showEditDialog(scope.row)">
|
||||
{{ $t('knowledgeBaseManagement.edit') }}
|
||||
</el-button>
|
||||
<el-button size="mini" type="text" @click="deleteSingleKnowledgeBase(scope.row)">
|
||||
{{ $t('knowledgeBaseManagement.delete') }}
|
||||
</el-button>
|
||||
</template>
|
||||
</el-table-column>
|
||||
</el-table>
|
||||
</div>
|
||||
|
||||
<div class="table_bottom">
|
||||
<div class="ctrl_btn">
|
||||
<el-button size="mini" type="primary" @click="toggleSelectAll">
|
||||
{{ isAllSelected ? $t('knowledgeBaseManagement.cancelSelectAll') : $t('knowledgeBaseManagement.selectAll') }}
|
||||
</el-button>
|
||||
<el-button size="mini" type="success" @click="showAddDialog">
|
||||
{{ $t('knowledgeBaseManagement.add') }}
|
||||
</el-button>
|
||||
<el-button size="mini" type="danger" icon="el-icon-delete" @click="deleteSelectedKnowledgeBase" :disabled="selectedKnowledgeBase.length === 0">
|
||||
{{ $t('knowledgeBaseManagement.delete') }}
|
||||
</el-button>
|
||||
</div>
|
||||
<div class="custom-pagination">
|
||||
<el-select v-model="pageSize" @change="handlePageSizeChange" class="page-size-select">
|
||||
<el-option v-for="item in pageSizeOptions" :key="item"
|
||||
:label="`${item}${$t('knowledgeBaseManagement.itemsPerPage')}`" :value="item">
|
||||
</el-option>
|
||||
</el-select>
|
||||
<button class="pagination-btn" :disabled="currentPage === 1" @click="goFirst">
|
||||
{{ $t('knowledgeBaseManagement.firstPage') }}
|
||||
</button>
|
||||
<button class="pagination-btn" :disabled="currentPage === 1" @click="goPrev">
|
||||
{{ $t('knowledgeBaseManagement.prevPage') }}
|
||||
</button>
|
||||
<button v-for="page in visiblePages" :key="page" class="pagination-btn"
|
||||
:class="{ active: page === currentPage }" @click="goToPage(page)">
|
||||
{{ page }}
|
||||
</button>
|
||||
<button class="pagination-btn" :disabled="currentPage === pageCount" @click="goNext">
|
||||
{{ $t('knowledgeBaseManagement.nextPage') }}
|
||||
</button>
|
||||
<span class="total-text">{{ $t('knowledgeBaseManagement.totalRecords', { total }) }}</span>
|
||||
</div>
|
||||
</div>
|
||||
</el-card>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<!-- 新增/编辑知识库对话框 -->
|
||||
<knowledge-base-dialog ref="knowledgeBaseDialog" :title="dialogTitle" :visible.sync="dialogVisible" :form="knowledgeBaseForm"
|
||||
@submit="handleSubmit" @cancel="dialogVisible = false" />
|
||||
|
||||
<el-footer>
|
||||
<version-footer />
|
||||
</el-footer>
|
||||
</div>
|
||||
</template>
|
||||
|
||||
<script>
|
||||
import Api from "@/apis/api";
|
||||
import HeaderBar from "@/components/HeaderBar.vue";
|
||||
import VersionFooter from "@/components/VersionFooter.vue";
|
||||
import KnowledgeBaseDialog from "@/components/KnowledgeBaseDialog.vue";
|
||||
|
||||
export default {
|
||||
components: { HeaderBar, VersionFooter, KnowledgeBaseDialog },
|
||||
data() {
|
||||
return {
|
||||
searchName: "",
|
||||
knowledgeBaseList: [],
|
||||
currentPage: 1,
|
||||
loading: false,
|
||||
pageSize: 10,
|
||||
pageSizeOptions: [10, 20, 50, 100],
|
||||
total: 0,
|
||||
dialogVisible: false,
|
||||
dialogTitle: "",
|
||||
selectedKnowledgeBase: [],
|
||||
isAllSelected: false,
|
||||
knowledgeBaseForm: {
|
||||
id: null,
|
||||
datasetId: null,
|
||||
name: "",
|
||||
description: "",
|
||||
status: 1
|
||||
}
|
||||
};
|
||||
},
|
||||
created() {
|
||||
this.fetchKnowledgeBaseList();
|
||||
},
|
||||
computed: {
|
||||
pageCount() {
|
||||
return Math.ceil(this.total / this.pageSize);
|
||||
},
|
||||
visiblePages() {
|
||||
const pages = [];
|
||||
const maxVisible = 3;
|
||||
let start = Math.max(1, this.currentPage - 1);
|
||||
let end = Math.min(this.pageCount, start + maxVisible - 1);
|
||||
|
||||
if (end - start + 1 < maxVisible) {
|
||||
start = Math.max(1, end - maxVisible + 1);
|
||||
}
|
||||
|
||||
for (let i = start; i <= end; i++) {
|
||||
pages.push(i);
|
||||
}
|
||||
return pages;
|
||||
},
|
||||
},
|
||||
methods: {
|
||||
handlePageSizeChange: function(val) {
|
||||
this.pageSize = val;
|
||||
this.currentPage = 1;
|
||||
this.fetchKnowledgeBaseList();
|
||||
},
|
||||
fetchKnowledgeBaseList: function() {
|
||||
this.loading = true;
|
||||
Api.knowledgeBase.getKnowledgeBaseList(
|
||||
{
|
||||
page: this.currentPage,
|
||||
page_size: this.pageSize,
|
||||
name: this.searchName,
|
||||
},
|
||||
(res) => {
|
||||
this.loading = false;
|
||||
console.log('getKnowledgeBaseList response:', res); // 添加调试日志
|
||||
|
||||
// 修复：从 res.data 获取分页数据，而不是 res.data.data
|
||||
// 因为 knowledgeBase.js 直接传递了整个响应对象
|
||||
if (res.data && res.data.code === 0) {
|
||||
const pageData = res.data.data || {};
|
||||
this.knowledgeBaseList = pageData.list || [];
|
||||
this.total = pageData.total || 0;
|
||||
console.log('Updated knowledgeBaseList:', this.knowledgeBaseList); // 添加调试日志
|
||||
} else {
|
||||
this.$message.error({
|
||||
message: res.data?.msg || this.$t('knowledgeBaseManagement.getKnowledgeBaseListFailed'),
|
||||
showClose: true
|
||||
});
|
||||
}
|
||||
},
|
||||
() => {
|
||||
this.loading = false;
|
||||
this.$message.error(this.$t('knowledgeBaseManagement.getKnowledgeBaseListFailed'));
|
||||
}
|
||||
);
|
||||
},
|
||||
handleSearch: function() {
|
||||
this.currentPage = 1;
|
||||
this.fetchKnowledgeBaseList();
|
||||
},
|
||||
handleSelectionChange: function(val) {
|
||||
this.selectedKnowledgeBase = val;
|
||||
},
|
||||
toggleSelectAll: function() {
|
||||
if (this.isAllSelected) {
|
||||
// 取消全选
|
||||
this.$refs.paramsTable.clearSelection();
|
||||
this.isAllSelected = false;
|
||||
} else {
|
||||
// 全选
|
||||
this.knowledgeBaseList.forEach(row => {
|
||||
this.$refs.paramsTable.toggleRowSelection(row, true);
|
||||
});
|
||||
this.isAllSelected = true;
|
||||
}
|
||||
},
|
||||
headerCellClassName: function({ row, column, rowIndex, columnIndex }) {
|
||||
if (columnIndex === 0) {
|
||||
return 'header-cell-first';
|
||||
}
|
||||
return 'header-cell';
|
||||
},
|
||||
showAddDialog: function() {
|
||||
console.log('showAddDialog called');
|
||||
this.dialogTitle = this.$t('knowledgeBaseManagement.addKnowledgeBase');
|
||||
this.knowledgeBaseForm = {
|
||||
id: null,
|
||||
datasetId: null,
|
||||
name: "",
|
||||
description: "",
|
||||
status: 1
|
||||
};
|
||||
this.dialogVisible = true;
|
||||
console.log('dialogVisible set to:', this.dialogVisible);
|
||||
},
|
||||
showViewDialog: function(row) {
|
||||
// 跳转到上传文件页面，传递知识库ID和名称
|
||||
this.$router.push({
|
||||
path: '/knowledge-file-upload',
|
||||
query: {
|
||||
datasetId: row.datasetId,
|
||||
knowledgeBaseName: row.name
|
||||
}
|
||||
});
|
||||
},
|
||||
showEditDialog: function(row) {
|
||||
this.dialogTitle = this.$t('knowledgeBaseManagement.editKnowledgeBase');
|
||||
this.knowledgeBaseForm = {
|
||||
id: row.id,
|
||||
datasetId: row.datasetId,
|
||||
name: row.name,
|
||||
description: row.description || "",
|
||||
status: row.status,
|
||||
ragModelId: row.ragModelId || null
|
||||
};
|
||||
this.dialogVisible = true;
|
||||
},
|
||||
handleSubmit: function(form) {
|
||||
console.log('handleSubmit called with form:', form);
|
||||
if (form.id) {
|
||||
// 编辑 - 使用dataset_id作为路径参数，form作为请求体
|
||||
console.log('Editing knowledge base:', form.datasetId);
|
||||
Api.knowledgeBase.updateKnowledgeBase(form.datasetId, form, (res) => {
|
||||
console.log('Update response:', res);
|
||||
// 修复：检查 res.data.code 而不是 res.code
|
||||
if (res.data && res.data.code === 0) {
|
||||
this.dialogVisible = false;
|
||||
this.fetchKnowledgeBaseList();
|
||||
this.$message.success(this.$t('knowledgeBaseManagement.updateSuccess'));
|
||||
} else {
|
||||
this.$message.error(res.data?.msg || this.$t('knowledgeBaseManagement.updateFailed'));
|
||||
}
|
||||
}, () => {
|
||||
this.$message.error(this.$t('knowledgeBaseManagement.updateFailed'));
|
||||
});
|
||||
} else {
|
||||
// 新增 - 只传递必要的字段，不传递id
|
||||
const createData = {
|
||||
name: form.name,
|
||||
description: form.description,
|
||||
status: form.status,
|
||||
ragModelId: form.ragModelId
|
||||
};
|
||||
console.log('Creating knowledge base with data:', createData);
|
||||
Api.knowledgeBase.createKnowledgeBase(createData, (res) => {
|
||||
console.log('Create response:', res);
|
||||
if (res.data && res.data.code === 0) {
|
||||
this.dialogVisible = false;
|
||||
this.fetchKnowledgeBaseList();
|
||||
this.$message.success(this.$t('knowledgeBaseManagement.addSuccess'));
|
||||
}
|
||||
}, (err) => {
|
||||
console.log('Error callback received:', err);
|
||||
// 错误回调处理后端返回的错误信息
|
||||
if (err && err.data) {
|
||||
console.log('后端返回错误消息:', err.data.msg || err.msg);
|
||||
this.$message.error(err.data.msg || err.msg || this.$t('knowledgeBaseManagement.addFailed'));
|
||||
} else {
|
||||
this.$message.error(this.$t('knowledgeBaseManagement.addFailed'));
|
||||
}
|
||||
});
|
||||
}
|
||||
},
|
||||
deleteSelectedKnowledgeBase: function() {
|
||||
if (this.selectedKnowledgeBase.length === 0) {
|
||||
this.$message.warning(this.$t('knowledgeBaseManagement.selectKnowledgeBaseFirst'));
|
||||
return;
|
||||
}
|
||||
|
||||
this.$confirm(
|
||||
this.$t('knowledgeBaseManagement.confirmBatchDelete', { count: this.selectedKnowledgeBase.length }),
|
||||
this.$t('common.warning'),
|
||||
{
|
||||
confirmButtonText: this.$t('common.confirm'),
|
||||
cancelButtonText: this.$t('common.cancel'),
|
||||
type: 'warning'
|
||||
}).then(() => {
|
||||
const ids = this.selectedKnowledgeBase.map(item => item.datasetId).join(',');
|
||||
Api.knowledgeBase.deleteKnowledgeBases(ids, (res) => {
|
||||
if (res.data && res.data.code === 0) {
|
||||
this.fetchKnowledgeBaseList();
|
||||
this.$message.success(this.$t('knowledgeBaseManagement.batchDeleteSuccess', { count: this.selectedKnowledgeBase.length }));
|
||||
} else {
|
||||
this.$message.error(res.data?.msg || this.$t('knowledgeBaseManagement.deleteFailed'));
|
||||
}
|
||||
}, () => {
|
||||
this.$message.error(this.$t('knowledgeBaseManagement.deleteFailed'));
|
||||
});
|
||||
}).catch(() => {
|
||||
this.$message({
|
||||
type: 'info',
|
||||
message: this.$t('knowledgeBaseManagement.operationCancelled'),
|
||||
duration: 1000
|
||||
});
|
||||
});
|
||||
},
|
||||
deleteSingleKnowledgeBase: function(row) {
|
||||
this.$confirm(
|
||||
this.$t('knowledgeBaseManagement.confirmBatchDelete', { count: 1 }),
|
||||
this.$t('common.warning'),
|
||||
{
|
||||
confirmButtonText: this.$t('common.confirm'),
|
||||
cancelButtonText: this.$t('common.cancel'),
|
||||
type: 'warning'
|
||||
}).then(() => {
|
||||
Api.knowledgeBase.deleteKnowledgeBase(row.datasetId, (res) => {
|
||||
if (res.data && res.data.code === 0) {
|
||||
this.fetchKnowledgeBaseList();
|
||||
this.$message.success(this.$t('knowledgeBaseManagement.batchDeleteSuccess', { count: 1 }));
|
||||
} else {
|
||||
this.$message.error(res.data?.msg || this.$t('knowledgeBaseManagement.deleteFailed'));
|
||||
}
|
||||
}, () => {
|
||||
this.$message.error(this.$t('knowledgeBaseManagement.deleteFailed'));
|
||||
});
|
||||
}).catch(() => {
|
||||
this.$message({
|
||||
type: 'info',
|
||||
message: this.$t('knowledgeBaseManagement.operationCancelled'),
|
||||
duration: 1000
|
||||
});
|
||||
});
|
||||
},
|
||||
handleStatusChange: function(row) {
|
||||
// 只传递需要更新的字段，确保包含id字段
|
||||
const updateForm = {
|
||||
id: row.id, // 添加id字段，后端需要此字段来定位记录
|
||||
datasetId: row.datasetId,
|
||||
name: row.name,
|
||||
description: row.description,
|
||||
status: row.status
|
||||
};
|
||||
console.log('Updating knowledge base status:', updateForm); // 添加调试日志
|
||||
Api.knowledgeBase.updateKnowledgeBase(row.datasetId, updateForm, (res) => {
|
||||
console.log('Status update response:', res); // 添加调试日志
|
||||
if (res.data && res.data.code !== 0) {
|
||||
// 恢复原来的状态
|
||||
this.fetchKnowledgeBaseList();
|
||||
this.$message.error(res.data?.msg || this.$t('knowledgeBaseManagement.updateFailed'));
|
||||
} else {
|
||||
// 更新成功，显示成功消息
|
||||
this.$message.success(this.$t('knowledgeBaseManagement.updateSuccess'));
|
||||
}
|
||||
}, () => {
|
||||
// 恢复原来的状态
|
||||
this.fetchKnowledgeBaseList();
|
||||
this.$message.error(this.$t('knowledgeBaseManagement.updateFailed'));
|
||||
});
|
||||
},
|
||||
goToPage: function(page) {
|
||||
if (page !== this.currentPage) {
|
||||
this.currentPage = page;
|
||||
this.fetchKnowledgeBaseList();
|
||||
}
|
||||
},
|
||||
goFirst: function() {
|
||||
if (this.currentPage !== 1) {
|
||||
this.currentPage = 1;
|
||||
this.fetchKnowledgeBaseList();
|
||||
}
|
||||
},
|
||||
goPrev: function() {
|
||||
if (this.currentPage > 1) {
|
||||
this.currentPage--;
|
||||
this.fetchKnowledgeBaseList();
|
||||
}
|
||||
},
|
||||
goNext: function() {
|
||||
if (this.currentPage < this.pageCount) {
|
||||
this.currentPage++;
|
||||
this.fetchKnowledgeBaseList();
|
||||
}
|
||||
},
|
||||
formatDate: function(dateString) {
|
||||
if (!dateString) return '';
|
||||
const date = new Date(dateString);
|
||||
return `${date.getFullYear()}-${String(date.getMonth() + 1).padStart(2, '0')}-${String(date.getDate()).padStart(2, '0')} ${String(date.getHours()).padStart(2, '0')}:${String(date.getMinutes()).padStart(2, '0')}`;
|
||||
}
|
||||
}
|
||||
};
|
||||
</script>
|
||||
|
||||
<style lang="scss" scoped>
|
||||
.welcome {
|
||||
min-width: 900px;
|
||||
min-height: 506px;
|
||||
height: 100vh;
|
||||
display: flex;
|
||||
position: relative;
|
||||
flex-direction: column;
|
||||
background-size: cover;
|
||||
background: linear-gradient(to bottom right, #dce8ff, #e4eeff, #e6cbfd) center;
|
||||
-webkit-background-size: cover;
|
||||
-o-background-size: cover;
|
||||
overflow: hidden;
|
||||
}
|
||||
|
||||
.main-wrapper {
|
||||
margin: 5px 22px;
|
||||
border-radius: 15px;
|
||||
min-height: calc(100vh - 24vh);
|
||||
height: auto;
|
||||
max-height: 80vh;
|
||||
box-shadow: 0 2px 12px rgba(0, 0, 0, 0.1);
|
||||
position: relative;
|
||||
background: rgba(237, 242, 255, 0.5);
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
}
|
||||
|
||||
.operation-bar {
|
||||
display: flex;
|
||||
justify-content: space-between;
|
||||
align-items: center;
|
||||
padding: 16px 24px;
|
||||
}
|
||||
|
||||
.page-title {
|
||||
font-size: 24px;
|
||||
margin: 0;
|
||||
}
|
||||
|
||||
.right-operations {
|
||||
display: flex;
|
||||
gap: 10px;
|
||||
margin-left: auto;
|
||||
}
|
||||
|
||||
.search-input {
|
||||
width: 240px;
|
||||
}
|
||||
|
||||
.btn-search {
|
||||
background: linear-gradient(135deg, #6b8cff, #a966ff);
|
||||
border: none;
|
||||
color: white;
|
||||
}
|
||||
|
||||
.content-panel {
|
||||
flex: 1;
|
||||
display: flex;
|
||||
overflow: hidden;
|
||||
height: 100%;
|
||||
border-radius: 15px;
|
||||
background: transparent;
|
||||
border: 1px solid #fff;
|
||||
}
|
||||
|
||||
.content-area {
|
||||
flex: 1;
|
||||
height: 100%;
|
||||
min-width: 600px;
|
||||
overflow: auto;
|
||||
background-color: white;
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
}
|
||||
|
||||
.params-card {
|
||||
background: white;
|
||||
flex: 1;
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
border: none;
|
||||
box-shadow: none;
|
||||
overflow: hidden;
|
||||
|
||||
::v-deep .el-card__body {
|
||||
padding: 15px;
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
flex: 1;
|
||||
overflow: hidden;
|
||||
}
|
||||
}
|
||||
|
||||
.table_bottom {
|
||||
display: flex;
|
||||
justify-content: space-between;
|
||||
align-items: center;
|
||||
margin-top: auto;
|
||||
padding-bottom: 10px;
|
||||
width: 100%;
|
||||
}
|
||||
|
||||
.ctrl_btn {
|
||||
display: flex;
|
||||
gap: 8px;
|
||||
padding-left: 26px;
|
||||
|
||||
.el-button {
|
||||
min-width: 72px;
|
||||
height: 32px;
|
||||
padding: 7px 12px 7px 10px;
|
||||
font-size: 12px;
|
||||
border-radius: 4px;
|
||||
line-height: 1;
|
||||
font-weight: 500;
|
||||
border: none;
|
||||
transition: all 0.3s ease;
|
||||
box-shadow: 0 2px 6px rgba(0, 0, 0, 0.1);
|
||||
|
||||
&:hover {
|
||||
transform: translateY(-1px);
|
||||
box-shadow: 0 4px 8px rgba(0, 0, 0, 0.15);
|
||||
}
|
||||
}
|
||||
|
||||
.el-button--primary {
|
||||
background: #5f70f3;
|
||||
color: white;
|
||||
}
|
||||
|
||||
.el-button--danger {
|
||||
background: #fd5b63;
|
||||
color: white;
|
||||
}
|
||||
}
|
||||
|
||||
.custom-pagination {
|
||||
display: flex;
|
||||
align-items: center;
|
||||
gap: 5px;
|
||||
|
||||
.el-select {
|
||||
margin-right: 8px;
|
||||
}
|
||||
|
||||
.pagination-btn:first-child,
|
||||
.pagination-btn:nth-child(2),
|
||||
.pagination-btn:nth-last-child(2),
|
||||
.pagination-btn:nth-child(3) {
|
||||
min-width: 60px;
|
||||
height: 32px;
|
||||
padding: 0 12px;
|
||||
border-radius: 4px;
|
||||
border: 1px solid #e4e7ed;
|
||||
background: #dee7ff;
|
||||
color: #606266;
|
||||
font-size: 14px;
|
||||
cursor: pointer;
|
||||
transition: all 0.3s ease;
|
||||
|
||||
&:hover {
|
||||
background: #d7dce6;
|
||||
}
|
||||
|
||||
&:disabled {
|
||||
opacity: 0.6;
|
||||
cursor: not-allowed;
|
||||
}
|
||||
}
|
||||
|
||||
.pagination-btn:not(:first-child):not(:nth-child(3)):not(:nth-child(2)):not(:nth-last-child(2)) {
|
||||
min-width: 28px;
|
||||
height: 32px;
|
||||
padding: 0;
|
||||
border-radius: 4px;
|
||||
border: 1px solid transparent;
|
||||
background: transparent;
|
||||
color: #606266;
|
||||
font-size: 14px;
|
||||
cursor: pointer;
|
||||
transition: all 0.3s ease;
|
||||
|
||||
&:hover {
|
||||
background: rgba(245, 247, 250, 0.3);
|
||||
}
|
||||
}
|
||||
|
||||
.pagination-btn.active {
|
||||
background: #5f70f3 !important;
|
||||
color: #ffffff !important;
|
||||
border-color: #5f70f3 !important;
|
||||
|
||||
&:hover {
|
||||
background: #6d7cf5 !important;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
.total-text {
|
||||
margin-left: 10px;
|
||||
color: #606266;
|
||||
font-size: 14px;
|
||||
}
|
||||
|
||||
.page-size-select {
|
||||
width: 100px;
|
||||
margin-right: 10px;
|
||||
|
||||
:deep(.el-input__inner) {
|
||||
height: 32px;
|
||||
line-height: 32px;
|
||||
border-radius: 4px;
|
||||
border: 1px solid #e4e7ed;
|
||||
background: #dee7ff;
|
||||
color: #606266;
|
||||
font-size: 14px;
|
||||
}
|
||||
|
||||
:deep(.el-input__suffix) {
|
||||
right: 6px;
|
||||
width: 15px;
|
||||
height: 20px;
|
||||
display: flex;
|
||||
justify-content: center;
|
||||
align-items: center;
|
||||
top: 6px;
|
||||
border-radius: 4px;
|
||||
}
|
||||
|
||||
:deep(.el-input__suffix-inner) {
|
||||
display: flex;
|
||||
align-items: center;
|
||||
justify-content: center;
|
||||
width: 100%;
|
||||
}
|
||||
|
||||
:deep(.el-icon-arrow-up:before) {
|
||||
content: "";
|
||||
display: inline-block;
|
||||
border-left: 6px solid transparent;
|
||||
border-right: 6px solid transparent;
|
||||
border-top: 9px solid #606266;
|
||||
position: relative;
|
||||
transform: rotate(0deg);
|
||||
transition: transform 0.3s;
|
||||
}
|
||||
}
|
||||
|
||||
:deep(.transparent-table) {
|
||||
background: white;
|
||||
flex: 1;
|
||||
width: 100%;
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
|
||||
.el-table__body-wrapper {
|
||||
flex: 1;
|
||||
overflow-y: auto;
|
||||
max-height: none !important;
|
||||
}
|
||||
|
||||
.el-table__header-wrapper {
|
||||
flex-shrink: 0;
|
||||
}
|
||||
|
||||
.el-table__header th {
|
||||
background: white !important;
|
||||
color: black;
|
||||
font-weight: 600;
|
||||
height: 40px;
|
||||
padding: 8px 0;
|
||||
font-size: 14px;
|
||||
border-bottom: 1px solid #e4e7ed;
|
||||
}
|
||||
|
||||
.el-table__body tr {
|
||||
background-color: white;
|
||||
|
||||
td {
|
||||
border-top: 1px solid rgba(0, 0, 0, 0.04);
|
||||
border-bottom: 1px solid rgba(0, 0, 0, 0.04);
|
||||
padding: 8px 0;
|
||||
height: 40px;
|
||||
color: #606266;
|
||||
font-size: 14px;
|
||||
}
|
||||
}
|
||||
|
||||
.el-table__row:hover>td {
|
||||
background-color: #f5f7fa !important;
|
||||
}
|
||||
|
||||
&::before {
|
||||
display: none;
|
||||
}
|
||||
}
|
||||
|
||||
:deep(.el-table .el-button--text) {
|
||||
color: #7079aa !important;
|
||||
}
|
||||
|
||||
:deep(.el-table .el-button--text:hover) {
|
||||
color: #5a64b5 !important;
|
||||
}
|
||||
|
||||
:deep(.el-checkbox__inner) {
|
||||
background-color: #eeeeee !important;
|
||||
border-color: #cccccc !important;
|
||||
}
|
||||
|
||||
:deep(.el-checkbox__inner:hover) {
|
||||
border-color: #cccccc !important;
|
||||
}
|
||||
|
||||
:deep(.el-checkbox__input.is-checked .el-checkbox__inner) {
|
||||
background-color: #5f70f3 !important;
|
||||
border-color: #5f70f3 !important;
|
||||
}
|
||||
|
||||
:deep(.el-loading-mask) {
|
||||
background-color: rgba(255, 255, 255, 0.6) !important;
|
||||
backdrop-filter: blur(2px);
|
||||
}
|
||||
|
||||
:deep(.el-loading-spinner .path) {
|
||||
stroke: #6b8cff;
|
||||
}
|
||||
|
||||
:deep(.el-table__empty-block) {
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
justify-content: center;
|
||||
align-items: center;
|
||||
min-height: 300px;
|
||||
}
|
||||
|
||||
.el-table {
|
||||
--table-max-height: calc(100vh - 40vh);
|
||||
max-height: var(--table-max-height);
|
||||
flex: 1;
|
||||
|
||||
.el-table__body-wrapper {
|
||||
max-height: calc(var(--table-max-height) - 40px);
|
||||
}
|
||||
}
|
||||
|
||||
@media (min-width: 1144px) {
|
||||
.table_bottom {
|
||||
display: flex;
|
||||
justify-content: space-between;
|
||||
align-items: center;
|
||||
margin-top: auto;
|
||||
}
|
||||
|
||||
:deep(.transparent-table) {
|
||||
.el-table__body tr {
|
||||
td {
|
||||
padding-top: 16px;
|
||||
padding-bottom: 16px;
|
||||
}
|
||||
|
||||
&+tr {
|
||||
margin-top: 10px;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
</style>
|
||||
File diff suppressed because it is too large
Load Diff
@@ -52,6 +52,9 @@
|
||||
<el-menu-item index="memory">
|
||||
<span class="menu-text">{{ $t("modelConfig.memory") }}</span>
|
||||
</el-menu-item>
|
||||
<el-menu-item index="rag">
|
||||
<span class="menu-text">{{ $t("modelConfig.rag") }}</span>
|
||||
</el-menu-item>
|
||||
</el-menu>
|
||||
|
||||
<!-- 右侧内容 -->
|
||||
|
||||
@@ -147,7 +147,8 @@ export default {
|
||||
{ value: "Intent", labelKey: 'providerManagement.modelType.Intent' },
|
||||
{ value: "Memory", labelKey: 'providerManagement.modelType.Memory' },
|
||||
{ value: "VAD", labelKey: 'providerManagement.modelType.VAD' },
|
||||
{ value: "Plugin", labelKey: 'providerManagement.modelType.Plugin' }
|
||||
{ value: "Plugin", labelKey: 'providerManagement.modelType.Plugin' },
|
||||
{ value: "RAG", labelKey: 'providerManagement.modelType.RAG' }
|
||||
],
|
||||
currentPage: 1,
|
||||
loading: false,
|
||||
@@ -369,7 +370,8 @@ export default {
|
||||
'LLM': 'danger',
|
||||
'Intent': 'info',
|
||||
'Memory': '',
|
||||
'VAD': 'primary'
|
||||
'VAD': 'primary',
|
||||
'RAG': 'warning'
|
||||
};
|
||||
return typeMap[type] || '';
|
||||
},
|
||||
|
||||
@@ -147,7 +147,15 @@ plugins:
|
||||
- ".wav"
|
||||
- ".p3"
|
||||
refresh_time: 300 # 刷新音乐列表的时间间隔，单位为秒
|
||||
|
||||
search_from_ragflow:
|
||||
# 知识库的描述信息，方便大语言模型知道什么时候调用
|
||||
description: "当用户问xxx时，调用本方法，使用知识库中的信息回答问题"
|
||||
# ragflow接口配置
|
||||
base_url: "http://192.168.0.8"
|
||||
# ragflow api访问令牌
|
||||
api_key: "ragflow-xxx"
|
||||
# ragflow知识库id
|
||||
dataset_ids: ["123456789"]
|
||||
# 声纹识别配置
|
||||
voiceprint:
|
||||
# 声纹接口地址
|
||||
@@ -239,6 +247,7 @@ Intent:
|
||||
functions:
|
||||
- change_role
|
||||
- get_weather
|
||||
# - search_from_ragflow
|
||||
# - get_news_from_chinanews
|
||||
- get_news_from_newsnow
|
||||
# play_music是服务器自带的音乐播放，hass_play_music是通过home assistant控制的独立外部程序音乐播放
|
||||
|
||||
@@ -5,7 +5,7 @@ from config.config_loader import load_config
|
||||
from config.settings import check_config_file
|
||||
from datetime import datetime
|
||||
|
||||
SERVER_VERSION = "0.8.6"
|
||||
SERVER_VERSION = "0.8.7"
|
||||
_logger_initialized = False
|
||||
|
||||
|
||||
|
||||
@@ -798,9 +798,9 @@ class ConnectionHandler:
|
||||
|
||||
if tools_call is not None and len(tools_call) > 0:
|
||||
tool_call_flag = True
|
||||
if tools_call[0].id is not None:
|
||||
if tools_call[0].id is not None and tools_call[0].id != "":
|
||||
function_id = tools_call[0].id
|
||||
if tools_call[0].function.name is not None:
|
||||
if tools_call[0].function.name is not None and tools_call[0].function.name != "":
|
||||
function_name = tools_call[0].function.name
|
||||
if tools_call[0].function.arguments is not None:
|
||||
function_arguments += tools_call[0].function.arguments
|
||||
|
||||
@@ -71,6 +71,19 @@ class ServerPluginExecutor(ToolExecutor):
|
||||
for func_name in all_required_functions:
|
||||
func_item = all_function_registry.get(func_name)
|
||||
if func_item:
|
||||
# 从函数注册中获取描述
|
||||
fun_description = (
|
||||
self.config.get("plugins", {})
|
||||
.get(func_name, {})
|
||||
.get("description", "")
|
||||
)
|
||||
if fun_description is not None and len(fun_description) > 0:
|
||||
if "function" in func_item.description and isinstance(
|
||||
func_item.description["function"], dict
|
||||
):
|
||||
func_item.description["function"][
|
||||
"description"
|
||||
] = fun_description
|
||||
tools[func_name] = ToolDefinition(
|
||||
name=func_name,
|
||||
description=func_item.description,
|
||||
|
||||
@@ -66,7 +66,9 @@ class PromptManager:
|
||||
def _load_base_template(self):
|
||||
"""加载基础提示词模板"""
|
||||
try:
|
||||
template_path = self.config.get("prompt_template", "agent-base-prompt.txt")
|
||||
template_path = self.config.get("prompt_template", None)
|
||||
if not template_path:
|
||||
template_path = "agent-base-prompt.txt"
|
||||
cache_key = f"prompt_template:{template_path}"
|
||||
|
||||
# 先从缓存获取
|
||||
@@ -117,8 +119,12 @@ class PromptManager:
|
||||
|
||||
def _get_current_time_info(self) -> tuple:
|
||||
"""获取当前时间信息"""
|
||||
from .current_time import get_current_date, get_current_weekday, get_current_lunar_date
|
||||
|
||||
from .current_time import (
|
||||
get_current_date,
|
||||
get_current_weekday,
|
||||
get_current_lunar_date,
|
||||
)
|
||||
|
||||
today_date = get_current_date()
|
||||
today_weekday = get_current_weekday()
|
||||
lunar_date = get_current_lunar_date() + "\n"
|
||||
@@ -192,9 +198,7 @@ class PromptManager:
|
||||
|
||||
try:
|
||||
# 获取最新的时间信息（不缓存）
|
||||
today_date, today_weekday, lunar_date = (
|
||||
self._get_current_time_info()
|
||||
)
|
||||
today_date, today_weekday, lunar_date = self._get_current_time_info()
|
||||
|
||||
# 获取缓存的上下文信息
|
||||
local_address = ""
|
||||
@@ -226,7 +230,8 @@ class PromptManager:
|
||||
emojiList=EMOJI_List,
|
||||
device_id=device_id,
|
||||
client_ip=client_ip,
|
||||
*args, **kwargs
|
||||
*args,
|
||||
**kwargs,
|
||||
)
|
||||
device_cache_key = f"device_prompt:{device_id}"
|
||||
self.cache_manager.set(
|
||||
|
||||
@@ -0,0 +1,98 @@
|
||||
import requests
|
||||
import sys
|
||||
from config.logger import setup_logging
|
||||
from plugins_func.register import register_function, ToolType, ActionResponse, Action
|
||||
|
||||
TAG = __name__
|
||||
logger = setup_logging()
|
||||
|
||||
# 定义基础的函数描述模板
|
||||
SEARCH_FROM_RAGFLOW_FUNCTION_DESC = {
|
||||
"type": "function",
|
||||
"function": {
|
||||
"name": "search_from_ragflow",
|
||||
"description": "从知识库中查询信息",
|
||||
"parameters": {
|
||||
"type": "object",
|
||||
"properties": {"question": {"type": "string", "description": "查询的问题"}},
|
||||
"required": ["question"],
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
@register_function(
|
||||
"search_from_ragflow", SEARCH_FROM_RAGFLOW_FUNCTION_DESC, ToolType.SYSTEM_CTL
|
||||
)
|
||||
def search_from_ragflow(conn, question=None):
|
||||
# 确保字符串参数正确处理编码
|
||||
if question and isinstance(question, str):
|
||||
# 确保问题参数是UTF-8编码的字符串
|
||||
pass
|
||||
else:
|
||||
question = str(question) if question is not None else ""
|
||||
|
||||
base_url = conn.config["plugins"]["search_from_ragflow"].get("base_url", "")
|
||||
api_key = conn.config["plugins"]["search_from_ragflow"].get("api_key", "")
|
||||
dataset_ids = conn.config["plugins"]["search_from_ragflow"].get("dataset_ids", [])
|
||||
|
||||
url = base_url + "/api/v1/retrieval"
|
||||
headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
|
||||
|
||||
# 确保payload中的字符串都是UTF-8编码
|
||||
payload = {"question": question, "dataset_ids": dataset_ids}
|
||||
|
||||
try:
|
||||
# 使用ensure_ascii=False确保JSON序列化时正确处理中文
|
||||
response = requests.post(
|
||||
url,
|
||||
json=payload,
|
||||
headers=headers,
|
||||
timeout=5,
|
||||
verify=False,
|
||||
)
|
||||
|
||||
# 显式设置响应的编码为utf-8
|
||||
response.encoding = "utf-8"
|
||||
|
||||
response.raise_for_status()
|
||||
|
||||
# 先获取文本内容，然后手动处理JSON解码
|
||||
response_text = response.text
|
||||
import json
|
||||
|
||||
result = json.loads(response_text)
|
||||
|
||||
if result.get("code") != 0:
|
||||
error_detail = response.get("error", {}).get("detail", "")
|
||||
# 安全地记录错误信息
|
||||
logger.bind(tag=TAG).error(
|
||||
"从RAGflow获取信息失败，原因：%s", str(error_detail)
|
||||
)
|
||||
return ActionResponse(Action.RESPONSE, None, "RAG接口返回异常")
|
||||
|
||||
chunks = result.get("data", {}).get("chunks", [])
|
||||
contents = []
|
||||
for chunk in chunks:
|
||||
content = chunk.get("content", "")
|
||||
if content:
|
||||
# 安全地处理内容字符串
|
||||
if isinstance(content, str):
|
||||
contents.append(content)
|
||||
elif isinstance(content, bytes):
|
||||
contents.append(content.decode("utf-8", errors="replace"))
|
||||
else:
|
||||
contents.append(str(content))
|
||||
|
||||
# 构建适合大模型的上下文内容（每段前加编号，段间两个换行）
|
||||
context_text = "\n\n".join(
|
||||
f"{i+1}. {c.strip()}" for i, c in enumerate(contents[:5])
|
||||
)
|
||||
if not context_text:
|
||||
context_text = "根据知识库查询结果，没有相关信息。"
|
||||
return ActionResponse(Action.REQLLM, context_text, None)
|
||||
|
||||
except Exception as e:
|
||||
# 使用安全的方式记录异常，避免编码问题
|
||||
logger.bind(tag=TAG).error("从RAGflow获取信息失败，原因：%s", str(e))
|
||||
return ActionResponse(Action.RESPONSE, None, "RAG接口返回异常")
|
||||
Reference in New Issue
Block a user