Skip to content

Commit 6ef0e20

Browse files
Tongzhou-Jiangcopybara-github
authored andcommitted
fix: investigate save artifact in _init_session
PiperOrigin-RevId: 834913008
1 parent bc26160 commit 6ef0e20

File tree

1 file changed

+8
-0
lines changed
  • vertexai/agent_engines/templates

1 file changed

+8
-0
lines changed

vertexai/agent_engines/templates/adk.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -607,6 +607,9 @@ async def _init_session(
607607
):
608608
"""Initializes the session, and returns the session id."""
609609
from google.adk.events.event import Event
610+
from google.cloud.aiplatform import base
611+
612+
_LOGGER = base.Logger(__name__)
610613

611614
session_state = None
612615
if request.authorizations:
@@ -633,6 +636,11 @@ async def _init_session(
633636
artifact.versions, key=lambda x: x["version"]
634637
):
635638
version_data = _ArtifactVersion(**version_data)
639+
# _LOGGER.info(
640+
# "Artifact '%s' type: %s",
641+
# version_data.data,
642+
# type(version_data.data)
643+
# )
636644
saved_version = await artifact_service.save_artifact(
637645
app_name=app.name if app else self._tmpl_attrs.get("app_name"),
638646
user_id=request.user_id,

0 commit comments

Comments
 (0)