|
@@ -8,13 +8,16 @@ from agent.core.presets import register_preset
|
|
|
from .prompts import (
|
|
from .prompts import (
|
|
|
COMPARISON_WORKER_PROMPT,
|
|
COMPARISON_WORKER_PROMPT,
|
|
|
COMPOSE_WORKER_PROMPT,
|
|
COMPOSE_WORKER_PROMPT,
|
|
|
|
|
+ DECODE_RETRIEVAL_WORKER_PROMPT,
|
|
|
DIRECTION_WORKER_PROMPT,
|
|
DIRECTION_WORKER_PROMPT,
|
|
|
ELEMENT_SET_WORKER_PROMPT,
|
|
ELEMENT_SET_WORKER_PROMPT,
|
|
|
|
|
+ EXTERNAL_RETRIEVAL_WORKER_PROMPT,
|
|
|
|
|
+ KNOWLEDGE_RETRIEVAL_WORKER_PROMPT,
|
|
|
PARAGRAPH_WORKER_PROMPT,
|
|
PARAGRAPH_WORKER_PROMPT,
|
|
|
|
|
+ PATTERN_RETRIEVAL_WORKER_PROMPT,
|
|
|
PLANNER_PROMPT,
|
|
PLANNER_PROMPT,
|
|
|
PORTFOLIO_WORKER_PROMPT,
|
|
PORTFOLIO_WORKER_PROMPT,
|
|
|
RETRIEVAL_VALIDATOR_PROMPT,
|
|
RETRIEVAL_VALIDATOR_PROMPT,
|
|
|
- RETRIEVAL_WORKER_PROMPT,
|
|
|
|
|
ROOT_VALIDATOR_PROMPT,
|
|
ROOT_VALIDATOR_PROMPT,
|
|
|
ROOT_WORKER_PROMPT,
|
|
ROOT_WORKER_PROMPT,
|
|
|
SCRIPT_CANDIDATE_VALIDATOR_PROMPT,
|
|
SCRIPT_CANDIDATE_VALIDATOR_PROMPT,
|
|
@@ -41,7 +44,7 @@ _FORBIDDEN = [
|
|
|
]
|
|
]
|
|
|
|
|
|
|
|
|
|
|
|
|
-def _worker(tools: list[str], *, prompt: str = RETRIEVAL_WORKER_PROMPT) -> AgentPreset:
|
|
|
|
|
|
|
+def _worker(tools: list[str], *, prompt: str) -> AgentPreset:
|
|
|
return AgentPreset(
|
|
return AgentPreset(
|
|
|
role=AgentRole.WORKER,
|
|
role=AgentRole.WORKER,
|
|
|
allowed_tools=list(dict.fromkeys([*tools, "submit_attempt"])),
|
|
allowed_tools=list(dict.fromkeys([*tools, "submit_attempt"])),
|
|
@@ -87,19 +90,31 @@ def register_script_presets() -> None:
|
|
|
)
|
|
)
|
|
|
register_preset(
|
|
register_preset(
|
|
|
"script_pattern_retrieval_worker",
|
|
"script_pattern_retrieval_worker",
|
|
|
- _worker(["read_input_snapshot", "query_pattern_qa"]),
|
|
|
|
|
|
|
+ _worker(
|
|
|
|
|
+ ["read_input_snapshot", "query_pattern_qa"],
|
|
|
|
|
+ prompt=PATTERN_RETRIEVAL_WORKER_PROMPT,
|
|
|
|
|
+ ),
|
|
|
)
|
|
)
|
|
|
register_preset(
|
|
register_preset(
|
|
|
"script_decode_retrieval_worker",
|
|
"script_decode_retrieval_worker",
|
|
|
- _worker(["read_input_snapshot", "search_script_decode_case"]),
|
|
|
|
|
|
|
+ _worker(
|
|
|
|
|
+ ["read_input_snapshot", "search_script_decode_case"],
|
|
|
|
|
+ prompt=DECODE_RETRIEVAL_WORKER_PROMPT,
|
|
|
|
|
+ ),
|
|
|
)
|
|
)
|
|
|
register_preset(
|
|
register_preset(
|
|
|
"script_external_retrieval_worker",
|
|
"script_external_retrieval_worker",
|
|
|
- _worker(["read_input_snapshot", "external_search_case", "load_images"]),
|
|
|
|
|
|
|
+ _worker(
|
|
|
|
|
+ ["read_input_snapshot", "external_search_case", "load_images"],
|
|
|
|
|
+ prompt=EXTERNAL_RETRIEVAL_WORKER_PROMPT,
|
|
|
|
|
+ ),
|
|
|
)
|
|
)
|
|
|
register_preset(
|
|
register_preset(
|
|
|
"script_knowledge_retrieval_worker",
|
|
"script_knowledge_retrieval_worker",
|
|
|
- _worker(["read_input_snapshot", "search_knowledge"]),
|
|
|
|
|
|
|
+ _worker(
|
|
|
|
|
+ ["read_input_snapshot", "search_knowledge"],
|
|
|
|
|
+ prompt=KNOWLEDGE_RETRIEVAL_WORKER_PROMPT,
|
|
|
|
|
+ ),
|
|
|
)
|
|
)
|
|
|
register_preset(
|
|
register_preset(
|
|
|
"script_direction_worker",
|
|
"script_direction_worker",
|
|
@@ -108,12 +123,13 @@ def register_script_presets() -> None:
|
|
|
prompt=DIRECTION_WORKER_PROMPT,
|
|
prompt=DIRECTION_WORKER_PROMPT,
|
|
|
),
|
|
),
|
|
|
)
|
|
)
|
|
|
|
|
+ structure_writes = [
|
|
|
|
|
+ "read_attempt_workspace",
|
|
|
|
|
+ "create_script_paragraphs",
|
|
|
|
|
+ ]
|
|
|
paragraph_writes = [
|
|
paragraph_writes = [
|
|
|
"read_attempt_workspace",
|
|
"read_attempt_workspace",
|
|
|
- "create_script_paragraph",
|
|
|
|
|
"create_script_paragraphs",
|
|
"create_script_paragraphs",
|
|
|
- "append_paragraph_atoms",
|
|
|
|
|
- "delete_paragraph_atom",
|
|
|
|
|
"batch_update_script_paragraphs",
|
|
"batch_update_script_paragraphs",
|
|
|
]
|
|
]
|
|
|
element_writes = [
|
|
element_writes = [
|
|
@@ -132,7 +148,7 @@ def register_script_presets() -> None:
|
|
|
register_preset(
|
|
register_preset(
|
|
|
"script_structure_worker",
|
|
"script_structure_worker",
|
|
|
_worker(
|
|
_worker(
|
|
|
- [*common_candidate_reads, *paragraph_writes],
|
|
|
|
|
|
|
+ [*common_candidate_reads, *structure_writes],
|
|
|
prompt=STRUCTURE_WORKER_PROMPT,
|
|
prompt=STRUCTURE_WORKER_PROMPT,
|
|
|
),
|
|
),
|
|
|
)
|
|
)
|
|
@@ -161,10 +177,7 @@ def register_script_presets() -> None:
|
|
|
"script_compose_worker",
|
|
"script_compose_worker",
|
|
|
_worker(
|
|
_worker(
|
|
|
[
|
|
[
|
|
|
- "read_input_snapshot",
|
|
|
|
|
"read_active_frontier",
|
|
"read_active_frontier",
|
|
|
- *paragraph_writes,
|
|
|
|
|
- *element_writes,
|
|
|
|
|
"save_structured_script_candidate",
|
|
"save_structured_script_candidate",
|
|
|
],
|
|
],
|
|
|
prompt=COMPOSE_WORKER_PROMPT,
|
|
prompt=COMPOSE_WORKER_PROMPT,
|