88 lines
3.2 KiB
Rust
88 lines
3.2 KiB
Rust
//! extract sub-Worker への入力テキスト組み立て。
|
||
//!
|
||
//! `crates/pod/src/pod.rs::build_summary_prompt` と同じ方針で
|
||
//! Item 列を flat な行に落とす（reasoning は省く、tool call は名前のみ、
|
||
//! tool result は summary のみ）。conversation 全体を Markdown の単一
|
||
//! セクションとして渡し、抽出指示は system prompt 側に寄せる。
|
||
|
||
use llm_worker::Item;
|
||
|
||
/// 与えられた `items` を extract sub-Worker の最初の user 入力に整形する。
|
||
pub fn build_extract_input(items: &[Item]) -> String {
|
||
let mut out = String::new();
|
||
out.push_str(
|
||
"Extract activity logs from the conversation slice below. \
|
||
Follow the system prompt's schema strictly and call `write_extracted` once.\n\n",
|
||
);
|
||
out.push_str("## Conversation slice\n");
|
||
out.push_str(&render_items(items));
|
||
out.push_str("\n\nWhen you are done, call `write_extracted` and end the turn.");
|
||
out
|
||
}
|
||
|
||
fn render_items(items: &[Item]) -> String {
|
||
let mut lines: Vec<String> = Vec::new();
|
||
for item in items {
|
||
match item {
|
||
Item::Message { role, content, .. } => {
|
||
let role_label = match role {
|
||
llm_worker::Role::User => "User",
|
||
llm_worker::Role::Assistant => "Assistant",
|
||
llm_worker::Role::System => "System",
|
||
};
|
||
let text: String = content
|
||
.iter()
|
||
.map(|p| p.as_text())
|
||
.collect::<Vec<_>>()
|
||
.join("");
|
||
lines.push(format!("[{role_label}] {text}"));
|
||
}
|
||
Item::ToolCall { name, .. } => {
|
||
lines.push(format!("[ToolCall] {name}"));
|
||
}
|
||
Item::ToolResult { summary, .. } => {
|
||
lines.push(format!("[ToolResult] {summary}"));
|
||
}
|
||
Item::Reasoning { .. } => {}
|
||
}
|
||
}
|
||
lines.join("\n\n")
|
||
}
|
||
|
||
#[cfg(test)]
|
||
mod tests {
|
||
use super::*;
|
||
|
||
#[test]
|
||
fn renders_user_assistant_pair_and_tool_calls() {
|
||
let items = vec![
|
||
Item::user_message("hello"),
|
||
Item::assistant_message("hi"),
|
||
Item::tool_call("c1", "read_file", "{}"),
|
||
Item::tool_result("c1", "ok"),
|
||
Item::reasoning("internal scratch — should be skipped"),
|
||
];
|
||
let s = build_extract_input(&items);
|
||
assert!(s.contains("[User] hello"));
|
||
assert!(s.contains("[Assistant] hi"));
|
||
assert!(s.contains("[ToolCall] read_file"));
|
||
assert!(s.contains("[ToolResult] ok"));
|
||
assert!(!s.contains("scratch"));
|
||
}
|
||
|
||
#[test]
|
||
fn tool_result_renders_summary_but_not_content() {
|
||
let huge_content = "raw-content-should-never-enter-extract-input".repeat(10_000);
|
||
let items = vec![Item::tool_result_with_content(
|
||
"c1",
|
||
"short summary kept for extraction",
|
||
huge_content.clone(),
|
||
)];
|
||
|
||
let s = build_extract_input(&items);
|
||
assert!(s.contains("[ToolResult] short summary kept for extraction"));
|
||
assert!(!s.contains("raw-content-should-never-enter-extract-input"));
|
||
assert!(!s.contains(&huge_content));
|
||
}
|
||
}
|