| | |
| | | Integer type = dto.getType(); |
| | | String reasoningContent = dto.getReasoningContent(); |
| | | String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid(); |
| | | String companyId = LoginUserUtil.getLoginUser().getCompanyId(); |
| | | |
| | | AiTalk aiTalk = this.getById(talkId); |
| | | DateTime date = DateUtil.date(); |
| | | if (StrUtil.isEmpty(talkId)){ |
| | | aiTalk = this.add(memberUuid,context, date); |
| | | aiTalk = this.add(memberUuid,companyId,context, date); |
| | | } |
| | | HashMap<String, String> stringStringHashMap = new HashMap<>(); |
| | | stringStringHashMap.put(LlmStrategyContextEnum.THINK.getName(),reasoningContent); |
| | | stringStringHashMap.put(LlmStrategyContextEnum.CONTENT.getName(),context); |
| | | aiTalkItemService.add(aiTalk.getId(), type, JSONUtil.toJsonStr(stringStringHashMap), memberUuid, date); |
| | | aiTalkItemService.add(aiTalk.getId(), companyId,type, JSONUtil.toJsonStr(stringStringHashMap), memberUuid, date); |
| | | |
| | | ApiTalkVo apiTalkVo = new ApiTalkVo(); |
| | | apiTalkVo.setTalkId(aiTalk.getId()); |
| | |
| | | } |
| | | |
| | | @Override |
| | | public AiTalk add(String memberUuid, String question, Date date) { |
| | | public AiTalk add(String memberUuid,String companyId, String question, Date date) { |
| | | AiTalk aiTalk = new AiTalk(); |
| | | aiTalk.setId(UUID.getSimpleUUIDString()); |
| | | aiTalk.setCompanyId(companyId); |
| | | aiTalk.setQuestion(question); |
| | | aiTalk.setCreatedTime(date); |
| | | aiTalk.setMemberId(memberUuid); |
| | |
| | | if (dto.getPrompt() != null){ |
| | | LlmStrategyDto llmStrategyDto = new LlmStrategyDto(); |
| | | llmStrategyDto.setRole(Role.SYSTEM.getValue()); |
| | | llmStrategyDto.setContent(AiPromptEnum.STREAM_NORMAL.getPrompt()); |
| | | llmStrategyDto.setContent(aiService.getSystemSetLTAiPrompt()); |
| | | llmStrategyDtoList.add(llmStrategyDto); |
| | | } |
| | | if (dto.getQuestion() != null){ |
| | | LlmStrategyDto llmStrategyDto = new LlmStrategyDto(); |
| | | llmStrategyDto.setRole(Role.USER.getValue()); |
| | | llmStrategyDto.setContent(dto.getQuestion()); |
| | | String format = StrUtil.format("请使用中文回答:{}", dto.getQuestion()); |
| | | llmStrategyDto.setContent(format); |
| | | llmStrategyDtoList.add(llmStrategyDto); |
| | | } |
| | | String modelName = LlmStrategyEnum.getName(aiService.getSystemSetAiType()); |