| | |
| | | } |
| | | |
| | | |
| | | |
| | | @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() |