feat: chat authentication

Author: shaohuzhang1

apps/application/flow/i_step_node.py

@@ -19,7 +19,7 @@
 
 from application.flow.common import Answer, NodeChunk
 from application.models import ChatRecord
-from application.models import ApplicationPublicAccessClient
+from application.models import ApplicationChatClientStats
 from common.constants.authentication_type import AuthenticationType
 from common.field.common import InstanceField
 
@@ -89,7 +89,7 @@ def handler(self, chat_id,
         chat_cache.set(chat_id,
                        self.chat_info, timeout=60 * 30)
         if self.client_type == AuthenticationType.APPLICATION_ACCESS_TOKEN.value:
-            application_public_access_client = (QuerySet(ApplicationPublicAccessClient)
+            application_public_access_client = (QuerySet(ApplicationChatClientStats)
                                                 .filter(client_id=self.client_id,
                                                         application_id=self.chat_info.application.id).first())
             if application_public_access_client is not None:

apps/application/flow/step_node/search_dataset_node/impl/base_search_dataset_node.py

@@ -9,19 +9,17 @@
 import os
 from typing import List, Dict
 
-from django.db.models import QuerySet
 from django.db import connection
+from django.db.models import QuerySet
+
 from application.flow.i_step_node import NodeResult
 from application.flow.step_node.search_dataset_node.i_search_dataset_node import ISearchDatasetStepNode
 from common.config.embedding_config import VectorStore
 from common.db.search import native_search
 from common.utils.common import get_file_content
-from knowledge.models import Document, Paragraph, Knowledge
-
-from models_provider.tools import get_model_instance_by_model_user_id
+from knowledge.models import Document, Paragraph, Knowledge, SearchMode
 from maxkb.conf import PROJECT_DIR
-
-SearchMode = None
+from models_provider.tools import get_model_instance_by_model_user_id
 
 
 def get_embedding_id(dataset_id_list):

apps/application/migrations/0002_chat_chatrecord_workflowversion_and_more.py

@@ -1,16 +1,16 @@
 # Generated by Django 5.2 on 2025-06-04 11:57
 
-import application.models.application_chat
-import common.encoder.encoder
+import uuid
+
 import django.contrib.postgres.fields
 import django.db.models.deletion
-import uuid
 import uuid_utils.compat
 from django.db import migrations, models
 
+import common.encoder.encoder
 
-class Migration(migrations.Migration):
 
+class Migration(migrations.Migration):
     dependencies = [
         ('application', '0001_initial'),
     ]
@@ -21,12 +21,13 @@ class Migration(migrations.Migration):
             fields=[
                 ('create_time', models.DateTimeField(auto_now_add=True, verbose_name='创建时间')),
                 ('update_time', models.DateTimeField(auto_now=True, verbose_name='修改时间')),
-                ('id', models.UUIDField(default=uuid.UUID('01973acd-fe4c-7fd1-94a8-f7cd668de562'), editable=False, primary_key=True, serialize=False, verbose_name='主键id')),
+                ('id', models.UUIDField(default=uuid.UUID('01973acd-fe4c-7fd1-94a8-f7cd668de562'), editable=False,
+                                        primary_key=True, serialize=False, verbose_name='主键id')),
                 ('abstract', models.CharField(max_length=1024, verbose_name='摘要')),
-                ('asker', models.JSONField(default=application.models.application_chat.default_asker, encoder=common.encoder.encoder.SystemEncoder, verbose_name='访问者')),
                 ('client_id', models.UUIDField(default=None, null=True, verbose_name='客户端id')),
                 ('is_deleted', models.BooleanField(default=False, verbose_name='')),
-                ('application', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='application.application')),
+                ('application',
+                 models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='application.application')),
             ],
             options={
                 'db_table': 'application_chat',
@@ -37,16 +38,25 @@ class Migration(migrations.Migration):
             fields=[
                 ('create_time', models.DateTimeField(auto_now_add=True, verbose_name='创建时间')),
                 ('update_time', models.DateTimeField(auto_now=True, verbose_name='修改时间')),
-                ('id', models.UUIDField(default=uuid_utils.compat.uuid1, editable=False, primary_key=True, serialize=False, verbose_name='主键id')),
-                ('vote_status', models.CharField(choices=[('-1', '未投票'), ('0', '赞同'), ('1', '反对')], default='-1', max_length=10, verbose_name='投票')),
+                ('id',
+                 models.UUIDField(default=uuid_utils.compat.uuid1, editable=False, primary_key=True, serialize=False,
+                                  verbose_name='主键id')),
+                ('vote_status',
+                 models.CharField(choices=[('-1', '未投票'), ('0', '赞同'), ('1', '反对')], default='-1', max_length=10,
+                                  verbose_name='投票')),
                 ('problem_text', models.CharField(max_length=10240, verbose_name='问题')),
                 ('answer_text', models.CharField(max_length=40960, verbose_name='答案')),
-                ('answer_text_list', django.contrib.postgres.fields.ArrayField(base_field=models.JSONField(), default=list, size=None, verbose_name='改进标注列表')),
+                ('answer_text_list',
+                 django.contrib.postgres.fields.ArrayField(base_field=models.JSONField(), default=list, size=None,
+                                                           verbose_name='改进标注列表')),
                 ('message_tokens', models.IntegerField(default=0, verbose_name='请求token数量')),
                 ('answer_tokens', models.IntegerField(default=0, verbose_name='响应token数量')),
                 ('const', models.IntegerField(default=0, verbose_name='总费用')),
-                ('details', models.JSONField(default=dict, encoder=common.encoder.encoder.SystemEncoder, verbose_name='对话详情')),
-                ('improve_paragraph_id_list', django.contrib.postgres.fields.ArrayField(base_field=models.UUIDField(blank=True), default=list, size=None, verbose_name='改进标注列表')),
+                ('details',
+                 models.JSONField(default=dict, encoder=common.encoder.encoder.SystemEncoder, verbose_name='对话详情')),
+                ('improve_paragraph_id_list',
+                 django.contrib.postgres.fields.ArrayField(base_field=models.UUIDField(blank=True), default=list,
+                                                           size=None, verbose_name='改进标注列表')),
                 ('run_time', models.FloatField(default=0, verbose_name='运行时长')),
                 ('index', models.IntegerField(verbose_name='对话下标')),
                 ('chat', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='application.chat')),
@@ -60,13 +70,17 @@ class Migration(migrations.Migration):
             fields=[
                 ('create_time', models.DateTimeField(auto_now_add=True, verbose_name='创建时间')),
                 ('update_time', models.DateTimeField(auto_now=True, verbose_name='修改时间')),
-                ('id', models.UUIDField(default=uuid_utils.compat.uuid1, editable=False, primary_key=True, serialize=False, verbose_name='主键id')),
-                ('workspace_id', models.CharField(db_index=True, default='default', max_length=64, verbose_name='工作空间id')),
+                ('id',
+                 models.UUIDField(default=uuid_utils.compat.uuid1, editable=False, primary_key=True, serialize=False,
+                                  verbose_name='主键id')),
+                ('workspace_id',
+                 models.CharField(db_index=True, default='default', max_length=64, verbose_name='工作空间id')),
                 ('name', models.CharField(default='', max_length=128, verbose_name='版本名称')),
                 ('publish_user_id', models.UUIDField(default=None, null=True, verbose_name='发布者id')),
                 ('publish_user_name', models.CharField(default='', max_length=128, verbose_name='发布者名称')),
                 ('work_flow', models.JSONField(default=dict, verbose_name='工作流数据')),
-                ('application', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='application.application')),
+                ('application',
+                 models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='application.application')),
             ],
             options={
                 'db_table': 'application_work_flow_version',
@@ -77,16 +91,20 @@ class Migration(migrations.Migration):
             fields=[
                 ('create_time', models.DateTimeField(auto_now_add=True, verbose_name='创建时间')),
                 ('update_time', models.DateTimeField(auto_now=True, verbose_name='修改时间')),
-                ('id', models.UUIDField(default=uuid.uuid1, editable=False, primary_key=True, serialize=False, verbose_name='主键id')),
+                ('id', models.UUIDField(default=uuid.uuid1, editable=False, primary_key=True, serialize=False,
+                                        verbose_name='主键id')),
                 ('client_id', models.UUIDField(default=uuid.uuid1, verbose_name='公共访问链接客户端id')),
                 ('client_type', models.CharField(max_length=64, verbose_name='客户端类型')),
                 ('access_num', models.IntegerField(default=0, verbose_name='访问总次数次数')),
                 ('intraday_access_num', models.IntegerField(default=0, verbose_name='当日访问次数')),
-                ('application', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='application.application', verbose_name='应用id')),
+                ('application',
+                 models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='application.application',
+                                   verbose_name='应用id')),
             ],
             options={
                 'db_table': 'application_public_access_client',
-                'indexes': [models.Index(fields=['application_id', 'client_id'], name='application_applica_8aaf45_idx')],
+                'indexes': [
+                    models.Index(fields=['application_id', 'client_id'], name='application_applica_8aaf45_idx')],
             },
         ),
     ]

