File tree Expand file tree Collapse file tree
tests/unittests/cli/utils Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -117,9 +117,9 @@ async def run_interactively(
117117 )
118118
119119 # Background task for periodic session saving
120- save_task : Optional [asyncio .Task [None ]] = None
120+ save_task : Optional [asyncio .Task [None ]] = None
121121
122- async def _periodic_save_session () -> None :
122+ async def _periodic_save_session () -> None :
123123 """Periodically save the session to disk every interval seconds."""
124124 nonlocal session , save_task
125125 try :
Original file line number Diff line number Diff line change @@ -683,7 +683,7 @@ def cli_run( # type: ignore[misc]
683683 artifact_service_uri : Optional [str ] = None ,
684684 memory_service_uri : Optional [str ] = None ,
685685 use_local_storage : bool = True ,
686- ) -> None :
686+ ) -> None :
687687 """Runs an interactive CLI for a certain agent.
688688
689689 AGENT: The path to the agent source code folder.
Original file line number Diff line number Diff line change @@ -283,15 +283,15 @@ async def test_run_cli_save_session(
283283 if session_file .exists ():
284284 session_file .unlink ()
285285
286- await cli .run_cli (
287- agent_parent_dir = str (parent_dir ),
288- agent_folder_name = folder_name ,
289- input_file = None ,
290- saved_session_file = None ,
291- save_session = False ,
292- save_session_on_runtime = False ,
293- interval = 60 ,
294- )
286+ await cli .run_cli (
287+ agent_parent_dir = str (parent_dir ),
288+ agent_folder_name = folder_name ,
289+ input_file = None ,
290+ saved_session_file = None ,
291+ save_session = True ,
292+ save_session_on_runtime = False ,
293+ interval = 60 ,
294+ )
295295
296296 assert session_file .exists ()
297297 data = json .loads (session_file .read_text ())
You can’t perform that action at this time.
0 commit comments