Skip to content

Commit 65d6da0

Browse files
google-genai-botcopybara-github
authored andcommitted
ADK changes
PiperOrigin-RevId: 814413627
1 parent b170a84 commit 65d6da0

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/google/adk/memory/vertex_ai_memory_bank_service.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,11 @@
2020

2121
from google.genai import types
2222
from typing_extensions import override
23-
import vertexai
23+
24+
try:
25+
import vertexai
26+
except ImportError:
27+
from google.cloud.aiplatform import vertexai
2428

2529
from .base_memory_service import BaseMemoryService
2630
from .base_memory_service import SearchMemoryResponse

0 commit comments

Comments
 (0)