diff --git a/backend/apps/agents/agents.py b/backend/apps/agents/agents.py index 5e5094a6..1dc1bc8a 100644 --- a/backend/apps/agents/agents.py +++ b/backend/apps/agents/agents.py @@ -306,7 +306,7 @@ async def compact_session(session_id: str): session = agent_manager.sessions.get(session_id) if not session: raise HTTPException(status_code=404, detail="session not found") - fired = agent_manager._maybe_compact(session, force=True) + fired = agent_manager.p_maybe_compact(session, force=True) if fired: from backend.apps.agents.core.ws_manager import ws_manager try: @@ -315,7 +315,7 @@ async def compact_session(session_id: str): "reason": "compacted", "compacted_through_msg_id": session.compacted_through_msg_id, }) - await agent_manager._emit_context_update( + await agent_manager.p_emit_context_update( session_id, session, input_tokens=_estimate_post_compact_input(session), @@ -346,7 +346,7 @@ async def clear_session(session_id: str): "status": session.status, "session": session.model_dump(mode="json"), }) - await agent_manager._emit_context_update( + await agent_manager.p_emit_context_update( session_id, session, input_tokens=0, diff --git a/backend/apps/agents/browser/browser_agent.py b/backend/apps/agents/browser/browser_agent.py index 779ea6e0..9452dd0a 100644 --- a/backend/apps/agents/browser/browser_agent.py +++ b/backend/apps/agents/browser/browser_agent.py @@ -848,7 +848,7 @@ async def run_browser_agent( except Exception: pass session.status = "completed" - agent_manager._sync_session_close(session) + agent_manager.p_sync_session_close(session) await ws_manager.send_to_session(session_id, "agent:status", { "session_id": session_id, "status": "completed", "session": session.model_dump(mode="json"), @@ -2149,7 +2149,7 @@ async def run_browser_agent( except Exception as e: logger.warning(f"[browser-agent {session_id}] keep_open persist failed: {e}") - agent_manager._sync_session_close(session) + agent_manager.p_sync_session_close(session) await ws_manager.send_to_session(session_id, "agent:status", { "session_id": session_id, "status": final_status, diff --git a/backend/main.py b/backend/main.py index 457c47d5..56b41b90 100644 --- a/backend/main.py +++ b/backend/main.py @@ -868,7 +868,7 @@ async def session_compact(session_id: str): session = agent_manager.sessions.get(session_id) if not session: return JSONResponse({"error": "session not found"}, status_code=404) - did_compact = agent_manager._maybe_compact(session, force=True) + did_compact = agent_manager.p_maybe_compact(session, force=True) if did_compact: session.needs_fresh_session = True await _ws.send_to_session(session_id, "agent:context_status", { diff --git a/backend/tests/test_browser_agent_loop.py b/backend/tests/test_browser_agent_loop.py index b9f1f662..5e7f135d 100644 --- a/backend/tests/test_browser_agent_loop.py +++ b/backend/tests/test_browser_agent_loop.py @@ -83,7 +83,7 @@ def _install(monkeypatch, primary, aux): monkeypatch.setattr(cred_mod, "get_anthropic_client_for_model", _client_for, raising=True) monkeypatch.setattr(BA, "load_builtin_permissions", lambda: {}, raising=True) - monkeypatch.setattr(am_mod.agent_manager, "_sync_session_close", lambda *a, **k: None, raising=False) + monkeypatch.setattr(am_mod.agent_manager, "p_sync_session_close", lambda *a, **k: None, raising=True) # fake WS: record browser commands, script results by action sent = []