fix: resolve session continuity issues from CodeRabbit review

- Fix cli_project_sessions logic to check both session_id and auto_save_name
- Add hasattr(self, 'args') guard in main.py session continuity hook
- Remove auto-save re-enabling fallback in project_sessions.py
- Add session continuity support to profiled execution paths

Fixes issues identified in PR review to ensure --no-save and --continue
work correctly together without silently re-enabling persistence.

Co-authored-by: Mervin Praison <MervinPraison@users.noreply.github.com>

Author: praisonai-triage-agent[bot]

src/praisonai/praisonai/cli/commands/run.py

@@ -256,7 +256,7 @@ class Args:
             args = Args()
             args.auto_save = auto_save_name
             args.resume_session = session_id
-            args.cli_project_sessions = bool(auto_save_name)
+            args.cli_project_sessions = bool(session_id or auto_save_name)
 
             praison.args = args
 
@@ -376,7 +376,7 @@ def _run_prompt(
 
             agent = Agent(**agent_config)
             if session_id or auto_save_name:
-                apply_cli_session_continuity(agent, session_id or auto_save_name)
+                apply_cli_session_continuity(agent, session_id or auto_save_name, auto_save=auto_save_name)
             result = agent.start(prompt)
 
             output.emit_result(
@@ -413,7 +413,7 @@ class Args:
         args.auto_save = auto_save_name
         args.history = None
         args.resume_session = session_id
-        args.cli_project_sessions = bool(auto_save_name)
+        args.cli_project_sessions = bool(session_id or auto_save_name)
         args.include_rules = None if no_rules else "auto"
         args.no_rules = no_rules
         args.workflow = None
@@ -499,6 +499,22 @@ def _run_from_file_profiled(
     )
     if model:
         praison.config_list[0]['model'] = model
+    
+    # Apply session continuity if requested
+    session_id, auto_save_name = resolve_session_params(
+        continue_session, session, fork, no_save
+    )
+    if session_id or auto_save_name:
+        class Args:
+            pass
+        
+        args = Args()
+        args.auto_save = auto_save_name
+        args.resume_session = session_id
+        args.cli_project_sessions = bool(session_id or auto_save_name)
+        
+        praison.args = args
+    
     profiler.mark_init_end()
 
     # Execution phase
@@ -560,7 +576,21 @@ def _run_prompt_profiled(
     if model:
         agent_config["llm"] = model
 
+    # Apply session continuity if requested
+    session_id, auto_save_name = resolve_session_params(
+        continue_session, session, fork, no_save
+    )
+    if session_id or auto_save_name:
+        from ..state.project_sessions import build_cli_memory_config, apply_cli_session_continuity
+        
+        memory_cfg = build_cli_memory_config(session_id, auto_save_name)
+        if memory_cfg is not None:
+            agent_config["memory"] = memory_cfg
+    
     agent = Agent(**agent_config)
+    if session_id or auto_save_name:
+        apply_cli_session_continuity(agent, session_id or auto_save_name, auto_save=auto_save_name)
+    
     profiler.mark_init_end()
 
     # Execution phase

src/praisonai/praisonai/cli/main.py

@@ -4818,11 +4818,12 @@ def level_based_approve(function_name, arguments, risk_level):
 
             agent = PraisonAgent(**agent_config)
 
-            if getattr(self.args, 'cli_project_sessions', False):
+            if hasattr(self, 'args') and getattr(self.args, 'cli_project_sessions', False):
                 session_id = getattr(self.args, 'resume_session', None) or getattr(self.args, 'auto_save', None)
+                auto_save = getattr(self.args, 'auto_save', None)
                 if session_id:
                     from .state.project_sessions import apply_cli_session_continuity
-                    apply_cli_session_continuity(agent, session_id)
+                    apply_cli_session_continuity(agent, session_id, auto_save=auto_save)
 
             # AutoRag - Automatic RAG retrieval decision
             if hasattr(self, 'args') and getattr(self.args, 'auto_rag', False):

src/praisonai/praisonai/cli/state/project_sessions.py

@@ -112,19 +112,18 @@ def build_cli_memory_config(
     from praisonaiagents import MemoryConfig
 
     sid = session_id or auto_save
-    save_name = auto_save or sid
-    return MemoryConfig(session_id=sid, auto_save=save_name, history=True)
+    return MemoryConfig(session_id=sid, auto_save=auto_save, history=True)
 
 
-def apply_cli_session_continuity(agent, session_id: str, project_path: Optional[str] = None) -> None:
+def apply_cli_session_continuity(agent, session_id: str, project_path: Optional[str] = None, auto_save: Optional[str] = None) -> None:
     """Wire an agent to the project session store and restore prior history."""
     store = get_project_session_store(project_path)
     agent._session_store = store
     agent._session_id = session_id
     agent._history_enabled = True
     agent._history_session_id = session_id
-    if not getattr(agent, "auto_save", None):
-        agent.auto_save = session_id
+    if auto_save is not None:
+        agent.auto_save = auto_save
 
     history = store.get_chat_history(session_id)
     if history and not agent.chat_history: