From 3749555542556cd63f21c29cc6cb20bfbdef9e4c Mon Sep 17 00:00:00 2001 From: Administrator <15274802129@163.com> Date: Mon, 01 Sep 2025 14:13:43 +0800 Subject: [PATCH] refactor(ai): 重构阿里云大模型调用逻辑 --- src/main/java/cc/mrbird/febs/ai/controller/TestController.java | 25 +++++++++++++++++++++++++ 1 files changed, 25 insertions(+), 0 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 fd74b56..b106fb4 100644 --- a/src/main/java/cc/mrbird/febs/ai/controller/TestController.java +++ b/src/main/java/cc/mrbird/febs/ai/controller/TestController.java @@ -170,6 +170,31 @@ } + + @ApiOperation("提问AI(非流式响应)V4") + @ApiResponses({ + @ApiResponse(code = 200, message = "非流式响应", response = ApiMemberTalkStreamVo.class), + }) + @PostMapping("/answerStreamV4") + public FebsResponse answerStreamV4(@RequestBody @Validated AiTalkAnswerStream dto) { + if (StrUtil.isEmpty(dto.getQuestion())){ + return new FebsResponse().fail().message("请输入问题"); + } + LlmStrategyDto llmStrategyDto = new LlmStrategyDto(); + Message systemMsg = Message.builder() + .role(Role.SYSTEM.getValue()) + .content(dto.getPrompt()) + .build(); + Message userMsg = Message.builder() + .role(Role.USER.getValue()) + .content(dto.getQuestion()) + .build(); + List<Message> messages = Arrays.asList(systemMsg, userMsg); + llmStrategyDto.setMessages(messages); + return llmStrategyFactory.getCalculationStrategyMap().get(LlmStrategyEnum.ALI.getName()).llmInvokeNonStreaming(llmStrategyDto); + } + + public static Flowable<GenerationResult> callWithMessageStream(String question,String prompt) throws NoApiKeyException, InputRequiredException { Generation gen = new Generation(); Message systemMsg = Message.builder() -- Gitblit v1.9.1