feat: Add import session API in the fast API

PiperOrigin-RevId: 771279971
This commit is contained in:
Google Team Member
2025-06-13 17:26:08 -07:00
committed by Copybara-Service
parent d129fd636b
commit 233fd20243
+8 -1
View File
@@ -412,12 +412,19 @@ def get_fast_api_app(
app_name: str, app_name: str,
user_id: str, user_id: str,
state: Optional[dict[str, Any]] = None, state: Optional[dict[str, Any]] = None,
events: Optional[list[Event]] = None,
) -> Session: ) -> Session:
logger.info("New session created") logger.info("New session created")
return await session_service.create_session( session = await session_service.create_session(
app_name=app_name, user_id=user_id, state=state app_name=app_name, user_id=user_id, state=state
) )
if events:
for event in events:
await session_service.append_event(session=session, event=event)
return session
def _get_eval_set_file_path(app_name, agents_dir, eval_set_id) -> str: def _get_eval_set_file_path(app_name, agents_dir, eval_set_id) -> str:
return os.path.join( return os.path.join(
agents_dir, agents_dir,