MaxKB/apps/application/models/application_chat.py at 8815a451837bcf2531d10bc933c3e030ba8eb668 · 1Panel-dev/MaxKB · GitHub

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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
# coding=utf-8
"""
    @project: MaxKB
    @Author：虎虎
    @file： application_chat_log.py
    @date：2025/5/29 17:12
    @desc:
"""
import uuid_utils.compat as uuid
from django.contrib.postgres.fields import ArrayField
from django.db import models
from django.utils.translation import gettext as _
from langchain_core.messages import HumanMessage, AIMessage

from application.models import Application
from common.encoder.encoder import SystemEncoder
from common.mixins.app_model_mixin import AppModelMixin
from users.models import User


class ChatUserType(models.TextChoices):
    ANONYMOUS_USER = "ANONYMOUS_USER", '匿名用户'
    CHAT_USER = "CHAT_USER", "对话用户"
    SYSTEM_API_KEY = "SYSTEM_API_KEY", "系统API_KEY"
    APPLICATION_API_KEY = "APPLICATION_API_KEY", "应用API_KEY"
    PLATFORM_USER = "PLATFORM_USER", "平台用户"


def default_asker():
    return {'username': '游客'}


class Chat(AppModelMixin):
    id = models.UUIDField(primary_key=True, max_length=128, default=uuid.uuid7, editable=False, verbose_name="主键id")
    application = models.ForeignKey(Application, on_delete=models.CASCADE)
    abstract = models.CharField(max_length=1024, verbose_name="摘要")
    chat_user_id = models.CharField(verbose_name="对话用户id", default=None, null=True)
    chat_user_type = models.CharField(max_length=64, verbose_name="客户端类型", choices=ChatUserType.choices,
                                      default=ChatUserType.ANONYMOUS_USER)
    is_deleted = models.BooleanField(verbose_name="逻辑删除", default=False)
    asker = models.JSONField(verbose_name="访问者", default=default_asker, encoder=SystemEncoder)
    meta = models.JSONField(verbose_name="元数据", default=dict)
    star_num = models.IntegerField(verbose_name="点赞数量", default=0)
    trample_num = models.IntegerField(verbose_name="点踩数量", default=0)
    chat_record_count = models.IntegerField(verbose_name="对话次数", default=0)
    mark_sum = models.IntegerField(verbose_name="标记数量", default=0)
    source = models.JSONField(verbose_name="来源", default=dict)
    ip_address = models.CharField(max_length=128, verbose_name="ip地址", default='')

    class Meta:
        db_table = "application_chat"


class VoteChoices(models.TextChoices):
    """订单类型"""
    UN_VOTE = "-1", '未投票'
    STAR = "0", '赞同'
    TRAMPLE = "1", '反对'


class VoteReasonChoices(models.TextChoices):
    ACCURATE = 'accurate', '内容准确'
    COMPLETE = 'complete', '内容完善'
    INACCURATE = 'inaccurate', '内容不准确'
    INCOMPLETE = 'incomplete', '内容不完善'
    OTHER = 'other', '其他'

class ShareLinkType(models.TextChoices):
    PUBLIC = "PUBLIC", 'public'
    PRIVATE = "PRIVATE", 'private'

class ChatSourceChoices(models.TextChoices):
    ONLINE = "ONLINE", "线上使用"
    API_CALL = "API_CALL", "API调用"
    ENTERPRISE_WECHAT = "ENTERPRISE_WECHAT", "企业微信"
    WECHAT_PUBLIC_ACCOUNT = "WECHAT_PUBLIC_ACCOUNT", "微信公众号"
    LARK = "LARK", "飞书"
    DINGTALK = "DINGTALK", "钉钉"
    ENTERPRISE_WECHAT_ROBOT = "ENTERPRISE_WECHAT_ROBOT", "企业微信机器人"
    TRIGGER = "TRIGGER", "触发器"
    SLACK = "SLACK", "Slack"


