feat: add generateCode function to streamline Python code generation

liuruibin · liuruibin · commit dc90be874d30 · 2026-03-27T15:34:11.000+08:00
diff --git a/apps/tools/api/tool.py b/apps/tools/api/tool.py
@@ -322,3 +322,4 @@ def get_parameters():
                 required=True,
             ),
         ]
+
diff --git a/apps/tools/serializers/tool.py b/apps/tools/serializers/tool.py
@@ -9,15 +9,17 @@
 import tempfile
 import zipfile
 from typing import Dict
-from django.core.cache import cache
+
 import requests
 import uuid_utils.compat as uuid
 from django.core import validators
+from django.core.cache import cache
 from django.db import transaction
 from django.db.models import QuerySet, Q, Subquery, OuterRef, CharField, Value, When, Case
 from django.http import HttpResponse
 from django.utils import timezone
 from django.utils.translation import gettext_lazy as _
+from langchain_core.messages import HumanMessage, AIMessage
 from langchain_mcp_adapters.client import MultiServerMCPClient
 from pylint.lint import Run
 from pylint.reporters import JSON2Reporter
@@ -36,6 +38,7 @@
 from common.utils.tool_code import ToolExecutor
 from knowledge.models import File, FileSourceType, Knowledge
 from maxkb.const import PROJECT_DIR
+from models_provider.models import Model
 from system_manage.models import AuthTargetType, WorkspaceUserResourcePermission
 from system_manage.models.resource_mapping import ResourceMapping
 from system_manage.serializers.resource_mapping_serializers import ResourceMappingSerializer
@@ -1108,6 +1111,68 @@ def upload(self):
             file.save(self.data.get('file').read())
             return file_id
 
+    class GenerateCodeSerializer(serializers.Serializer):
+        workspace_id = serializers.CharField(required=True, label=_('Workspace ID'))
+        model_id = serializers.UUIDField(required=True, label=_('Model ID'))
+        prompt = serializers.CharField(required=True, label=_('Prompt'))
+        messages = serializers.ListField(required=True, label=_('Messages'))
+        model_params_setting = serializers.DictField(required=False, default=dict, label=_('Model Params Setting'))
+        init_field_list = serializers.ListField(required=False, default=list, label=_('Init Field List'))
+        input_field_list = serializers.ListField(required=False, default=list, label=_('Input Field List'))
+
+        def generate_code(self):
+            from models_provider.tools import get_model_instance_by_model_workspace_id
+            from application.flow.tools import to_stream_response_simple
+
+            self.is_valid(raise_exception=True)
+
+            workspace_id = self.data.get('workspace_id')
+            model_id = self.data.get('model_id')
+            prompt = self.data.get('prompt')
+            messages = self.data.get('messages')
+            model_params_setting = self.data.get('model_params_setting')
+            init_field_list = self.data.get('init_field_list')
+            input_field_list = self.data.get('input_field_list')
+
+            message = messages[-1]['content']
+            q = prompt.replace(
+                "{userInput}", message
+            ).replace(
+                "{initFieldList}", json.dumps(init_field_list)
+            ).replace(
+                "{inputFieldList}", json.dumps(input_field_list)
+            )
+
+            messages[-1]['content'] = q
+            SUPPORTED_MODEL_TYPES = ["LLM"]
+            model_exist = QuerySet(Model).filter(
+                id=model_id,
+                model_type__in=SUPPORTED_MODEL_TYPES
+            ).exists()
+            if not model_exist:
+                raise Exception(_("Model does not exists or is not an LLM model"))
+
+            def process():
+                model = get_model_instance_by_model_workspace_id(
+                    model_id=model_id, workspace_id=workspace_id, **model_params_setting
+                )
+                try:
+                    for r in model.stream([
+                        # SystemMessage(content=SYSTEM_ROLE),
+                        *[
+                            HumanMessage(
+                                content=m.get('content')
+                            ) if m.get('role') == 'user' else AIMessage(
+                                content=m.get('content')
+                            ) for m in messages
+                        ]
+                    ]):
+                        yield 'data: ' + json.dumps({'content': r.content}) + '\n\n'
+                except Exception as e:
+                    yield 'data: ' + json.dumps({'error': str(e)}) + '\n\n'
+
+            return to_stream_response_simple(process())
+
 
 class ToolTreeSerializer(serializers.Serializer):
     class Query(serializers.Serializer):
