Skip to content

Commit 8ec301b

Browse files
Naming fix + change to traces for datadog metrics
1 parent 734a9a2 commit 8ec301b

File tree

3 files changed

+9
-9
lines changed

3 files changed

+9
-9
lines changed

crates/goose/src/agents/tool_execution.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ impl Agent {
8888
// Log user decision if this was a security alert
8989
if let Some(finding_id) = get_security_finding_id_from_results(&request.id, inspection_results) {
9090
tracing::info!(
91-
counter.goose.prompt_injection_user_decisions = 1,
91+
monotonic_counter.goose.prompt_injection_user_decisions = 1,
9292
decision = ?confirmation.permission,
9393
finding_id = %finding_id,
9494
tool_request_id = %request.id,

crates/goose/src/security/mod.rs

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ impl SecurityManager {
6161
) -> Result<Vec<SecurityResult>> {
6262
if !self.is_prompt_injection_detection_enabled() {
6363
tracing::debug!(
64-
counter.goose.prompt_injection_scanner_disabled = 1,
64+
monotonic_counter.goose.prompt_injection_scanner_disabled = 1,
6565
"Security scanning disabled"
6666
);
6767
return Ok(vec![]);
@@ -74,7 +74,7 @@ impl SecurityManager {
7474
match PromptInjectionScanner::with_ml_detection() {
7575
Ok(s) => {
7676
tracing::info!(
77-
counter.goose.prompt_injection_scanner_enabled = 1,
77+
monotonic_counter.goose.prompt_injection_scanner_enabled = 1,
7878
"Security scanner initialized with ML-based detection"
7979
);
8080
s
@@ -90,7 +90,7 @@ impl SecurityManager {
9090
}
9191
} else {
9292
tracing::info!(
93-
counter.goose.prompt_injection_scanner_enabled = 1,
93+
monotonic_counter.goose.prompt_injection_scanner_enabled = 1,
9494
"Security scanner initialized with pattern-based detection only"
9595
);
9696
PromptInjectionScanner::new()
@@ -124,7 +124,7 @@ impl SecurityManager {
124124
serde_json::to_string(&tool_call).unwrap_or_else(|_| "{}".to_string());
125125

126126
tracing::warn!(
127-
counter.goose.prompt_injection_finding = 1,
127+
monotonic_counter.goose.prompt_injection_finding = 1,
128128
threat_type = "command_injection",
129129
above_threshold = above_threshold,
130130
tool_name = %tool_call.name,
@@ -164,7 +164,7 @@ impl SecurityManager {
164164
}
165165

166166
tracing::info!(
167-
counter.goose.prompt_injection_analysis_performed = 1,
167+
monotonic_counter.goose.prompt_injection_analysis_performed = 1,
168168
security_issues_found = results.len(),
169169
"Prompt injection detection: Security analysis complete"
170170
);

crates/goose/src/security/scanner.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -161,9 +161,9 @@ impl PromptInjectionScanner {
161161
tool_confidence = %tool_result.confidence,
162162
context_confidence = ?context_result.ml_confidence,
163163
final_confidence = %final_confidence,
164-
has_command_ml = tool_result.ml_confidence.is_some(),
165-
has_prompt_ml = context_result.ml_confidence.is_some(),
166-
has_patterns = tool_result.used_pattern_detection,
164+
used_command_ml = tool_result.ml_confidence.is_some(),
165+
used_prompt_ml = context_result.ml_confidence.is_some(),
166+
used_pattern_detection = tool_result.used_pattern_detection,
167167
threshold = %threshold,
168168
malicious = final_confidence >= threshold,
169169
"Security analysis complete"

0 commit comments

Comments
 (0)