Fixup error printing

block · Dec 20, 2024 · b2023fc · b2023fc
1 parent 7f6793e
commit b2023fc
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/crates/goose/src/providers/openai.rs b/crates/goose/src/providers/openai.rs
@@ -135,7 +135,13 @@ 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 {