class ChatRecord(AppModelMixin):
    """
    对话日志 详情
    """
    id = models.UUIDField(primary_key=True, max_length=128, default=uuid.uuid7, editable=False, verbose_name="主键id")
    chat = models.ForeignKey(Chat, on_delete=models.CASCADE)
    vote_status = models.CharField(verbose_name='投票', max_length=10, choices=VoteChoices.choices,
                                   default=VoteChoices.UN_VOTE)
    vote_reason = models.CharField(verbose_name='投票原因', max_length=50, choices=VoteReasonChoices.choices, null=True,
                                   blank=True)
    vote_other_content = models.CharField(verbose_name='其他原因', max_length=1024, default='')
    problem_text = models.CharField(max_length=10240, verbose_name="问题")
    answer_text = models.CharField(max_length=40960, verbose_name="答案")
    answer_text_list = ArrayField(verbose_name="改进标注列表",
                                  base_field=models.JSONField()
                                  , default=list)
    message_tokens = models.IntegerField(verbose_name="请求token数量", default=0)
    answer_tokens = models.IntegerField(verbose_name="响应token数量", default=0)
    const = models.IntegerField(verbose_name="总费用", default=0)
    details = models.JSONField(verbose_name="对话详情", default=dict, encoder=SystemEncoder)
    improve_paragraph_id_list = ArrayField(verbose_name="改进标注列表",
                                           base_field=models.UUIDField(max_length=128, blank=True)
                                           , default=list)
    run_time = models.FloatField(verbose_name="运行时长", default=0)
    index = models.IntegerField(verbose_name="对话下标")
    source = models.JSONField(verbose_name="来源", default=dict)
    ip_address = models.CharField(max_length=128, verbose_name="ip地址", default='')

    def get_human_message(self):
        if 'problem_padding' in self.details:
            return HumanMessage(content=self.details.get('problem_padding').get('padding_problem_text'))
        return HumanMessage(content=self.problem_text)

    def get_ai_message(self):
        answer_text = self.answer_text
        if answer_text is None or len(str(answer_text).strip()) == 0:
            answer_text = _(
                'Sorry, no relevant content was found. Please re-describe your problem or provide more information. ')
        return AIMessage(content=answer_text)

    def get_node_details_runtime_node_id(self, runtime_node_id):
        return self.details.get(runtime_node_id, None)

    class Meta:
        db_table = "application_chat_record"


class ApplicationChatUserStats(AppModelMixin):
    id = models.UUIDField(primary_key=True, max_length=128, default=uuid.uuid7, editable=False, verbose_name="主键id")
    chat_user_id = models.UUIDField(max_length=128, default=uuid.uuid7, verbose_name="对话用户id")
    chat_user_type = models.CharField(max_length=64, verbose_name="对话用户类型", choices=ChatUserType.choices,
                                      default=ChatUserType.ANONYMOUS_USER)
    application = models.ForeignKey(Application, on_delete=models.CASCADE, verbose_name="应用id")
    access_num = models.IntegerField(default=0, verbose_name="访问总次数次数")
    intraday_access_num = models.IntegerField(default=0, verbose_name="当日访问次数")

    class Meta:
        db_table = "application_chat_user_stats"
        indexes = [
            models.Index(fields=['application_id', 'chat_user_id']),
        ]

class ChatShareLink(AppModelMixin):
    id = models.UUIDField(primary_key=True, max_length=128, default=uuid.uuid7, editable=False, verbose_name="主键id")
    chat = models.ForeignKey(Chat, on_delete=models.CASCADE)
    application = models.ForeignKey(Application,on_delete=models.CASCADE)
    share_type = models.CharField(max_length=20, choices=ShareLinkType.choices, default=ShareLinkType.PUBLIC)
    user = models.ForeignKey(User, on_delete=models.SET_NULL, db_constraint=False, blank=True, null=True)
    chat_record_ids = ArrayField(base_field=models.UUIDField(max_length=128))

    class Meta:
        db_table = "application_chat_share_link"