mirror of https://github.com/langgenius/dify.git
chore: fix test cases
This commit is contained in:
parent
5f65466551
commit
3d02f425d0
|
|
@ -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 {}
|
||||
|
||||
|
|
|
|||
|
|
@ -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():
|
||||
|
|
|
|||
Loading…
Reference in New Issue