diff --git a/apps/tools/urls.py b/apps/tools/urls.py
@@ -15,6 +15,7 @@
     path('workspace/<str:workspace_id>/tool/tool_list', views.ToolView.Query.as_view()),
     path('workspace/<str:workspace_id>/tool/test_connection', views.ToolView.TestConnection.as_view()),
     path('workspace/<str:workspace_id>/tool/upload_skill_file', views.ToolView.UploadSkillFile.as_view()),
+    path('workspace/<str:workspace_id>/tool/generate_code', views.ToolView.GenerateCode.as_view()),
     path('workspace/<str:workspace_id>/tool/<str:tool_id>', views.ToolView.Operate.as_view()),
     path('workspace/<str:workspace_id>/tool/<str:tool_id>/publish', views.ToolWorkflowView.Publish.as_view()),
     path('workspace/<str:workspace_id>/tool/<str:tool_id>/debug', views.ToolWorkflowDebugView.as_view()),
diff --git a/apps/tools/views/tool.py b/apps/tools/views/tool.py
@@ -596,3 +596,20 @@ def put(self, request: Request, workspace_id: str):
                 'user_id': request.user.id,
                 'file': request.FILES.get('file'),
             }).upload())
+
+    class GenerateCode(APIView):
+        authentication_classes = [TokenAuth]
+
+        @has_permissions(
+            PermissionConstants.TOOL_CREATE.get_workspace_permission(),
+            PermissionConstants.TOOL_CREATE.get_workspace_permission_workspace_manage_role(),
+            PermissionConstants.TOOL_EDIT.get_workspace_permission(),
+            PermissionConstants.TOOL_EDIT.get_workspace_permission_workspace_manage_role(),
+            RoleConstants.WORKSPACE_MANAGE.get_workspace_role(),
+            RoleConstants.USER.get_workspace_role()
+        )
+        def post(self, request: Request, workspace_id: str):
+            return ToolSerializer.GenerateCodeSerializer(data={
+                'workspace_id': workspace_id,
+                **request.data
+            }).generate_code()
diff --git a/ui/src/api/system-resource-management/tool.ts b/ui/src/api/system-resource-management/tool.ts
@@ -1,5 +1,5 @@
 import { Result } from '@/request/Result'
-import { get, post, del, put, exportFile } from '@/request/index'
+import { get, post, del, put, exportFile, postStream } from '@/request/index'
 import { type Ref } from 'vue'
 import type { pageRequest } from '@/api/type/common'
 import type { toolData } from '@/api/type/tool'
@@ -156,6 +156,15 @@ const uploadSkillFile: (data: toolData, loading?: Ref<boolean>) => Promise<Resul
   return put(`${prefix}/upload_skill_file`, data, undefined, loading)
 }
 
+const generateCode: (data:any) => Promise<Result<any>> = (
+  data: any,
+) => {
+  const p = (window.MaxKB?.prefix ? window.MaxKB?.prefix : '/admin') + '/api'
+  return postStream(
+    `${p}${prefix}/generate_code`,
+    data,
+  )
+}
 
 
 export default {
@@ -173,4 +182,5 @@ export default {
   pageToolRecord,
   getToolRecordDetail,
   uploadSkillFile,
+  generateCode,
 }
diff --git a/ui/src/api/system-shared/tool.ts b/ui/src/api/system-shared/tool.ts
@@ -1,5 +1,5 @@
 import { Result } from '@/request/Result'
-import { get, post, del, put, exportFile } from '@/request/index'
+import { get, post, del, put, exportFile, postStream } from '@/request/index'
 import { type Ref } from 'vue'
 import type { pageRequest } from '@/api/type/common'
 import type { toolData, AddInternalToolParam } from '@/api/type/tool'
@@ -200,6 +200,15 @@ const uploadSkillFile: (data: toolData, loading?: Ref<boolean>) => Promise<Resul
   return put(`${prefix}/upload_skill_file`, data, undefined, loading)
 }
 
