0606

admin

2025-06-09 4eb46966002c6ca24cbb8cc8b519a05610e81649

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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
package tech.aiflowy.ai.controller;
 
import org.springframework.web.bind.annotation.*;
import tech.aiflowy.ai.entity.AiBotMessage;
import tech.aiflowy.ai.service.AiBotMessageService;
import tech.aiflowy.common.domain.Result;
import tech.aiflowy.common.web.controller.BaseCurdController;
import tech.aiflowy.common.satoken.util.SaTokenUtil;
import com.agentsflex.core.util.Maps;
import com.agentsflex.core.util.StringUtil;
import com.mybatisflex.core.query.QueryWrapper;
import tech.aiflowy.common.web.jsonbody.JsonBody;
 
import java.math.BigInteger;
import java.util.ArrayList;
import java.util.List;
 
/**
 * Bot 消息记录表 控制层。
 *
 * @author michael
 * @since 2024-11-04
 */
@RestController
@RequestMapping("/api/v1/aiBotMessage")
public class AiBotMessageController extends BaseCurdController<AiBotMessageService, AiBotMessage> {
    private final AiBotMessageService aiBotMessageService;
 
    public AiBotMessageController(AiBotMessageService service, AiBotMessageService aiBotMessageService) {
        super(service);
        this.aiBotMessageService = aiBotMessageService;
    }
 
 
    @GetMapping("list")
    @Override
    public Result list(AiBotMessage entity, Boolean asTree, String sortKey, String sortType) {
 
        if (entity.getBotId() == null || StringUtil.noText(entity.getSessionId())) {
            return Result.fail();
        }
 
        QueryWrapper queryWrapper = QueryWrapper.create();
        queryWrapper.eq(AiBotMessage::getBotId, entity.getBotId());
        queryWrapper.eq(AiBotMessage::getAccountId, SaTokenUtil.getLoginAccount().getId());
        queryWrapper.eq(AiBotMessage::getSessionId, entity.getSessionId());
        queryWrapper.orderBy(AiBotMessage::getCreated, true);
 
        List<AiBotMessage> list = service.list(queryWrapper);
 
        if (list == null || list.isEmpty()) {
            return Result.fail();
        }
 
        List<Maps> maps = new ArrayList<>();
        for (AiBotMessage aiBotMessage : list) {
            maps.add(Maps.of("id", aiBotMessage.getId())
                    .set("content", aiBotMessage.getContent())
                    .set("role", aiBotMessage.getRole())
                    .set("created", aiBotMessage.getCreated().getTime())
                    .set("updateAt", aiBotMessage.getCreated().getTime())
            );
        }
        return Result.success(maps);
    }
 
 
    @Override
    protected Result onSaveOrUpdateBefore(AiBotMessage entity, boolean isSave) {
        entity.setAccountId(SaTokenUtil.getLoginAccount().getId());
        return super.onSaveOrUpdateBefore(entity, isSave);
    }
 
 
 
    @GetMapping("messageList")
    public Result messageList(@RequestParam(value = "botId") String botId,
                              @RequestParam(value = "sessionId") String sessionId,
                              @RequestParam(value = "isExternalMsg") int isExternalMsg
                              ) {
 
        return aiBotMessageService.messageList(botId, sessionId, isExternalMsg);
    }
 
    @PostMapping("removeMsg")
    public Result removeMsg(  @JsonBody(value = "botId", required = true) String botId,
                              @JsonBody(value = "sessionId", required = true) String sessionId,
                              @JsonBody(value = "isExternalMsg", required = true) int isExternalMsg
    ) {
 
        return aiBotMessageService.removeMsg(botId, sessionId, isExternalMsg);
    }
 
 
}