feat: Support internal functionlib

--story=1017939 --user=刘瑞斌 【函数库】- 支持 系统内置函数 https://www.tapd.cn/57709429/s/1665943

Author: liuruibin

apps/application/flow/step_node/function_lib_node/impl/base_function_lib_node.py

@@ -108,13 +108,10 @@ def execute(self, function_lib_id, input_field_list, **kwargs) -> NodeResult:
                                              ), **field}
                    for field in
                    function_lib.input_field_list]}
-        init_field_list = json.loads(rsa_long_decrypt(function_lib.init_field_list))
-        init_params = {field.get('field'): field.get('value') if field.get('value',
-                                                                           None) is not None else field.get(
-            'default_value') for field in init_field_list}
+
         self.context['params'] = params
         # 合并初始化参数
-        all_params = init_params | params
+        all_params = json.loads(rsa_long_decrypt(function_lib.init_params)) | params
         result = function_executor.exec_code(function_lib.code, all_params)
         return NodeResult({'result': result}, {}, _write_context=write_context)
 

apps/function_lib/migrations/0003_functionlib_function_type_functionlib_icon_and_more.py

@@ -1,9 +1,30 @@
-# Generated by Django 4.2.15 on 2025-03-13 02:56
+# Generated by Django 4.2.15 on 2025-03-13 07:21
 
 from django.db import migrations, models
 