apps/application/models/application_api_key.py

@@ -24,16 +24,3 @@ class Meta:
         db_table = "application_api_key"
 
 
-class ApplicationPublicAccessClient(AppModelMixin):
-    id = models.UUIDField(primary_key=True, max_length=128, default=uuid.uuid1, editable=False, verbose_name="主键id")
-    client_id = models.UUIDField(max_length=128, default=uuid.uuid1, verbose_name="公共访问链接客户端id")
-    client_type = models.CharField(max_length=64, verbose_name="客户端类型")
-    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_public_access_client"
-        indexes = [
-            models.Index(fields=['application_id', 'client_id']),
-        ]

apps/application/models/application_chat.py

@@ -17,17 +17,20 @@
 from common.mixins.app_model_mixin import AppModelMixin
 
 
-def default_asker():
-    return {'user_name': '游客'}
+class ClientType(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"
 
 
 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="摘要")
-    asker = models.JSONField(verbose_name="访问者", default=default_asker, encoder=SystemEncoder)
     client_id = models.UUIDField(verbose_name="客户端id", default=None, null=True)
-    is_deleted = models.BooleanField(verbose_name="", default=False)
+    client_type = models.CharField(max_length=64, verbose_name="客户端类型", choices=ClientType.choices)
+    is_deleted = models.BooleanField(verbose_name="逻辑删除", default=False)
 
     class Meta:
         db_table = "application_chat"