+const generateCode: (data:any) => Promise<Result<any>> = (
+  data: any,
+) => {
+  const p = (window.MaxKB?.prefix ? window.MaxKB?.prefix : '/admin') + '/api'
+  return postStream(
+    `${p}${prefix}/generate_code`,
+    data,
+  )
+}
 
 
 export default {
@@ -222,4 +231,5 @@ export default {
   pageToolRecord,
   getToolRecordDetail,
   uploadSkillFile,
+  generateCode,
 }
diff --git a/ui/src/api/tool/tool.ts b/ui/src/api/tool/tool.ts
@@ -299,6 +299,18 @@ const debugToolWorkflow: (tool_id: string, data: any) => Promise<any> = (tool_id
   const p = (window.MaxKB?.prefix ? window.MaxKB?.prefix : '/admin') + '/api'
   return postStream(`${p}${prefix.value}/${tool_id}/debug`, data)
 }
+
+const generateCode: (data:any) => Promise<Result<any>> = (
+  data: any,
+) => {
+  const p = (window.MaxKB?.prefix ? window.MaxKB?.prefix : '/admin') + '/api'
+  return postStream(
+    `${p}${prefix.value}/generate_code`,
+    data,
+  )
+}
+
+
 export default {
   getToolList,
   getAllToolList,
@@ -326,4 +338,5 @@ export default {
   publish,
   exportToolWorkflow,
   debugToolWorkflow,
+  generateCode,
 }
diff --git a/ui/src/views/tool/ToolFormDrawer.vue b/ui/src/views/tool/ToolFormDrawer.vue
@@ -170,13 +170,23 @@
         </el-table-column>
       </el-table>
 
-      <h4 class="title-decoration-1 mb-16">
-        {{ $t('views.tool.form.param.code') }}
-        <span class="color-danger" style="margin-left: -10px">*</span>
-        <el-text type="info" class="color-secondary">
-          {{ $t('views.tool.form.param.paramInfo2') }}
-        </el-text>
-      </h4>
+      <div class="flex-between">
+        <h4 class="title-decoration-1 mb-16">
+          {{ $t('views.tool.form.param.code') }}
+          <span class="color-danger" style="margin-left: -10px">*</span>
+          <el-text type="info" class="color-secondary">
+            {{ $t('views.tool.form.param.paramInfo2') }}
+          </el-text>
+        </h4>
+        <el-button
+          type="primary"
+           @click="openGenerateCodeDialog"
+          link
+        >
+          <AppIcon iconName="app-generate-star" class="mr-4"></AppIcon>
+          {{ $t('views.application.generateDialog.label') }}
+        </el-button>
+      </div>
 
       <div class="mb-8" v-if="showEditor">
         <CodemirrorEditor
@@ -217,6 +227,7 @@
     <FieldFormDialog ref="FieldFormDialogRef" @refresh="refreshFieldList" />
     <UserFieldFormDialog ref="UserFieldFormDialogRef" @refresh="refreshInitFieldList" />
     <EditAvatarDialog ref="EditAvatarDialogRef" @refresh="refreshTool" />
+    <GenerateCodeDialog ref="GenerateCodeDialogRef" :toolData="form" @replace="replaceCode"/>
   </el-drawer>
 </template>
 
@@ -237,6 +248,7 @@ import { useRoute } from 'vue-router'
 import useStore from '@/stores'
 import permissionMap from '@/permission'
 import { loadSharedApi } from '@/utils/dynamics-api/shared-api'
+import GenerateCodeDialog from "@/views/tool/component/GenerateCodeDialog.vue";
 const route = useRoute()
 
 const props = defineProps({
@@ -265,6 +277,7 @@ const UserFieldFormDialogRef = ref()
 const EditAvatarDialogRef = ref()
 const initFieldTableRef = ref()
 const inputFieldTableRef = ref()
+const GenerateCodeDialogRef = ref()
 
 const FormRef = ref()
 
@@ -392,6 +405,15 @@ function openEditAvatar() {
   EditAvatarDialogRef.value.open(form.value)
 }
 
+function openGenerateCodeDialog() {
+  GenerateCodeDialogRef.value?.open(form.value.init_field_list, form.value.input_field_list)
+}
+
+function replaceCode(code: string) {
+  const match = code.replace('```python', '').replace('```', '')
+  form.value.code = match
+}
+
 const submit = async (formEl: FormInstance | undefined) => {
   if (!formEl) return
   await formEl.validate((valid: any) => {
diff --git a/ui/src/views/tool/component/GenerateCodeDialog.vue b/ui/src/views/tool/component/GenerateCodeDialog.vue

Original file line number	Diff line number	Diff line change
`@@ -322,3 +322,4 @@ def get_parameters():`
`322`	`322`	`required=True,`
`323`	`323`	`),`
`324`	`324`	`]`
	`325`	`+`