-function_template = """
-INSERT INTO function_lib (create_time, update_time, id, name, "desc", code, input_field_list, user_id, is_active, permission_type, function_type, icon, init_field_list, template_id) VALUES ('2025-02-26 03:36:48.187286 +00:00', '2025-03-11 07:23:46.123972 +00:00', 'e89ad2ae-f3f2-11ef-ad09-0242ac110002', 'Google Search', 'Google Web Search', e'def google_search(query, apikey, cx):
+function_template = '''
+INSERT INTO function_lib (create_time, update_time, id, name, "desc", code, input_field_list, user_id, is_active, permission_type, function_type, icon, init_field_list, init_params, template_id) VALUES ('2025-03-10 06:20:35.945414 +00:00', '2025-03-10 09:19:23.608026 +00:00', 'c75cb48e-fd77-11ef-84d2-5618c4394482', '博查AI', '从博查搜索任何信息和网页URL', e'def bocha_search(query, apikey):
+    import requests
+    import json
+    url = "https://api.bochaai.com/v1/web-search"
+    payload = json.dumps({
+        "query": query,
+        "Boolean": "true",
+        "count": 8
+    })
+
+    headers = {
+        "Authorization": "Bearer " + apikey, #鉴权参数，示例：Bearer xxxxxx，API KEY请先前往博查AI开放平台（https://open.bochaai.com）> API KEY 管理中获取。
+        "Content-Type": "application/json"
+    }
+
+    response = requests.request("POST", url, headers=headers, data=payload)
+    if response.status_code == 200:
+        return response.json()
+    else:
+        raise Exception(f"API请求失败: {response.status_code}, 错误信息: {response.text}")
+        return (response.text)', '{"{\\"name\\": \\"query\\", \\"type\\": \\"string\\", \\"source\\": \\"reference\\", \\"is_required\\": true}"}', 'f0dd8f71-e4ee-11ee-8c84-a8a1595801ab', TRUE, 'PUBLIC', 'INTERNAL', '/src/assets/fx/bochaai/icon.png', '[{"attrs": {"type": "password", "maxlength": 200, "minlength": 1, "show-password": true, "show-word-limit": true}, "field": "apikey", "label": "apikey", "required": true, "input_type": "PasswordInput", "props_info": {"rules": [{"message": "apikey 为必填属性", "required": true}, {"max": 200, "min": 1, "message": "apikey长度在 1 到 200 个字符", "trigger": "blur"}]}, "default_value": "x", "show_default_value": false}]', '', NULL);
+INSERT INTO function_lib (create_time, update_time, id, name, "desc", code, input_field_list, user_id, is_active, permission_type, function_type, icon, init_field_list, init_params, template_id) VALUES ('2025-02-26 03:36:48.187286 +00:00', '2025-03-11 07:23:46.123972 +00:00', 'e89ad2ae-f3f2-11ef-ad09-0242ac110002', 'Google Search', 'Google Web Search', e'def google_search(query, apikey, cx):
     import requests
     import json
     url = "https://customsearch.googleapis.com/customsearch/v1"
@@ -16,64 +37,40 @@
 
     response = requests.get(url, params=params)
     if response.status_code == 200:
-        contexts = response.json()[\'items\']
-        max_context = min(len(contexts), 10)
-        format_contexts = []
-        for i in range(max_context):
-            formatted_context = f"[[引用:{i + 1}]]\\n网页标题:{contexts[i][\'title\']}\\n网页链接:{contexts[i][\'link\']}\\n网页内容:{contexts[i][\'snippet\']}\\n网站名称:{contexts[i][\'displayLink\']}"
-            format_contexts.append(formatted_context)
-
-            context = "\\n\\n".join(format_contexts)
-        return context
+        return response.json()
     else:
-        print(f"Error: {response.status_code}")
-        return None', '{"{\"name\": \"query\", \"type\": \"string\", \"source\": \"reference\", \"is_required\": true}"}', 'f0dd8f71-e4ee-11ee-8c84-a8a1595801ab', true, 'PUBLIC', 'INTERNAL', '/src/assets/fx/google_search/icon.png', 'd3OH0CJYyODrbL1+EgFQ0umVWmb3ZpLsZkv5eqskvXeikcuHXoF0Ta/sVBvdLBDcm4LCLKEgZkUOq9JQb+9qiVA6IHq0ZSFj+JpMIDgl3YVnFbhyTpGp3nAxOm2Y7ufmeMtVIOq20Is1ynOrdIB7UHWrvEOmSq+O8kY612X6VtMHrshkMFL9huiRdQThhnsEZdLdhHgkp9BjpCBJl8SbzTjJmVX4QeswobF5Kr3IRPZBPzrFh0S4jM6Ie0dW6OA5UwgCOBeIVyBxNytarj629kz9EWluns/UGVY8IoXQsphB9m63tXADZ9+djKfRV6TUBosi963DZLi893yY+M7o5UN+eK2u+6FQfy8pTwwiSdNIXtC16pcpvJvG+2jmNeTF1bzOS1fok+L0k/kHhOyzARk8BG1DDyvCsn73QIlnTbKrqCYN9q5zsRQvZPhBMg/g1/oOL147pRArs41htrzOObFP6MbQVM5DXX8tbvro24F7EjGMktoYpPLhqtiMNv9qn7tQbpRfDMYvMHUMiXfLXSroQH2u9a2pb7rBk0QtNJF18069t3p6nXRw3UsUsOXHvYZ1AyBQb6Jy+JbT2OIyUVKfqCvv86ZgnX6JNGqCQXHiFMuxmFI4i62b8f4uPpPBGw9PS5LFZYTm4VWHALoOKQyyOdrbIxk2+lCii5DeT2NkYNi+EaHCQ8/TKHhyG1DGymZPZ7VEK+WxmCwRlr6DEtQfEntlFyWgAimhHtxaTTM7uVViAU3ddCDUtmrOQBteK5c0BzvsMcLMSvsev5UKKo7Q1jipPORgKCun+d0El/qg4Fe38yqWHXBS4jbTuwtiNDsdabrZ0rJnDm3BRJQydouak1rg9GDXcCsHligdR5hJJqP1VqLLUDuoRC8V9urFI0udNQLWKFYPog3J4S/XYbdeQZvio8ZfDQRSWtADWj9B8voLRK8aJMT6HkKXsveRw8GlDNKu/JK44KS7uLCaE9QsIp4zYxRfpkYTx29VXI5vm+a6zvPGz9sJO9nowIOjTV5RMZ3MoS49JM7oGAjJ/5gTk5ZYWC56kmr7712NKZYxiDhDWhs+IYewujcEnj17L9hULDWQCqL+xGR25ayMGukWp4TREVTZxn3p0H7VZrlSrSiLWCISl6+dLED3reclcVsUsqR1YxHCNnPv+0VKmxS8MiMANYf3mf2QoI0F3tHst+sMSq08eJNLbYqCTCP6hOhG5rVbh3pDvSbpfMaPsRumzi3R0FY6mudBoG9g31Jl4N4HF82luL7Oj/kIkzA18aq1Zcl2UTy7Nnqjn8oTRqc8I3RjFxV1BY2zaiK3pvMPKDw7QBSWTa8T6xdq1/17cETKQRwVjjb8gWy1R8Q9khkEMN49TaZDoMLIsCN6EzvsmHXmUdxFxh5kBmG3GLmbttpYpKOPwGCFbykUf8x8jxubOLrgLsyUhNk/vMI/Y4ZHWVIAjZNlRowJYklyHMSmYFwSywczvg1mkE4qmAcKPJGOuBWzlh788VlQrc5E8yil7VexqW8nN/ho/aFd+G+RSPECaE6AvklZa6CPZTZzyhUPqHLVSG0Yq7lWxqc1XUbW3BJWPRF64ghf8VCAguSZpwSoTMBXBKGWa9er0cLSCZheTQSs5tIOkt93zhnATjeebvJPR/IhOZpM6m83o9JHtNLR6jsXS8R87+VivMJA+fp1NyWemwqBm9BNKt3A2E8=', null);
-INSERT INTO function_lib (create_time, update_time, id, name, "desc", code, input_field_list, user_id, is_active, permission_type, function_type, icon, init_field_list, template_id) VALUES ('2025-03-10 06:20:35.945414 +00:00', '2025-03-10 09:19:23.608026 +00:00', 'c75cb48e-fd77-11ef-84d2-5618c4394482', '博查AI', '从博查搜索任何信息和网页URL', e'def bocha_search(query, apikey):
-    import requests
+        raise Exception(f"API请求失败: {response.status_code}, 错误信息: {response.text}")
+        return (response.text)', '{"{\\"name\\": \\"query\\", \\"type\\": \\"string\\", \\"source\\": \\"reference\\", \\"is_required\\": true}"}', 'f0dd8f71-e4ee-11ee-8c84-a8a1595801ab', TRUE, 'PUBLIC', 'INTERNAL', '/src/assets/fx/google_search/icon.png', '[{"attrs": {"type": "password", "maxlength": 200, "minlength": 1, "show-password": true, "show-word-limit": true}, "field": "apikey", "label": "apikey", "required": true, "input_type": "PasswordInput", "props_info": {"rules": [{"message": "apikey 为必填属性", "required": true}, {"max": 200, "min": 1, "message": "apikey长度在 1 到 200 个字符", "trigger": "blur"}]}, "default_value": "x", "show_default_value": false}, {"attrs": {"maxlength": 200, "minlength": 1, "show-word-limit": true}, "field": "cx", "label": "cx", "required": true, "input_type": "TextInput", "props_info": {"rules": [{"message": "cx 为必填属性", "required": true}, {"max": 200, "min": 1, "message": "cx长度在 1 到 200 个字符", "trigger": "blur"}]}, "default_value": "x", "show_default_value": false}]', '', NULL);
+INSERT INTO function_lib (create_time, update_time, id, name, "desc", code, input_field_list, user_id, is_active, permission_type, function_type, icon, init_field_list, init_params, template_id) VALUES ('2025-02-25 07:44:40.141515 +00:00', '2025-03-11 06:33:53.248495 +00:00', '5e912f00-f34c-11ef-8a9c-5618c4394482', 'LangSearch API', e'A Web Search API supporting natural language search
+', e'
+def langsearch(query, apikey):
     import json
-    url = "https://api.bochaai.com/v1/web-search"
-    payload = json.dumps({
-        "query": query,
-        "Boolean": \'true\',
-        "count": 8
-    })
-
-    headers = {
-        \'Authorization\': \'Bearer \' + apikey, #鉴权参数，示例：Bearer xxxxxx，API KEY请先前往博查AI开放平台（https://open.bochaai.com）> API KEY 管理中获取。
-        \'Content-Type\': \'application/json\'
-    }
-
-    response = requests.request("POST", url, headers=headers, data=payload)
-    return response.json()', '{"{\"name\": \"query\", \"type\": \"string\", \"source\": \"reference\", \"is_required\": true}"}', 'f0dd8f71-e4ee-11ee-8c84-a8a1595801ab', true, 'PUBLIC', 'INTERNAL', '/src/assets/fx/bochaai/icon.png', 'X1ukYeWnI4usm6Emy2bVb0NigczyEZVe4MVst83D1Hgiqti9CNOiLNcANkuX05K+huX1D5p2g7LVFghgpM8ddoSMax5faAsG7tw1UZtPoexYp0ssjXDHBfFCbZcIs/X4Gjio5Lk33X7FQojOM1BIR1P/itxv0U5G1Q45FKg51kdiZoafKYuf24aqqMNDscCnLLQxwVt259BGU5JY4zIuv9hxAxX/7QCXJmgu5CrheuLiTqjRS4TzqBUrLtB/AT4QjrlILeZBwnQ5k2BKFp0mbVeAA+rAaJzBg0GmmnljsJolrKjdPuCWEwCnNw2uptlE3DePXvELsAPEuADBoeE/L5q4RFFzNvkgREsfWn2PaeSkutLUAItE+S3+Ol0QGmnIE7OabifpAgNDH6aalRjYcPNuWZ0V17XzcjFXRwpQFHd6TjkfsgpI5J2m+rZyGayZ9+4XFNbzv9u84TurOvLTGa5IHTbP9Yj4ND4xXyGT+zN0TeOZ3c7XeyoqJWVMNcsp6dc7hiQT5jc9rpRmNUtYo47KPcZfRHq3ToQjdJhx7ZpLA1Uun1RcW+lKhrDpy/ce2f8TjC2+gE7HpDIYeYMB+W0eDX/LHICAWdw/v8gaTn0wlmw9TVW8loqgVBQVlWdEUV2ErkJNfxliwcCD/AxWXlrvaF1FpLOGy6wgUPExoetnDS5BftFumvOEj6R3jfy7jThdEI3+YtNGTg6QA9cBlJ7Ey3nrixZGt/+j9TwAnPhNXcjs/SVGUzxBx+VmUN+jCXMJr93D/yOwnlUkkHljRisyoWIXXOlvOx9y41p2yJRXA1SWQ5n7O/FUA38Z/6w89OMNnPJHfMZgztV8SxgBsayR2echA6XXdmvBLbZlgyr1AugwcR+ZN2t3ayRG8muRYL9S6gNSywxOyaLKFuesOCDO+vU1NdegTFMKYoJuXts0W7bIzYjDB1vQyX9ggAzodTWLUz9sjzAh4zDMbnxPJdNXGVx4O61H0qoU40bPGLZGnPl1+b+mahgCVM7Q28UA', null);
-INSERT INTO function_lib (create_time, update_time, id, name, "desc", code, input_field_list, user_id, is_active, permission_type, function_type, icon, init_field_list, template_id) VALUES ('2025-02-25 07:44:40.141515 +00:00', '2025-03-11 06:33:53.248495 +00:00', '5e912f00-f34c-11ef-8a9c-5618c4394482', 'LangSearch API', e'A Web Search API supporting natural language search
-', e'def search(query, apikey):
     import requests
-    import json
+    
     url = "https://api.langsearch.com/v1/web-search"
     payload = json.dumps({
         "query": query,
         "summary": True,
         "freshness": "noLimit",
-        "summary": True,
         "livecrawl": True,
         "count": 20
     })
     headers = {
-        \'Authorization\': apikey,
-        \'Content-Type\': \'application/json\'
+        "Authorization": apikey,
+        "Content-Type": "application/json"
     }
     # key从官网申请 https://langsearch.com/
     response = requests.request("POST", url, headers=headers, data=payload)
     if response.status_code == 200:
-        return str(response.json())
+        return response.json()
     else:
         raise Exception(f"API请求失败: {response.status_code}, 错误信息: {response.text}")
-    return(response.text)', '{"{\"name\": \"query\", \"type\": \"string\", \"source\": \"reference\", \"is_required\": true}"}', 'f0dd8f71-e4ee-11ee-8c84-a8a1595801ab', true, 'PUBLIC', 'INTERNAL', '/src/assets/fx/langsearch/icon.png', 'DGp5NooY3zGFJjzQ3t+uscqoTZbZHax2RdHvfyZ6NCcb/5JhcJHsB9rzCoGjQVvewugnFzjIdATJcryhjis3enkl4l1TAFdOxDT7Py63JpuOapoE52pkhsE9QYW5f1tg6Nd9gssYY6rYnyqMvSf4wUjQJBDQOxv29+CqRCTWSUnQfHD8aWUaPe4rXZObdFnLLg42ROg8cVu5UJkJXdA6EcRF8G082NqQ+LOB49OSffkfhg+fk4nibKIvYcFo3Y1MP+bTuqO23O5tzbNUnPmqsqGPnxbJqZmvJvkksZg1L/NTzYSvvZYuPNUbGfR7wC9FucFN/TFFRMGFdCxtiXGlUQhrZvz8W5vjMKlzsRfnKXTFS4Cg62MmCvQFK6yybLMx0QMOc/yftFt4MkZdLepjim08ctLav3mNApaFNtZ0Isz+A40jpimrZkj5Ahg1KpXCQXUQyOtmQXc6MUVVFOzFlXTQfa3ua4tkiAdAwZEHvaYgjQh28dq4nrtgqFhe+hojJ7kIlJsqtPvb9/+RYwIWVlwef4jVDV2a3Eu+1fMlid1ZJT/KwNkCv2179jRaYhXGkXfU8ii1OOhL9mIGtSJ5lThuJouXgQvNyRr7MajMCVDZYUEOf+kolH/SWEtCFxImT4zaMVFcuZyzLWGh48BX9KuKoAhfPToSs4RgTk0G9h1hQqCrZP4GSXXiBgpMiz2niFGmWk1oWRa6XfU3Dy3Xkft4Q1+iQbnSXcfzZggYoZ4g6qbaf7iBUKN6mW/BvLbSVuqftgy9KF1dHMD95bSMBX6o47rJyAUbaJduLRsH086uNTDTaIE2f/F0ahD5WbokCQGzhWp6W8h4YgIcj8SC3uHiyCOeW5nIm/zEM2+bY4BuClr1IuvYR32GBu0wDma1f7gxEU7Exu49oDojjSdOaFb8isvsaJ9dCNOP/V/ONldBS9VOcZ9l4RNawWwLhrzZhhml5TUI5ZCUywHDFqu/0uBnTI1AHlg+NAs4iJ4VZoVJztZrQmNa9etbRWvhU8yd', null);
-"""
+        return (response.text)', '{"{\\"name\\": \\"query\\", \\"type\\": \\"string\\", \\"source\\": \\"reference\\", \\"is_required\\": true}"}', 'f0dd8f71-e4ee-11ee-8c84-a8a1595801ab', TRUE, 'PUBLIC', 'INTERNAL', '/src/assets/fx/langsearch/icon.png', '[{"attrs": {"type": "password", "maxlength": 200, "minlength": 1, "show-password": true, "show-word-limit": true}, "field": "apikey", "label": "apikey", "required": true, "input_type": "PasswordInput", "props_info": {"rules": [{"message": "apikey 为必填属性", "required": true}, {"max": 200, "min": 1, "message": "apikey长度在 1 到 200 个字符", "trigger": "blur"}]}, "default_value": "x", "show_default_value": false}]', '', NULL);
 
