Skip to content

Commit a685a95

Browse files
google-genai-botcopybara-github
authored andcommitted
feat: Add import session API in the UI side
PiperOrigin-RevId: 771138494
1 parent d129fd6 commit a685a95

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/google/adk/cli/fast_api.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -412,12 +412,19 @@ async def create_session(
412412
app_name: str,
413413
user_id: str,
414414
state: Optional[dict[str, Any]] = None,
415+
events: Optional[list[Event]] = None,
415416
) -> Session:
416417
logger.info("New session created")
417-
return await session_service.create_session(
418+
session = await session_service.create_session(
418419
app_name=app_name, user_id=user_id, state=state
419420
)
420421

422+
if events:
423+
for event in events:
424+
await session_service.append_event(session=session, event=event)
425+
426+
return session
427+
421428
def _get_eval_set_file_path(app_name, agents_dir, eval_set_id) -> str:
422429
return os.path.join(
423430
agents_dir,

0 commit comments

Comments
 (0)