From 65d6da081c5e0d86dfd618ace2858d83f6162667 Mon Sep 17 00:00:00 2001
From: Google Team Member <genai-sdk-bot@google.com>
Date: Thu, 2 Oct 2025 15:38:43 -0700
Subject: [PATCH] ADK changes

PiperOrigin-RevId: 814413627
---
 src/google/adk/memory/vertex_ai_memory_bank_service.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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