Compare commits
No commits in common. "2a2179a1384d0e2484ca53f8c3ba9768df40de3e" and "244578cc0120edab2766053ab62a513408434c29" have entirely different histories.
2a2179a138
...
244578cc01
@ -16,7 +16,7 @@ spec:
|
|||||||
labels:
|
labels:
|
||||||
app: atlasbot
|
app: atlasbot
|
||||||
annotations:
|
annotations:
|
||||||
checksum/atlasbot-configmap: manual-atlasbot-88
|
checksum/atlasbot-configmap: manual-atlasbot-87
|
||||||
vault.hashicorp.com/agent-inject: "true"
|
vault.hashicorp.com/agent-inject: "true"
|
||||||
vault.hashicorp.com/role: "comms"
|
vault.hashicorp.com/role: "comms"
|
||||||
vault.hashicorp.com/agent-inject-secret-turn-secret: "kv/data/atlas/comms/turn-shared-secret"
|
vault.hashicorp.com/agent-inject-secret-turn-secret: "kv/data/atlas/comms/turn-shared-secret"
|
||||||
|
|||||||
@ -3605,33 +3605,10 @@ def _open_ended_multi(
|
|||||||
|
|
||||||
def _open_ended_total_steps(mode: str) -> int:
|
def _open_ended_total_steps(mode: str) -> int:
|
||||||
if mode == "fast":
|
if mode == "fast":
|
||||||
return 2
|
return 4
|
||||||
return 7
|
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(
|
def _open_ended_fast(
|
||||||
prompt: str,
|
prompt: str,
|
||||||
*,
|
*,
|
||||||
@ -3641,13 +3618,14 @@ def _open_ended_fast(
|
|||||||
history_lines: list[str],
|
history_lines: list[str],
|
||||||
state: ThoughtState | None = None,
|
state: ThoughtState | None = None,
|
||||||
) -> str:
|
) -> str:
|
||||||
model = _model_for_mode("fast")
|
return _open_ended_multi(
|
||||||
return _open_ended_fast_single(
|
|
||||||
prompt,
|
prompt,
|
||||||
fact_pack=fact_pack,
|
fact_pack=fact_pack,
|
||||||
|
fact_lines=fact_lines,
|
||||||
|
fact_meta=fact_meta,
|
||||||
history_lines=history_lines,
|
history_lines=history_lines,
|
||||||
|
mode="fast",
|
||||||
state=state,
|
state=state,
|
||||||
model=model,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user