diff --git a/src/mistralai/extra/observability/redaction_policies.py b/src/mistralai/extra/observability/redaction_policies.py index d99d2a3f..8017e2cb 100644 --- a/src/mistralai/extra/observability/redaction_policies.py +++ b/src/mistralai/extra/observability/redaction_policies.py @@ -85,6 +85,7 @@ def redact_status_description(self, description: str | None) -> str | None: "password", "payload", "prompt", + "result", "secret", "set_cookie", "token", diff --git a/src/mistralai/extra/tests/test_redaction_policies.py b/src/mistralai/extra/tests/test_redaction_policies.py index 5d0c279b..d8677923 100644 --- a/src/mistralai/extra/tests/test_redaction_policies.py +++ b/src/mistralai/extra/tests/test_redaction_policies.py @@ -44,6 +44,17 @@ def test_fragment_match_redacted(self, attribute_policy: AttributeRedactionPolic ) assert out["custom.prompt.text"] == DEFAULT_REDACTED_VALUE + def test_workflow_arguments_and_result_redacted( + self, attribute_policy: AttributeRedactionPolicy + ): + out = attribute_policy.redact_attributes( + {"wf.arguments": "x", "wf.result": "x"} + ) + assert out == { + "wf.arguments": DEFAULT_REDACTED_VALUE, + "wf.result": DEFAULT_REDACTED_VALUE, + } + def test_token_pattern_on_kept_string( self, attribute_policy: AttributeRedactionPolicy ):