From c1e94d56c857ff127650ddba67edd386b9f49400 Mon Sep 17 00:00:00 2001 From: Brad Stein Date: Wed, 28 Jan 2026 01:52:23 -0300 Subject: [PATCH] atlasbot: simplify fast path --- services/comms/scripts/atlasbot/bot.py | 32 ++++++++++++++++++++++---- 1 file changed, 27 insertions(+), 5 deletions(-) diff --git a/services/comms/scripts/atlasbot/bot.py b/services/comms/scripts/atlasbot/bot.py index 50fed4b..d0d46ef 100644 --- a/services/comms/scripts/atlasbot/bot.py +++ b/services/comms/scripts/atlasbot/bot.py @@ -3605,10 +3605,33 @@ def _open_ended_multi( def _open_ended_total_steps(mode: str) -> int: if mode == "fast": - return 4 + return 2 return 7 +def _open_ended_fast_single( + prompt: str, + *, + fact_pack: str, + history_lines: list[str], + state: ThoughtState | None = None, + model: str, +) -> str: + if state: + state.update("drafting", step=2, note="summarizing") + context = fact_pack + reply = _ollama_call( + ("atlasbot_fast", "atlasbot_fast"), + prompt, + context=context, + use_history=False, + model=model, + ) + if state: + state.update("done", step=_open_ended_total_steps("fast")) + return _ensure_scores(reply) + + def _open_ended_fast( prompt: str, *, @@ -3618,14 +3641,13 @@ def _open_ended_fast( history_lines: list[str], state: ThoughtState | None = None, ) -> str: - return _open_ended_multi( + model = _model_for_mode("fast") + return _open_ended_fast_single( prompt, fact_pack=fact_pack, - fact_lines=fact_lines, - fact_meta=fact_meta, history_lines=history_lines, - mode="fast", state=state, + model=model, )