fix: make function get_session_data from auto agent client public

Author: shangkunwang01

MaxKernel/auto_agent/agent_client/auto_agent_client.py

@@ -50,7 +50,7 @@ async def create_session(self) -> None:
       session_id=self.session_id,
     )
 
-  def _get_session_data(self) -> dict:
+  def get_session_data(self) -> dict:
     if not self.session:
       raise ValueError("Session has not been created yet.")
 

MaxKernel/auto_agent/agent_client/run_batch_agent_call.py

@@ -50,7 +50,7 @@ def get_optimized_kernel_path(session_file_path: str):
 
 def save_session_to_file(client: Any, file_path: str):
   """Fetches full session data and saves it to a local JSON file."""
-  session_data = client._get_session_data()
+  session_data = client.get_session_data()
   with open(file_path, "w") as f:
     json.dump(session_data, f, indent=2)
   logger.info(f"Saved session data to {file_path}")