@@ -80,3 +83,18 @@ def get_node_details_runtime_node_id(self, runtime_node_id):
 
     class Meta:
         db_table = "application_chat_record"
+
+
+class ApplicationChatClientStats(AppModelMixin):
+    id = models.UUIDField(primary_key=True, max_length=128, default=uuid.uuid7, editable=False, verbose_name="主键id")
+    client_id = models.UUIDField(max_length=128, default=uuid.uuid7, verbose_name="公共访问链接客户端id")
+    client_type = models.CharField(max_length=64, verbose_name="客户端类型", choices=ClientType.choices)
+    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_client_stats"
+        indexes = [
+            models.Index(fields=['application_id', 'client_id']),
+        ]

apps/application/views/application_api_key.py

@@ -20,10 +20,10 @@ class ApplicationKey(APIView):
         parameters=ApplicationKeyCreateAPI.get_parameters(),
         tags=[_('Application Api Key')]  # type: ignore
     )
-    def post(self,request: Request, application_id: str, workspace_id: str):
+    def post(self, request: Request, workspace_id: str, application_id: str):
         return result.success(ApplicationKeySerializer(
-                data={'application_id': application_id, 'user_id': request.user.id,
-                      'workspace_id':workspace_id}).generate())
+            data={'application_id': application_id, 'user_id': request.user.id,
+                  'workspace_id': workspace_id}).generate())
 
     @extend_schema(
         methods=['GET'],
@@ -33,11 +33,10 @@ def post(self,request: Request, application_id: str, workspace_id: str):
         parameters=ApplicationKeyCreateAPI.get_parameters(),
         tags=[_('Application Api Key')]  # type: ignore
     )
-    def get(self,request: Request, application_id: str, workspace_id: str):
-        return result,success(ApplicationKeySerializer(
-            data={'application_id':application_id, 'user_id':request.user.id,
-                  'workspace_id':workspace_id}).list())
-
+    def get(self, request: Request, workspace_id: str, application_id: str, ):
+        return result, success(ApplicationKeySerializer(
+            data={'application_id': application_id, 'user_id': request.user.id,
+                  'workspace_id': workspace_id}).list())
 
     class Operate(APIView):
         authentication_classes = [TokenAuth]
@@ -50,9 +49,8 @@ class Operate(APIView):
             parameters=ApplicationKeyCreateAPI.get_parameters(),
             tags=[_('Application Api Key')]  # type: ignore
         )
-        def put(self, request: Request, application_id: str, workspace_id: str):
+        def put(self, request: Request, workspace_id: str, application_id: str):
             return result.success(ApplicationKeySerializer.Operate(
 
             )
-                                  )
-
+            )

apps/application/views/application_version.py

@@ -12,7 +12,7 @@
 from rest_framework.views import APIView
 
 from application.api.application_version import ApplicationVersionListAPI, ApplicationVersionPageAPI, \
-    ApplicationVersionAPI, ApplicationVersionOperateAPI
+    ApplicationVersionOperateAPI
 from application.serializers.application_version import ApplicationVersionSerializer
 from common import result
 from common.auth import TokenAuth
@@ -88,6 +88,7 @@ def get(self, request: Request, workspace_id: str, application_id: str, work_flo
             responses=ApplicationVersionOperateAPI.get_response(),
             tags=[_('Application/Version')]  # type: ignore
         )
+        @has_permissions(PermissionConstants.APPLICATION_EDIT.get_workspace_application_permission())
         def put(self, request: Request, workspace_id: str, application_id: str, work_flow_version_id: str):
             return result.success(
                 ApplicationVersionSerializer.Operate(

apps/chat/api/chat_authentication_api.py

@@ -0,0 +1,24 @@
+# coding=utf-8
+"""
+    @project: MaxKB
+    @Author：虎虎
+    @file： chat_authentication_api.py
+    @date：2025/6/6 19:59
+    @desc:
+"""
+from chat.serializers.chat_authentication import AuthenticationSerializer
+from common.mixins.api_mixin import APIMixin
+
+
+class ChatAuthenticationAPI(APIMixin):
+    @staticmethod
+    def get_request():
+        return AuthenticationSerializer()
+
+    @staticmethod
+    def get_parameters():
+        pass
+
+    @staticmethod
+    def get_response():
+        pass