feat: reintroduce FPE handling

devenv.lock

@@ -3,10 +3,10 @@
     "devenv": {
       "locked": {
         "dir": "src/modules",
-        "lastModified": 1769011019,
+        "lastModified": 1771419682,
         "owner": "cachix",
         "repo": "devenv",
-        "rev": "e4c0fb64efda6ffa563a97a98c8c75c336f6bff1",
+        "rev": "f77fc4de35c184d9ef9a32d5d7e9033351bcdfdc",
         "type": "github"
       },
       "original": {
@@ -55,10 +55,10 @@
         ]
       },
       "locked": {
-        "lastModified": 1768935149,
+        "lastModified": 1770726378,
         "owner": "cachix",
         "repo": "git-hooks.nix",
-        "rev": "18cbede9ff6da05b911c5c4802a397c2686ac8fa",
+        "rev": "5eaaedde414f6eb1aea8b8525c466dc37bba95ae",
         "type": "github"
       },
       "original": {
@@ -88,11 +88,14 @@
       }
     },
     "nixpkgs": {
+      "inputs": {
+        "nixpkgs-src": "nixpkgs-src"
+      },
       "locked": {
-        "lastModified": 1767052823,
+        "lastModified": 1770434727,
         "owner": "cachix",
         "repo": "devenv-nixpkgs",
-        "rev": "538a5124359f0b3d466e1160378c87887e3b51a4",
+        "rev": "8430f16a39c27bdeef236f1eeb56f0b51b33d348",
         "type": "github"
       },
       "original": {
@@ -110,10 +113,10 @@
         ]
       },
       "locked": {
-        "lastModified": 1765052656,
+        "lastModified": 1770193939,
         "owner": "cachix",
         "repo": "nixpkgs-python",
-        "rev": "04b27dbad2e004cb237db202f21154eea3c4f89f",
+        "rev": "0630618bfe33895453257fb606af75aa71247393",
         "type": "github"
       },
       "original": {
@@ -122,6 +125,23 @@
         "type": "github"
       }
     },
+    "nixpkgs-src": {
+      "flake": false,
+      "locked": {
+        "lastModified": 1769922788,
+        "narHash": "sha256-H3AfG4ObMDTkTJYkd8cz1/RbY9LatN5Mk4UF48VuSXc=",
+        "owner": "NixOS",
+        "repo": "nixpkgs",
+        "rev": "207d15f1a6603226e1e223dc79ac29c7846da32e",
+        "type": "github"
+      },
+      "original": {
+        "owner": "NixOS",
+        "ref": "nixpkgs-unstable",
+        "repo": "nixpkgs",
+        "type": "github"
+      }
+    },
     "root": {
       "inputs": {
         "devenv": "devenv",

devenv.nix

@@ -4,11 +4,8 @@
   config,
   inputs,
   ...
-}: {
-  packages = with pkgs; [
-    git
-  ];
-  starship.enable = true;
+}:
+{
   languages.python = {
     enable = true;
     version = "3.12";

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "crowdstrike-aidr-openai"
-version = "0.1.1"
+version = "0.2.0"
 description = "A wrapper around the OpenAI Python library that wraps the Responses API with CrowdStrike AIDR"
 readme = "README.md"
 license = "MIT"
@@ -10,7 +10,7 @@ authors = [
 requires-python = ">=3.12"
 dependencies = [
     "anyio ~=4.12.0",
-    "crowdstrike-aidr ~=0.4.2",
+    "crowdstrike-aidr ~=0.5.0",
     "httpx ~=0.28.1",
     "openai ~=2.8.1",
     "pydantic-core ~=2.41.5",

src/crowdstrike_aidr_openai/resources/responses/responses.py

@@ -986,15 +986,14 @@ def create(
             if c.type == "output_text"
         ]
 
-        # TODO: reintroduce.
         # FPE unredact.
-        # if guard_input_response.result.fpe_context is not None:
-        #     redact_response = self._client.redact_client.unredact(
-        #         output_messages,
-        #         fpe_context=guard_input_response.result.fpe_context,
-        #     )
-        #     assert redact_response.result is not None
-        #     output_messages = list_message_adapter.validate_python(redact_response.result.data)
+        if guard_input_response.result.fpe_context is not None:
+            redact_response = self._client.ai_guard_client.unredact(
+                redacted_data=output_messages,
+                fpe_context=guard_input_response.result.fpe_context,
+            )
+            assert redact_response.result is not None
+            output_messages = list_message_adapter.validate_python(redact_response.result.data)
 
         guard_output_response = self._client.ai_guard_client.guard_chat_completions(
             guard_input={"messages": messages + output_messages},
@@ -1870,15 +1869,14 @@ async def create(
             if c.type == "output_text"
         ]
 
-        # TODO: reintroduce.
         # FPE unredact.
-        # if guard_input_response.result.fpe_context is not None:
-        #     redact_response = await self._client.redact_client.unredact(
-        #         output_messages,
-        #         fpe_context=guard_input_response.result.fpe_context,
-        #     )
-        #     assert redact_response.result is not None
-        #     output_messages = list_message_adapter.validate_python(redact_response.result.data)
+        if guard_input_response.result.fpe_context is not None:
+            redact_response = self._client.ai_guard_client.unredact(
+                redacted_data=output_messages,
+                fpe_context=guard_input_response.result.fpe_context,
+            )
+            assert redact_response.result is not None
+            output_messages = list_message_adapter.validate_python(redact_response.result.data)
 
         guard_output_response = self._client.ai_guard_client.guard_chat_completions(
             guard_input={"messages": messages + output_messages}, event_type="output"