From 1fb280aa3db8940591a831c72c70b5ec1639e34b Mon Sep 17 00:00:00 2001
From: wxg0103 <727495428@qq.com>
Date: Fri, 12 Sep 2025 15:34:13 +0800
Subject: [PATCH] feat: add source field to user serializer and update user
 creation logic

---
 apps/users/serializers/user.py | 10 +++++++++-
 ui/src/workflow/common/data.ts |  2 +-
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/apps/users/serializers/user.py b/apps/users/serializers/user.py
index cca96534d..529889305 100644
--- a/apps/users/serializers/user.py
+++ b/apps/users/serializers/user.py
@@ -61,6 +61,7 @@ class CreateUserSerializer(serializers.Serializer):
     email = serializers.EmailField(required=True, label=_('Email'))
     nick_name = serializers.CharField(required=False, label=_('Nick name'))
     phone = serializers.CharField(required=False, label=_('Phone'))
+    source = serializers.CharField(required=False, label=_('Source'), default='LOCAL')
 
 
 def is_workspace_manage(user_id: str, workspace_id: str):
@@ -172,6 +173,13 @@ class UserManageSerializer(serializers.Serializer):
             allow_null=True,
             allow_blank=True
         )
+        source = serializers.CharField(
+            required=False,
+            label=_("Source"),
+            max_length=20,
+            default="LOCAL"
+        )
+
 
         def is_valid(self, *, raise_exception=True):
             super().is_valid(raise_exception=True)
@@ -332,7 +340,7 @@ class UserManageSerializer(serializers.Serializer):
             username=instance.get('username'),
             password=password_encrypt(instance.get('password')),
             role=RoleConstants.USER.name,
-            source="LOCAL",
+            source=instance.get('source', 'LOCAL'),
             is_active=True
         )
         update_user_role(instance, user, user_id)
diff --git a/ui/src/workflow/common/data.ts b/ui/src/workflow/common/data.ts
index 9ac8c89ca..01c925981 100644
--- a/ui/src/workflow/common/data.ts
+++ b/ui/src/workflow/common/data.ts
@@ -523,7 +523,7 @@ export const nodeDict: any = {
   [WorkflowType.VariableAssignNode]: variableAssignNode,
   [WorkflowType.McpNode]: mcpNode,
   [WorkflowType.TextToVideoGenerateNode]: textToVideoNode,
-  [WorkflowType.ImageToVideoGenerateNode]: imageToVideoNode
+  [WorkflowType.ImageToVideoGenerateNode]: imageToVideoNode,
   [WorkflowType.IntentNode]: intentNode,
 }
 export function isWorkFlow(type: string | undefined) {