chore: fix test cases

This commit is contained in:
Novice 2025-10-29 14:51:19 +08:00
parent 5f65466551
commit 3d02f425d0
No known key found for this signature in database
GPG Key ID: EE3F68E3105DAAAB
2 changed files with 3 additions and 4 deletions

View File

@ -14,7 +14,6 @@ from core.helper.provider_cache import NoOpProviderCredentialCache
from core.mcp.types import OAuthClientInformation, OAuthClientMetadata, OAuthTokens
from core.tools.entities.common_entities import I18nObject
from core.tools.entities.tool_entities import ToolProviderType
from core.tools.utils.encryption import create_provider_encrypter
if TYPE_CHECKING:
from models.tools import MCPToolProvider
@ -272,6 +271,9 @@ class MCPProviderEntity(BaseModel):
def _decrypt_dict(self, data: dict[str, Any]) -> dict[str, Any]:
"""Generic method to decrypt dictionary fields"""
# Lazy import to avoid circular dependency
from core.tools.utils.encryption import create_provider_encrypter
if not data:
return {}

View File

@ -395,9 +395,6 @@ def test_client_capabilities_default():
# Assert default capabilities
assert received_capabilities is not None
assert received_capabilities.sampling is not None
assert received_capabilities.roots is not None
assert received_capabilities.roots.listChanged is True
def test_client_capabilities_with_custom_callbacks():