Feat/resume

wake_ai/cli.py

@@ -301,6 +301,7 @@ def main(ctx: click.Context, working_dir: str | None, model: str, resume: bool,
         ctx.obj["model"] = model
         ctx.obj["working_dir"] = working_dir
         ctx.obj["execution_dir"] = execution_dir
+        ctx.obj["resume"] = resume
         ctx.obj["cleanup_working_dir"] = not no_cleanup
         ctx.obj["show_progress"] = not no_progress
         ctx.obj["console"] = console  # Pass console for coordinated output

wake_ai/core/claude.py

@@ -520,6 +520,7 @@ def query(
         prompt: str,
         max_turns: Optional[int] = None,
         continue_session: bool = False,
+        resume_session: Optional[str] = None,
     ) -> ClaudeCodeResponse:
         """Execute a query with Claude Code (synchronous wrapper).
 
@@ -534,15 +535,23 @@ def query(
 
         # Note: Session resumption logic is handled in the async version
 
+        if resume_session and continue_session:
+            raise ValueError(
+                "resume_session and continue_session cannot be used together"
+            )
+
         if continue_session:
             logger.debug(f"Continuing session: {continue_session}")
 
         # Execute async version using asyncio event loop
         response = asyncio.run(
             self.query_async(
-                prompt=prompt, max_turns=max_turns, continue_session=continue_session
+                prompt=prompt, max_turns=max_turns, continue_session=continue_session, resume_session=resume_session
             )
         )
+        # update only it could reach here.
+        # it can update last session, however other handiling needs to be done.
+        self.last_session_id = response.session_id
 
         return response