diff --git a/src/google/adk/memory/vertex_ai_memory_bank_service.py b/src/google/adk/memory/vertex_ai_memory_bank_service.py index a3fb9b6a..03f4c392 100644 --- a/src/google/adk/memory/vertex_ai_memory_bank_service.py +++ b/src/google/adk/memory/vertex_ai_memory_bank_service.py @@ -20,11 +20,7 @@ from typing import TYPE_CHECKING from google.genai import types from typing_extensions import override - -try: - import vertexai -except ImportError: - from google.cloud.aiplatform import vertexai +import vertexai from .base_memory_service import BaseMemoryService from .base_memory_service import SearchMemoryResponse