Sync Space session state naming

README.md

@@ -65,7 +65,7 @@ These behaviors are intentional Space-only deltas:
   managed run directory under `RH_SPACE_RUNS_DIR`.
 - The runtime layout is always:
   `run_.../agent_workspace/` for agent-visible files and
-  `run_.../agent_trace/` for traces and `_session_state.json`.
+  `run_.../agent_trace/` for traces and `session_state_*.json`.
 - Uploaded images are saved under `agent_workspace/inputs/images/` and are also
   passed to the model as image inputs when supported.
 - Users can download files created or handled by the agent with the
@@ -133,7 +133,7 @@ Configure these as Hugging Face Space secrets before starting the app:
 └── run_YYYYMMDD_HHMMSS_<random>/
     ├── agent_workspace/
     │   └── inputs/images/        # user uploaded images, when present
-    └── agent_trace/              # trace JSONL and _session_state.json
+    └── agent_trace/              # trace JSONL and session_state_*.json
 ```
 
 The frontend exposes the chat UI and a single `Download workspace.zip` action

agent_base/react_agent.py

@@ -913,7 +913,7 @@ class MultiTurnReactAgent(BaseAgent):
             on_event=event_callback,
         )
         self.trace_path = trace_writer.path
-        self.session_state_path = resolve_session_state_path(trace_dir) if trace_dir else None
+        self.session_state_path = resolve_session_state_path(self.trace_path) if self.trace_path else None
         session_state = AgentSessionState(
             run_id=trace_writer.run_id,
             model_name=self.model,

agent_base/session_state.py

@@ -9,7 +9,7 @@ from agent_base.model_profiles import ModelProfile
 from agent_base.utils import safe_jsonable
 
 
-SESSION_STATE_FILENAME = "_session_state.json"
+SESSION_STATE_PREFIX = "session_state"
 
 
 @dataclass
@@ -74,8 +74,11 @@ class AgentSessionState:
         }
 
 
-def resolve_session_state_path(trace_dir: str | Path) -> Path:
-    return Path(trace_dir) / SESSION_STATE_FILENAME
+def resolve_session_state_path(trace_path: str | Path) -> Path:
+    trace = Path(trace_path)
+    stem = trace.stem
+    suffix = stem[len("trace_") :] if stem.startswith("trace_") else stem
+    return trace.with_name(f"{SESSION_STATE_PREFIX}_{suffix}.json")
 
 
 def persist_session_state(path: str | Path, state: AgentSessionState) -> None: