Fix bandit.

Author: blkt

src/codegate/muxing/ollama_mappers.py

@@ -215,7 +215,7 @@ def ollama_generate_from_openai(
 
 def _gen_tool_call_id():
     letter_bytes = string.ascii_lowercase + string.digits
-    b = [letter_bytes[random.randint(0, len(letter_bytes) - 1)] for _ in range(8)]
+    b = [letter_bytes[random.randint(0, len(letter_bytes) - 1)] for _ in range(8)]  # nosec
     return "call_" + "".join(b).lower()
 
 

src/codegate/pipeline/pii/pii.py

@@ -320,12 +320,12 @@ async def process_chunk(  # noqa: C901
                 end_idx = text.find(self.marker_end, start_idx + 1)
                 if end_idx == -1:
                     # Incomplete marker, buffer the rest only if it can be a UUID
-                    if start_idx + 1 < len(content) and not can_be_uuid(content[start_idx + 1 :]):
+                    if start_idx + 1 < len(text) and not can_be_uuid(text[start_idx + 1 :]):
                         # the buffer can't be a UUID, so we can't process it, just return
-                        result.append(content[current_pos:])
+                        result.append(text[current_pos:])
                     else:
                         # this can still be a UUID
-                        context.prefix_buffer = content[current_pos:]
+                        context.prefix_buffer = text[current_pos:]
                     break
 
                 # Add text before marker

src/codegate/types/anthropic/_generators.py

@@ -155,5 +155,5 @@ async def message_wrapper(lines):
                 raise ValueError(f"anthropic: unexpected event type '{event_type}'")
 
     # The following should always hold when we get here
-    assert event_type == "message_stop" or event_type == "error"
+    assert event_type == "message_stop" or event_type == "error"  # nosec
     return

tests/pipeline/pii/test_pi.py

@@ -132,27 +132,27 @@ async def test_process_chunk_with_uuid(self, unredaction_step):
 
     @pytest.mark.asyncio
     async def test_detect_not_an_uuid(self, unredaction_step):
-        chunk1 = ModelResponse(
+        chunk1 = StreamingChatCompletion(
             id="test",
             choices=[
-                StreamingChoices(
+                ChoiceDelta(
                     finish_reason=None,
                     index=0,
-                    delta=Delta(content="#"),
+                    delta=MessageDelta(content="#"),
                     logprobs=None,
                 )
             ],
             created=1234567890,
             model="test-model",
             object="chat.completion.chunk",
         )
-        chunk2 = ModelResponse(
+        chunk2 = StreamingChatCompletion(
             id="test",
             choices=[
-                StreamingChoices(
+                ChoiceDelta(
                     finish_reason=None,
                     index=0,
-                    delta=Delta(content=" filepath"),
+                    delta=MessageDelta(content=" filepath"),
                     logprobs=None,
                 )
             ],