More logging for command injection classifier model training (#7779)

This commit is contained in:
dorien-koelemeijer
2026-03-11 23:03:18 +10:00
committed by GitHub
parent 55023ca873
commit 6d6c6b3228
2 changed files with 10 additions and 2 deletions
+7 -2
View File
@@ -165,13 +165,18 @@ impl SecurityManager {
tool_request_id: tool_request.id.clone(),
});
}
} else {
} else if analysis_result.scanned {
let tool_call_json =
serde_json::to_string(&tool_call).unwrap_or_else(|_| "{}".to_string());
tracing::info!(
monotonic_counter.goose.prompt_injection_tool_call_passed = 1,
tool_name = %tool_call.name,
tool_request_id = %tool_request.id,
tool_call_json = %tool_call_json,
confidence = analysis_result.confidence,
explanation = %sanitized_explanation,
"Current tool call passed security analysis"
"Current tool call passed security analysis"
);
}
}
+3
View File
@@ -21,6 +21,7 @@ pub struct ScanResult {
pub is_malicious: bool,
pub confidence: f32,
pub explanation: String,
pub scanned: bool,
}
struct DetailedScanResult {
@@ -127,6 +128,7 @@ impl PromptInjectionScanner {
is_malicious: false,
confidence: 0.0,
explanation: "Tool call skipped: only shell commands are scanned".to_string(),
scanned: false,
});
}
@@ -180,6 +182,7 @@ impl PromptInjectionScanner {
is_malicious: final_confidence >= threshold,
confidence: final_confidence,
explanation: self.build_explanation(&final_result, threshold, &tool_content),
scanned: true,
})
}