From ddf31ac4e871f442dfcc7ce1543f18159bc52077 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 02 Sep 2025 10:29:03 +0800
Subject: [PATCH] refactor(ai): 重构 AI服务调用方法
---
src/main/java/cc/mrbird/febs/ai/controller/TestController.java | 8 +++++---
1 files changed, 5 insertions(+), 3 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/controller/TestController.java b/src/main/java/cc/mrbird/febs/ai/controller/TestController.java
index 16f0d63..70d4a61 100644
--- a/src/main/java/cc/mrbird/febs/ai/controller/TestController.java
+++ b/src/main/java/cc/mrbird/febs/ai/controller/TestController.java
@@ -4,6 +4,7 @@
import cc.mrbird.febs.ai.mapper.AiMemberMapper;
import cc.mrbird.febs.ai.req.talk.AiTalkAnswerStream;
import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkStreamVo;
+import cc.mrbird.febs.ai.service.AiService;
import cc.mrbird.febs.ai.strategy.enumerates.LlmStrategyEnum;
import cc.mrbird.febs.ai.strategy.LlmStrategyFactory;
import cc.mrbird.febs.ai.strategy.param.LlmStrategyDto;
@@ -55,6 +56,7 @@
private final MallMemberMapper mallMemberMapper;
private final AiMemberMapper aiMemberMapper;
+ private final AiService aiService;
private final RedisUtils redisUtils;
private final LlmStrategyFactory llmStrategyFactory;
@ApiOperation(value = "登录测试", notes = "登录测试")
@@ -164,9 +166,9 @@
llmStrategyDto.setContent(dto.getQuestion());
llmStrategyDtoList.add(llmStrategyDto);
}
- String modelName = LlmStrategyEnum.getName(dto.getType());
+ String modelName = LlmStrategyEnum.getName(aiService.getSystemSetAiType());
- return llmStrategyFactory.getCalculationStrategyMap().get(modelName).llmInvokeStreaming(llmStrategyDtoList);
+ return llmStrategyFactory.getCalculationStrategyMap().get(modelName).llmInvokeStreamingWithThink(llmStrategyDtoList);
}
@@ -193,7 +195,7 @@
llmStrategyDto.setContent(dto.getQuestion());
llmStrategyDtoList.add(llmStrategyDto);
}
- String modelName = LlmStrategyEnum.getName(dto.getType());
+ String modelName = LlmStrategyEnum.getName(aiService.getSystemSetAiType());
return llmStrategyFactory.getCalculationStrategyMap().get(modelName).llmInvokeNonStreaming(llmStrategyDtoList);
}
--
Gitblit v1.9.1