feat: i18n

Author: shaohuzhang1

apps/application/migrations/0024_applicationaccesstoken_language.py

@@ -0,0 +1,19 @@
+# Generated by Django 4.2.15 on 2025-01-20 03:20
+
+from django.db import migrations, models
+import smartdoc.conf
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('application', '0023_application_stt_autosend'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='applicationaccesstoken',
+            name='language',
+            field=models.CharField(default=smartdoc.conf.Config.get_language_code, max_length=10, verbose_name='语言'),
+        ),
+    ]

apps/application/models/api_key_model.py

@@ -13,6 +13,7 @@
 
 from application.models import Application
 from common.mixins.app_model_mixin import AppModelMixin
+from smartdoc.const import CONFIG
 from users.models import User
 
 
@@ -45,6 +46,8 @@ class ApplicationAccessToken(AppModelMixin):
                             , default=list)
     show_source = models.BooleanField(default=False, verbose_name="是否显示知识来源")
 
+    language = models.CharField(max_length=10, verbose_name="语言", default=CONFIG.get_language_code)
+
     class Meta:
         db_table = "application_access_token"
 

apps/application/serializers/application_serializers.py

@@ -336,6 +336,8 @@ class AccessTokenEditSerializer(serializers.Serializer):
             show_source = serializers.BooleanField(required=False,
                                                    error_messages=ErrMessage.boolean(
                                                        _("Whether to display knowledge sources")))
+            language = serializers.CharField(required=False, allow_blank=True, allow_null=True,
+                                             error_messages=ErrMessage.char(_("language")))
 
         def edit(self, instance: Dict, with_valid=True):
             if with_valid:
@@ -358,6 +360,8 @@ def edit(self, instance: Dict, with_valid=True):
                 application_access_token.white_list = instance.get('white_list')
             if 'show_source' in instance and instance.get('show_source') is not None:
                 application_access_token.show_source = instance.get('show_source')
+            if 'language' in instance and instance.get('language') is not None:
+                application_access_token.language = instance.get('language')
             application_access_token.save()
             application_setting_model = DBModelManage.get_model('application_setting')
             xpack_cache = DBModelManage.get_model('xpack_cache')
@@ -980,6 +984,7 @@ def profile(self, with_valid=True):
                  'file_upload_setting': application.file_upload_setting,
                  'work_flow': application.work_flow,
                  'show_source': application_access_token.show_source,
+                 'language': application_access_token.language,
                  **application_setting_dict})
 
         @transaction.atomic

apps/application/swagger_api/application_api.py

@@ -165,6 +165,9 @@ def get_request_body_api():
                     'show_source': openapi.Schema(type=openapi.TYPE_BOOLEAN,
                                                   title=_("Whether to display knowledge sources"),
                                                   description=_("Whether to display knowledge sources")),
+                    'language': openapi.Schema(type=openapi.TYPE_STRING,
+                                               title=_("language"),
+                                               description=_("language"))
                 }
             )
 
@@ -282,24 +285,24 @@ def get_request_body_api():
                 properties={
                     'prompt': openapi.Schema(type=openapi.TYPE_STRING, title=_("Prompt word"),
                                              description=_("Prompt word"),
-                                             default=_("""
-Known information:
-{data}
-Answer requirements:
-- If you don't know the answer or don't get the answer, please answer "No relevant information found in the knowledge base, it is recommended to consult relevant technical support or refer to official documents for operation".
-- Avoid mentioning that you got the knowledge from <data></data>.
-- Please keep the answer consistent with the description in <data></data>.
-- Please use markdown syntax to optimize the format of the answer.
-- Please return the image link, link address and script language in <data></data> completely.
-- Please answer in the same language as the question.
-Question:
-{question}
-                                                    """)),
+                                             default=_(("Known information:\n"
+                                                        "{data}\n"
+                                                        "Answer requirements:\n"
+                                                        "- If you don't know the answer or don't get the answer, please answer \"No relevant information found in the knowledge base, it is recommended to consult relevant technical support or refer to official documents for operation\".\n"
+                                                        "- Avoid mentioning that you got the knowledge from <data></data>.\n"
+                                                        "- Please keep the answer consistent with the description in <data></data>.\n"
+                                                        "- Please use markdown syntax to optimize the format of the answer.\n"
+                                                        "- Please return the image link, link address and script language in <data></data> completely.\n"
+                                                        "- Please answer in the same language as the question.\n"
+                                                        "Question:\n"
+                                                        "{question}"))),
 
                     'system': openapi.Schema(type=openapi.TYPE_STRING, title=_("System prompt words (role)"),
                                              description=_("System prompt words (role)")),
