From 2368782db48ed7ba8468e981ced4d9b518b98c4d Mon Sep 17 00:00:00 2001 From: shangkunwang Date: Fri, 24 Jul 2026 20:04:01 +0000 Subject: [PATCH] fix: save session.json even when auto agent client fail --- MaxKernel/auto_search/worker.py | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/MaxKernel/auto_search/worker.py b/MaxKernel/auto_search/worker.py index 7329de1..a5f9adc 100644 --- a/MaxKernel/auto_search/worker.py +++ b/MaxKernel/auto_search/worker.py @@ -133,14 +133,16 @@ async def _run_agent( ) await client.create_session(initial_state) - await client.run_async() try: - session_json_path = os.path.join(session_dir, "session.json") - with open(session_json_path, "w") as f: - json.dump(client.get_session_data(), f, indent=2) - logger.info(f"Saved session data to {session_json_path}") - except Exception as se: - logger.warning(f"Failed to save session data for {node_id}: {se}") + await client.run_async() + finally: + try: + session_json_path = os.path.join(session_dir, "session.json") + with open(session_json_path, "w") as f: + json.dump(client.get_session_data(), f, indent=2) + logger.info(f"Saved session data to {session_json_path}") + except Exception as se: + logger.warning(f"Failed to save session data for {node_id}: {se}") return client.get_state() def _process_results(