diff --git a/api/services/enterprise/plugin_manager_service.py b/api/services/enterprise/plugin_manager_service.py index 67e8b2133e..f1d5f7d1a6 100644 --- a/api/services/enterprise/plugin_manager_service.py +++ b/api/services/enterprise/plugin_manager_service.py @@ -3,7 +3,7 @@ import logging from pydantic import BaseModel -from configs import dify_config +from services.enterprise.base import EnterprisePluginManagerRequest from services.enterprise.base import EnterprisePluginManagerRequest from services.errors.base import BaseServiceError @@ -37,6 +37,7 @@ class PreUninstallPluginRequest(BaseModel): class CredentialPolicyViolationError(BaseServiceError): pass + class PluginManagerService: @classmethod def check_credential_policy_compliance(cls, body: CheckCredentialPolicyComplianceRequest): @@ -69,7 +70,6 @@ class PluginManagerService: "POST", "/pre-uninstall-plugin", json=body.model_dump(), - timeout=dify_config.ENTERPRISE_REQUEST_TIMEOUT, ) except Exception: logger.exception( diff --git a/api/tests/unit_tests/services/enterprise/test_plugin_manager_service.py b/api/tests/unit_tests/services/enterprise/test_plugin_manager_service.py index 6ee328ae2c..dc084b6eb8 100644 --- a/api/tests/unit_tests/services/enterprise/test_plugin_manager_service.py +++ b/api/tests/unit_tests/services/enterprise/test_plugin_manager_service.py @@ -9,7 +9,6 @@ from unittest.mock import patch from httpx import HTTPStatusError -from configs import dify_config from services.enterprise.plugin_manager_service import ( PluginManagerService, PreUninstallPluginRequest, @@ -34,7 +33,6 @@ class TestTryPreUninstallPlugin: "POST", "/pre-uninstall-plugin", json={"tenant_id": "tenant-123", "plugin_unique_identifier": "com.example.my_plugin"}, - timeout=dify_config.ENTERPRISE_REQUEST_TIMEOUT, ) def test_try_pre_uninstall_plugin_http_error_soft_fails(self): @@ -61,7 +59,6 @@ class TestTryPreUninstallPlugin: "POST", "/pre-uninstall-plugin", json={"tenant_id": "tenant-456", "plugin_unique_identifier": "com.example.other_plugin"}, - timeout=dify_config.ENTERPRISE_REQUEST_TIMEOUT, ) mock_logger.exception.assert_called_once() @@ -85,6 +82,5 @@ class TestTryPreUninstallPlugin: "POST", "/pre-uninstall-plugin", json={"tenant_id": "tenant-789", "plugin_unique_identifier": "com.example.failing_plugin"}, - timeout=dify_config.ENTERPRISE_REQUEST_TIMEOUT, ) mock_logger.exception.assert_called_once()