Administrator
2025-09-01 3749555542556cd63f21c29cc6cb20bfbdef9e4c
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()