fix

Author: loveyana

tests/test_ve_identity_mcp_tool.py

@@ -16,8 +16,7 @@
 """Unit tests for VeIdentityMcpTool."""
 
 import pytest
-from unittest.mock import Mock, AsyncMock, patch, MagicMock
-from google.adk.tools.tool_context import ToolContext
+from unittest.mock import Mock, AsyncMock, patch
 
 from veadk.integrations.ve_identity.mcp_tool import VeIdentityMcpTool
 from veadk.integrations.ve_identity.auth_config import api_key_auth, oauth2_auth

tests/test_ve_identity_mcp_toolset.py

@@ -15,8 +15,7 @@
 """Unit tests for VeIdentityMcpToolset."""
 
 import pytest
-from unittest.mock import Mock, AsyncMock, patch, MagicMock
-from google.adk.agents.readonly_context import ReadonlyContext
+from unittest.mock import Mock, AsyncMock, patch
 
 from veadk.integrations.ve_identity.mcp_toolset import VeIdentityMcpToolset
 from veadk.integrations.ve_identity.auth_config import api_key_auth, oauth2_auth