-                    'no_references_prompt': openapi.Schema(type=openapi.TYPE_STRING, title=_("No citation segmentation prompt"),
-                                                           default="{question}", description=_("No citation segmentation prompt"))
+                    'no_references_prompt': openapi.Schema(type=openapi.TYPE_STRING,
+                                                           title=_("No citation segmentation prompt"),
+                                                           default="{question}",
+                                                           description=_("No citation segmentation prompt"))
 
                 }
             )
@@ -323,22 +326,29 @@ def get_request_body_api():
                 required=['name', 'desc', 'model_id', 'dialogue_number', 'dataset_setting', 'model_setting',
                           'problem_optimization', 'stt_model_enable', 'stt_model_enable', 'tts_type'],
                 properties={
-                    'name': openapi.Schema(type=openapi.TYPE_STRING, title=_("Application Name"), description=_("Application Name")),
-                    'desc': openapi.Schema(type=openapi.TYPE_STRING, title=_("Application Description"), description=_("Application Description")),
-                    'model_id': openapi.Schema(type=openapi.TYPE_STRING, title=_("Model id"), description=_("Model id")),
-                    "dialogue_number": openapi.Schema(type=openapi.TYPE_NUMBER, title=_("Number of multi-round conversations"),
+                    'name': openapi.Schema(type=openapi.TYPE_STRING, title=_("Application Name"),
+                                           description=_("Application Name")),
+                    'desc': openapi.Schema(type=openapi.TYPE_STRING, title=_("Application Description"),
+                                           description=_("Application Description")),
+                    'model_id': openapi.Schema(type=openapi.TYPE_STRING, title=_("Model id"),
+                                               description=_("Model id")),
+                    "dialogue_number": openapi.Schema(type=openapi.TYPE_NUMBER,
+                                                      title=_("Number of multi-round conversations"),
                                                       description=_("Number of multi-round conversations")),
-                    'prologue': openapi.Schema(type=openapi.TYPE_STRING, title=_("Opening remarks"), description=_("Opening remarks")),
+                    'prologue': openapi.Schema(type=openapi.TYPE_STRING, title=_("Opening remarks"),
+                                               description=_("Opening remarks")),
                     'dataset_id_list': openapi.Schema(type=openapi.TYPE_ARRAY,
                                                       items=openapi.Schema(type=openapi.TYPE_STRING),
-                                                      title=_("List of associated knowledge base IDs"), description=_("List of associated knowledge base IDs")),
+                                                      title=_("List of associated knowledge base IDs"),
+                                                      description=_("List of associated knowledge base IDs")),
                     'dataset_setting': ApplicationApi.DatasetSetting.get_request_body_api(),
                     'model_setting': ApplicationApi.ModelSetting.get_request_body_api(),
                     'problem_optimization': openapi.Schema(type=openapi.TYPE_BOOLEAN, title=_("Problem Optimization"),
                                                            description=_("Problem Optimization"), default=True),
                     'type': openapi.Schema(type=openapi.TYPE_STRING, title=_("Application Type"),
                                            description=_("Application Type   SIMPLE |  WORK_FLOW")),
-                    'problem_optimization_prompt': openapi.Schema(type=openapi.TYPE_STRING, title=_('Question optimization tips'),
+                    'problem_optimization_prompt': openapi.Schema(type=openapi.TYPE_STRING,
+                                                                  title=_('Question optimization tips'),
                                                                   description=_("Question optimization tips"),
                                                                   default=_(
                                                                       "() contains the user's question. Answer the guessed user's question based on the context ({question}) Requirement: Output a complete question and put it in the <data></data> tag")),

apps/common/mixins/app_model_mixin.py

@@ -7,11 +7,11 @@
     @desc:
 """
 from django.db import models
-from django.utils.translation import gettext_lazy as _
+
 
 class AppModelMixin(models.Model):
-    create_time = models.DateTimeField(verbose_name=_('Create time'), auto_now_add=True)
-    update_time = models.DateTimeField(verbose_name=_('Update time'), auto_now=True)
+    create_time = models.DateTimeField(verbose_name="创建时间", auto_now_add=True)
+    update_time = models.DateTimeField(verbose_name="修改时间", auto_now=True)
 
     class Meta:
         abstract = True