Secrets step simplification (#1074)

* Fix typo

* Rely on redacted_secrets_count to detect if this was the first message

Author: jhrozek

src/codegate/pipeline/secrets/secrets.py

@@ -271,7 +271,7 @@ def _redact_text(
         return text_encryptor.obfuscate(text, snippet)
 
     async def process(
-        self, reques: Any, context: PipelineContext
+        self, request: Any, context: PipelineContext
     ) -> PipelineResult:
         """
         Process the request to find and protect secrets in all messages.
@@ -498,33 +498,30 @@ async def process_chunk(
         )
 
         # Check if this is the first chunk (delta role will be present, others will not)
-        for _ in itertools.takewhile(lambda x: x[0] == 1, enumerate(chunk.get_content())):
-            redacted_count = input_context.metadata["redacted_secrets_count"]
-            secret_text = "secret" if redacted_count == 1 else "secrets"
-            # Create notification chunk
-            if tool_name in ["cline", "kodu"]:
-                # NOTE: Original code was ensuring that role was
-                # "assistant" here, we might have to do that as well,
-                # but I believe it was defensive programming or
-                # leftover of some refactoring.
-                notification_chunk = self._create_chunk(
-                    chunk,
-                    f"<thinking>\n🛡️ [CodeGate prevented {redacted_count} {secret_text}]"
-                    f"(http://localhost:9090/?search=codegate-secrets) from being leaked "
-                    f"by redacting them.</thinking>\n\n",
-                )
-            else:
-                notification_chunk = self._create_chunk(
-                    chunk,
-                    f"\n🛡️ [CodeGate prevented {redacted_count} {secret_text}]"
-                    f"(http://localhost:9090/?search=codegate-secrets) from being leaked "
-                    f"by redacting them.\n\n",
-                )
-
-            # Reset the counter
-            input_context.metadata["redacted_secrets_count"] = 0
+        redacted_count = input_context.metadata["redacted_secrets_count"]
+        secret_text = "secret" if redacted_count == 1 else "secrets"
+        # Create notification chunk
+        if tool_name in ["cline", "kodu"]:
+            # NOTE: Original code was ensuring that role was
+            # "assistant" here, we might have to do that as well,
+            # but I believe it was defensive programming or
+            # leftover of some refactoring.
+            notification_chunk = self._create_chunk(
+                chunk,
+                f"<thinking>\n🛡️ [CodeGate prevented {redacted_count} {secret_text}]"
+                f"(http://localhost:9090/?search=codegate-secrets) from being leaked "
+                f"by redacting them.</thinking>\n\n",
+            )
+        else:
+            notification_chunk = self._create_chunk(
+                chunk,
+                f"\n🛡️ [CodeGate prevented {redacted_count} {secret_text}]"
+                f"(http://localhost:9090/?search=codegate-secrets) from being leaked "
+                f"by redacting them.\n\n",
+            )
 
-            # Return both the notification and original chunk
-            return [notification_chunk, chunk]
+        # Reset the counter
+        input_context.metadata["redacted_secrets_count"] = 0
 
-        return [chunk]
+        # Return both the notification and original chunk
+        return [notification_chunk, chunk]