From 83063532a0ea67ccb4d9506936fc746dfd13b5d8 Mon Sep 17 00:00:00 2001 From: Yeuoly <45712896+Yeuoly@users.noreply.github.com> Date: Thu, 1 Feb 2024 09:10:32 +0800 Subject: [PATCH] Fix/api tool (#2317) --- api/controllers/console/workspace/tool_providers.py | 4 ++-- api/services/tools_manage_service.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/api/controllers/console/workspace/tool_providers.py b/api/controllers/console/workspace/tool_providers.py index 196747a163..1d5fd74314 100644 --- a/api/controllers/console/workspace/tool_providers.py +++ b/api/controllers/console/workspace/tool_providers.py @@ -171,8 +171,8 @@ class ToolApiProviderUpdateApi(Resource): parser.add_argument('schema', type=str, required=True, nullable=False, location='json') parser.add_argument('provider', type=str, required=True, nullable=False, location='json') parser.add_argument('original_provider', type=str, required=True, nullable=False, location='json') - parser.add_argument('icon', type=str, required=True, nullable=False, location='json') - parser.add_argument('privacy_policy', type=str, required=True, nullable=False, location='json') + parser.add_argument('icon', type=dict, required=True, nullable=False, location='json') + parser.add_argument('privacy_policy', type=str, required=True, nullable=True, location='json') args = parser.parse_args() diff --git a/api/services/tools_manage_service.py b/api/services/tools_manage_service.py index 4b9371b6b6..6fb1853871 100644 --- a/api/services/tools_manage_service.py +++ b/api/services/tools_manage_service.py @@ -362,7 +362,7 @@ class ToolManageService: @staticmethod def update_api_tool_provider( - user_id: str, tenant_id: str, provider_name: str, original_provider: str, icon: str, credentials: dict, + user_id: str, tenant_id: str, provider_name: str, original_provider: str, icon: dict, credentials: dict, schema_type: str, schema: str, privacy_policy: str ): """ @@ -387,7 +387,7 @@ class ToolManageService: # update db provider provider.name = provider_name - provider.icon = icon + provider.icon = json.dumps(icon) provider.schema = schema provider.description = extra_info.get('description', '') provider.schema_type_str = ApiProviderSchemaType.OPENAPI.value