Fixup error printing

block · Dec 20, 2024 · eb54b6b · eb54b6b
1 parent 7f6793e
commit eb54b6b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/crates/goose/src/providers/openai.rs b/crates/goose/src/providers/openai.rs
@@ -135,7 +135,11 @@ impl Moderation for OpenAiProvider {
             let category_scores = response_json["results"][0]["category_scores"].clone();
             return Ok(ModerationResult::new(
                 flagged,
-                Some(categories.keys().cloned().collect()),
+                Some(categories
+                    .iter()
+                    .filter(|(_, value)| value.as_bool().unwrap_or(false))
+                    .map(|(key, _)| key.to_string())
+                    .collect()),
                 Some(category_scores),
             ));
         } else {