+'''
 
-class Migration(migrations.Migration):
 
+class Migration(migrations.Migration):
     dependencies = [
         ('function_lib', '0002_functionlib_is_active_functionlib_permission_type'),
     ]
@@ -82,17 +79,24 @@ class Migration(migrations.Migration):
         migrations.AddField(
             model_name='functionlib',
             name='function_type',
-            field=models.CharField(choices=[('INTERNAL', '内置'), ('PUBLIC', '公开')], default='PUBLIC', max_length=20, verbose_name='函数类型'),
+            field=models.CharField(choices=[('INTERNAL', '内置'), ('PUBLIC', '公开')],
+                                   default='PUBLIC', max_length=20, verbose_name='函数类型'),
         ),
         migrations.AddField(
             model_name='functionlib',
             name='icon',
-            field=models.CharField(default='/ui/favicon.ico', max_length=256, verbose_name='函数库icon'),
+            field=models.CharField(default='/ui/favicon.ico', max_length=256,
+                                   verbose_name='函数库icon'),
         ),
         migrations.AddField(
             model_name='functionlib',
             name='init_field_list',
-            field=models.CharField(max_length=102400, null=True, verbose_name='启动字段列表'),
+            field=models.JSONField(default=list, verbose_name='启动字段列表'),
+        ),
+        migrations.AddField(
+            model_name='functionlib',
+            name='init_params',
+            field=models.CharField(max_length=102400, null=True, verbose_name='初始化参数'),
         ),
         migrations.AddField(
             model_name='functionlib',

apps/function_lib/models/function.py

@@ -33,14 +33,15 @@ class FunctionLib(AppModelMixin):
     input_field_list = ArrayField(verbose_name="输入字段列表",
                                   base_field=models.JSONField(verbose_name="输入字段", default=dict)
                                   , default=list)
-    init_field_list = models.CharField(max_length=102400, verbose_name="启动字段列表", null=True)
+    init_field_list = models.JSONField(verbose_name="启动字段列表", default=list)
     icon = models.CharField(max_length=256, verbose_name="函数库icon", default="/ui/favicon.ico")
     is_active = models.BooleanField(default=True)
     permission_type = models.CharField(max_length=20, verbose_name='权限类型', choices=PermissionType.choices,
                                        default=PermissionType.PRIVATE)
     function_type = models.CharField(max_length=20, verbose_name='函数类型', choices=FunctionType.choices,
                                        default=FunctionType.PUBLIC)
     template_id =  models.UUIDField(max_length=128, verbose_name="模版id", null=True, default=None)
+    init_params = models.CharField(max_length=102400, verbose_name="初始化参数", null=True)
 
     class Meta:
         db_table = "function_lib"