use super::response_support::{
apply_contextual_guidance, bounded_result_payload, budget_hint, compact_response_payload,
effective_budget_for_tool, max_result_size_chars_for_tool, routing_hint_for_payload,
success_jsonrpc_response, text_payload_for_response,
};
use crate::AppState;
use crate::error::CodeLensError;
use crate::mutation_gate::{MutationGateAllowance, MutationGateFailure, is_verifier_source_tool};
use crate::protocol::{JsonRpcResponse, SuggestedNextCall, ToolCallResponse, ToolResponseMeta};
use crate::tool_defs::{ToolSurface, tool_definition};
use crate::tools;
use serde_json::json;
pub(crate) struct SuccessResponseInput<'a> {
pub name: &'a str,
pub payload: serde_json::Value,
pub meta: ToolResponseMeta,
pub state: &'a AppState,
pub surface: ToolSurface,
pub active_surface: &'a str,
pub arguments: &'a serde_json::Value,
pub logical_session_id: &'a str,
pub recent_tools: Vec<String>,
pub gate_allowance: Option<&'a MutationGateAllowance>,
pub compact: bool,
pub harness_phase: Option<&'a str>,
pub request_budget: usize,
pub start: std::time::Instant,
pub id: Option<serde_json::Value>,
pub doom_loop_count: usize,
pub doom_loop_rapid: bool,
}
pub(crate) fn build_success_response(input: SuccessResponseInput<'_>) -> JsonRpcResponse {
let SuccessResponseInput {
name,
payload,
meta,
state,
surface,
active_surface,
arguments,
logical_session_id,
recent_tools,
gate_allowance,
compact,
harness_phase,
request_budget,
start,
id,
doom_loop_count,
doom_loop_rapid,
} = input;
let elapsed_ms = start.elapsed().as_millis();
let effective_budget = effective_budget_for_tool(name, request_budget);
if is_verifier_source_tool(name) {
state.record_recent_preflight_from_payload(
name,
active_surface,
logical_session_id,
arguments,
&payload,
);
}
let had_caution = gate_allowance.map(|a| a.caution) == Some(true);
if had_caution {
state.metrics().record_mutation_with_caution();
}
let missing_preflight = had_caution;
let has_output_schema = tool_definition(name)
.and_then(|tool| tool.output_schema.as_ref())
.is_some();
let structured_content = has_output_schema.then(|| payload.clone());
let mut resp = ToolCallResponse::success(payload, meta);
let payload_estimate = serde_json::to_string(&resp.data)
.map(|s| tools::estimate_tokens(&s))
.unwrap_or(0);
let mut hint = budget_hint(name, payload_estimate, effective_budget);
if missing_preflight {
hint = format!("{hint} Tip: run verify_change_readiness before mutations for safer edits.");
}
if doom_loop_count >= 3 {
if doom_loop_rapid {
hint = format!(
"{hint} Rapid retry burst detected ({doom_loop_count}x in <10s). \
Use start_analysis_job for heavy analysis, or narrow scope with path/max_tokens."
);
} else {
hint = format!(
"{hint} Repeated low-level chain detected. Prefer verify_change_readiness, \
find_minimal_context_for_change, analyze_change_request for compressed context."
);
}
}
resp.token_estimate = Some(payload_estimate);
resp.budget_hint = Some(hint);
resp.elapsed_ms = Some(elapsed_ms as u64);
resp.routing_hint = Some(routing_hint_for_payload(&resp));
resp.reasoning_scaffold = tools::reasoning_scaffold::reasoning_scaffold_for(name);
let emitted_composite_guidance =
apply_contextual_guidance(&mut resp, name, &recent_tools, harness_phase, surface);
if doom_loop_count >= 3 {
if doom_loop_rapid {
resp.suggested_next_tools = Some(vec![
"start_analysis_job".to_owned(),
"find_minimal_context_for_change".to_owned(),
"get_ranked_context".to_owned(),
]);
} else {
resp.suggested_next_tools = Some(vec![
"verify_change_readiness".to_owned(),
"find_minimal_context_for_change".to_owned(),
"analyze_change_request".to_owned(),
]);
}
}
if let Some(ref next_tools) = resp.suggested_next_tools {
resp.suggestion_reasons = Some(tools::suggestion_reasons_for(next_tools, name));
let calls = build_suggested_next_calls(name, arguments, next_tools, resp.data.as_ref());
if !calls.is_empty() {
resp.suggested_next_calls = Some(calls);
}
}
if compact {
compact_response_payload(&mut resp);
}
let effort_offset = state.effort_level().compression_threshold_offset();
let text = text_payload_for_response(&resp, structured_content.as_ref());
let (text, structured_content, truncated) = bounded_result_payload(
text,
structured_content,
payload_estimate,
effective_budget,
effort_offset,
);
state.metrics().record_call_with_tokens_for_session(
name,
elapsed_ms as u64,
true,
payload_estimate,
active_surface,
truncated,
harness_phase,
Some(logical_session_id),
);
if emitted_composite_guidance
&& !matches!(name, "get_tool_metrics" | "set_profile" | "set_preset")
{
state.metrics().record_composite_guidance_emitted(name);
}
let max_result_size = max_result_size_chars_for_tool(name, truncated);
success_jsonrpc_response(id, text, structured_content, Some(max_result_size))
}
#[allow(clippy::too_many_arguments)]
pub(crate) fn build_error_response(
name: &str,
error: CodeLensError,
gate_failure: Option<MutationGateFailure>,
active_surface: &str,
logical_session_id: &str,
state: &AppState,
start: std::time::Instant,
id: Option<serde_json::Value>,
) -> JsonRpcResponse {
let elapsed_ms = start.elapsed().as_millis();
state.metrics().record_call_with_tokens_for_session(
name,
elapsed_ms as u64,
false,
0,
active_surface,
false,
None,
Some(logical_session_id),
);
if error.is_protocol_error() {
return JsonRpcResponse::error(id, error.jsonrpc_code(), error.to_string());
}
let recovery_hint = error.recovery_hint();
let mut resp = ToolCallResponse::error(error.to_string());
resp.recovery_hint = recovery_hint;
if let Some(failure) = gate_failure {
let analysis_hint = failure
.analysis_id
.as_ref()
.map(|analysis_id| format!(" Last related analysis_id: `{analysis_id}`."))
.unwrap_or_default();
resp.error = Some(format!(
"[{:?}] {}{}",
failure.kind, failure.message, analysis_hint
));
resp.suggested_next_tools = Some(failure.suggested_next_tools);
resp.budget_hint = Some(failure.budget_hint);
}
let text = text_payload_for_response(&resp, None);
JsonRpcResponse::result(
id,
json!({
"content": [{ "type": "text", "text": text }],
"isError": true
}),
)
}
fn build_suggested_next_calls(
current_tool: &str,
current_args: &serde_json::Value,
next_tools: &[String],
payload: Option<&serde_json::Value>,
) -> Vec<SuggestedNextCall> {
let analysis_id = payload
.and_then(|p| p.get("analysis_id"))
.and_then(|v| v.as_str());
let task = current_args.get("task").and_then(|v| v.as_str());
let changed_files = current_args.get("changed_files").cloned();
let path = current_args.get("path").and_then(|v| v.as_str());
let file_path = current_args
.get("file_path")
.and_then(|v| v.as_str())
.or_else(|| current_args.get("relative_path").and_then(|v| v.as_str()));
let symbol = current_args
.get("symbol")
.and_then(|v| v.as_str())
.or_else(|| current_args.get("symbol_name").and_then(|v| v.as_str()))
.or_else(|| current_args.get("name").and_then(|v| v.as_str()))
.or_else(|| current_args.get("function_name").and_then(|v| v.as_str()))
.or_else(|| current_args.get("entrypoint").and_then(|v| v.as_str()));
let new_name = current_args.get("new_name").and_then(|v| v.as_str());
let target_path = file_path.or(path);
let single_changed_file = current_args
.get("changed_files")
.and_then(|v| v.as_array())
.and_then(|items| {
if items.len() == 1 {
items.first().and_then(|value| value.as_str())
} else {
None
}
});
let diagnostic_path = target_path.or(single_changed_file);
let mut calls = Vec::new();
for next in next_tools.iter().take(3) {
let call = match (current_tool, next.as_str()) {
("explore_codebase", "review_architecture") => {
target_path.map(|p| SuggestedNextCall {
tool: "review_architecture".to_owned(),
arguments: json!({ "path": p }),
reason: "Escalate the same scoped exploration into an architecture review instead of starting over.".to_owned(),
})
}
("explore_codebase", "analyze_change_impact" | "impact_report") => {
target_path.map(|p| SuggestedNextCall {
tool: next.clone(),
arguments: json!({ "path": p }),
reason: "Carry the same scoped path into an impact pass without re-entering the target.".to_owned(),
})
}
("trace_request_path", "plan_safe_refactor") => symbol.map(|sym| SuggestedNextCall {
tool: "plan_safe_refactor".to_owned(),
arguments: json!({ "symbol": sym }),
reason: "Use the traced entrypoint as the symbol anchor for refactor planning.".to_owned(),
}),
("review_architecture", "plan_safe_refactor") => {
target_path.map(|p| SuggestedNextCall {
tool: "plan_safe_refactor".to_owned(),
arguments: json!({ "path": p }),
reason: "Reuse the reviewed scope as the refactor planning target.".to_owned(),
})
}
("review_architecture", "analyze_change_impact" | "impact_report") => {
target_path.map(|p| SuggestedNextCall {
tool: next.clone(),
arguments: json!({ "path": p }),
reason: "Take the same architecture scope into an impact report instead of re-specifying it.".to_owned(),
})
}
("plan_safe_refactor", "trace_request_path") => symbol.map(|sym| SuggestedNextCall {
tool: "trace_request_path".to_owned(),
arguments: json!({ "symbol": sym }),
reason: "Trace the same symbol before editing to confirm the execution path.".to_owned(),
}),
("plan_safe_refactor", "analyze_change_impact" | "impact_report") => {
target_path.map(|p| SuggestedNextCall {
tool: next.clone(),
arguments: json!({ "path": p }),
reason: "Assess blast radius for the same refactor scope without restating the target.".to_owned(),
})
}
("verify_change_readiness", "get_analysis_section") => analysis_id.map(|aid| {
SuggestedNextCall {
tool: "get_analysis_section".to_owned(),
arguments: json!({
"analysis_id": aid,
"section": "verifier_diagnostics",
}),
reason: "Expand the diagnostics section of this readiness report instead of re-running verifier work.".to_owned(),
}
}),
("analyze_change_request", "verify_change_readiness") => task.map(|t| {
let mut args = json!({ "task": t });
if let Some(cf) = changed_files.clone() {
args["changed_files"] = cf;
}
SuggestedNextCall {
tool: "verify_change_readiness".to_owned(),
arguments: args,
reason: "Gate the same task through the verifier before any edit.".to_owned(),
}
}),
("impact_report", "diff_aware_references") => changed_files.clone().map(|cf| {
SuggestedNextCall {
tool: "diff_aware_references".to_owned(),
arguments: json!({ "changed_files": cf }),
reason: "Drill into classified references for the same file set without re-running impact.".to_owned(),
}
}),
("impact_report", "verify_change_readiness") => task.map(|t| {
let mut args = json!({ "task": t });
if let Some(cf) = changed_files.clone() {
args["changed_files"] = cf;
}
SuggestedNextCall {
tool: "verify_change_readiness".to_owned(),
arguments: args,
reason: "Promote the impact evidence into a readiness verdict for the same scope.".to_owned(),
}
}),
("review_changes", "impact_report") => {
let mut args = json!({});
if let Some(cf) = changed_files.clone() {
args["changed_files"] = cf;
Some(SuggestedNextCall {
tool: "impact_report".to_owned(),
arguments: args,
reason: "Re-run the broader impact view over the same changed files.".to_owned(),
})
} else if let Some(p) = target_path {
args["path"] = json!(p);
Some(SuggestedNextCall {
tool: "impact_report".to_owned(),
arguments: args,
reason: "Expand this review into a broader impact report for the same scope.".to_owned(),
})
} else {
None
}
}
("review_changes", "diagnose_issues") => diagnostic_path.map(|p| SuggestedNextCall {
tool: "diagnose_issues".to_owned(),
arguments: json!({ "path": p }),
reason: "Drill into diagnostics for the same reviewed file while the scope is still warm.".to_owned(),
}),
("diagnose_issues", "review_changes") => diagnostic_path.map(|p| SuggestedNextCall {
tool: "review_changes".to_owned(),
arguments: json!({ "path": p }),
reason: "Promote the same file-level issue into a broader change review.".to_owned(),
}),
("diagnose_issues", "find_symbol") => symbol.map(|sym| {
let mut args = json!({ "name": sym });
if let Some(p) = target_path {
args["file_path"] = json!(p);
}
SuggestedNextCall {
tool: "find_symbol".to_owned(),
arguments: args,
reason: "Jump directly to the implicated symbol instead of searching from scratch.".to_owned(),
}
}),
("safe_rename_report", "rename_symbol") => {
match (file_path, symbol) {
(Some(fp), Some(sym)) => {
let mut args = json!({
"file_path": fp,
"symbol_name": sym,
"dry_run": true,
});
if let Some(nn) = new_name {
args["new_name"] = json!(nn);
}
Some(SuggestedNextCall {
tool: "rename_symbol".to_owned(),
arguments: args,
reason: "Execute the rename previewed here; keep `dry_run=true` until diagnostics pass.".to_owned(),
})
}
_ => None,
}
}
(_, "get_analysis_section") => analysis_id.map(|aid| SuggestedNextCall {
tool: "get_analysis_section".to_owned(),
arguments: json!({ "analysis_id": aid, "section": "summary" }),
reason: "Pull the summary section of this handle instead of re-running the workflow."
.to_owned(),
}),
_ => None,
};
if let Some(c) = call {
calls.push(c);
}
}
calls
}