1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
package com.iailab.module.ai.api.chat;
 
import com.iailab.module.ai.controller.admin.chat.vo.conversation.AiChatConversationCreateEnergyReqVO;
import com.iailab.module.ai.dal.dataobject.chat.AiChatConversationDO;
import com.iailab.module.ai.service.chat.AiChatConversationService;
import jakarta.annotation.Resource;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.RestController;
 
import java.util.List;
 
import static com.iailab.framework.security.core.util.SecurityFrameworkUtils.getLoginUserId;
 
@RestController // 提供 RESTful API 接口，给 Feign 调用
@Validated
public class AiChatConversionApiImpl implements AiChatConversationApi {
 
    @Resource
    private AiChatConversationService chatConversationService;
 
    @Override
    public Long chatEnergyConversation(String modelName) {
        Long conversationId;
        List<AiChatConversationDO> list = chatConversationService.getChatConversationList(getLoginUserId(), modelName);
        if(list.size() == 0) {
            AiChatConversationCreateEnergyReqVO createReqVO = new AiChatConversationCreateEnergyReqVO();
            createReqVO.setModelName(modelName);
            conversationId = chatConversationService.createChatConversationEnergy(createReqVO);
        } else {
            conversationId = list.get(0).getId();
        }
        return conversationId;
    }
}