Skip to content

Commit b84657a

Browse files
yeesiancopybara-github
authored andcommitted
fix: broken import of VertexAiSessionService in AdkApp
PiperOrigin-RevId: 750214698
1 parent 5efcd6b commit b84657a

File tree

1 file changed

+3
-1
lines changed
  • vertexai/preview/reasoning_engines/templates

1 file changed

+3
-1
lines changed

vertexai/preview/reasoning_engines/templates/adk.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -408,7 +408,9 @@ def set_up(self):
408408
if session_service_builder:
409409
self._tmpl_attrs["session_service"] = session_service_builder()
410410
elif "GOOGLE_CLOUD_AGENT_ENGINE_ID" in os.environ:
411-
from google.adk.sessions import VertexAiSessionService
411+
from google.adk.sessions.vertex_ai_session_service import (
412+
VertexAiSessionService,
413+
)
412414

413415
self._tmpl_attrs["session_service"] = VertexAiSessionService(
414416
project=project,

0 commit comments

Comments
 (0)