From 2dca704d5038710ec0d60463612fd741a3047ac5 Mon Sep 17 00:00:00 2001
From: Jakub Hrozek <jakub@stacklok.com>
Date: Tue, 4 Mar 2025 21:00:03 +0100
Subject: [PATCH] Run make all to reformat code automatically

I don't know why our lint CI step doesn't catch this (it did slap me the
other day for other reasons, so it is working), but running `make all`
produced a bunch of reformats.
---
 src/codegate/pipeline/pii/analyzer.py                 | 3 +--
 src/codegate/pipeline/pii/pii.py                      | 1 -
 src/codegate/pipeline/sensitive_data/manager.py       | 3 ++-
 src/codegate/pipeline/sensitive_data/session_store.py | 2 +-
 tests/pipeline/pii/test_analyzer.py                   | 1 -
 tests/pipeline/sensitive_data/test_manager.py         | 3 ++-
 tests/pipeline/sensitive_data/test_session_store.py   | 2 +-
 tests/test_server.py                                  | 1 -
 8 files changed, 7 insertions(+), 9 deletions(-)

diff --git a/src/codegate/pipeline/pii/analyzer.py b/src/codegate/pipeline/pii/analyzer.py
index 96442824..706deb9b 100644
--- a/src/codegate/pipeline/pii/analyzer.py
+++ b/src/codegate/pipeline/pii/analyzer.py
@@ -1,10 +1,9 @@
-from typing import Any, List, Optional
+from typing import List, Optional
 
 import structlog
 from presidio_analyzer import AnalyzerEngine
 from presidio_anonymizer import AnonymizerEngine
 
-from codegate.db.models import AlertSeverity
 from codegate.pipeline.base import PipelineContext
 from codegate.pipeline.sensitive_data.session_store import SessionStore
 
diff --git a/src/codegate/pipeline/pii/pii.py b/src/codegate/pipeline/pii/pii.py
index fde89428..d7f33d67 100644
--- a/src/codegate/pipeline/pii/pii.py
+++ b/src/codegate/pipeline/pii/pii.py
@@ -1,5 +1,4 @@
 from typing import Any, Dict, List, Optional, Tuple
-import uuid
 
 import regex as re
 import structlog
diff --git a/src/codegate/pipeline/sensitive_data/manager.py b/src/codegate/pipeline/sensitive_data/manager.py
index 89506d15..bf467878 100644
--- a/src/codegate/pipeline/sensitive_data/manager.py
+++ b/src/codegate/pipeline/sensitive_data/manager.py
@@ -1,7 +1,8 @@
-import json
 from typing import Dict, Optional
+
 import pydantic
 import structlog
+
 from codegate.pipeline.sensitive_data.session_store import SessionStore
 
 logger = structlog.get_logger("codegate")
diff --git a/src/codegate/pipeline/sensitive_data/session_store.py b/src/codegate/pipeline/sensitive_data/session_store.py
index 5e508847..7a33abd2 100644
--- a/src/codegate/pipeline/sensitive_data/session_store.py
+++ b/src/codegate/pipeline/sensitive_data/session_store.py
@@ -1,5 +1,5 @@
-from typing import Dict, Optional
 import uuid
+from typing import Dict, Optional
 
 
 class SessionStore:
diff --git a/tests/pipeline/pii/test_analyzer.py b/tests/pipeline/pii/test_analyzer.py
index d626b8cf..e856653c 100644
--- a/tests/pipeline/pii/test_analyzer.py
+++ b/tests/pipeline/pii/test_analyzer.py
@@ -1,7 +1,6 @@
 from unittest.mock import MagicMock, patch
 
 import pytest
-from presidio_analyzer import RecognizerResult
 
 from codegate.pipeline.pii.analyzer import PiiAnalyzer
 
diff --git a/tests/pipeline/sensitive_data/test_manager.py b/tests/pipeline/sensitive_data/test_manager.py
index 6115ad14..66305388 100644
--- a/tests/pipeline/sensitive_data/test_manager.py
+++ b/tests/pipeline/sensitive_data/test_manager.py
@@ -1,6 +1,7 @@
-import json
 from unittest.mock import MagicMock, patch
+
 import pytest
+
 from codegate.pipeline.sensitive_data.manager import SensitiveData, SensitiveDataManager
 from codegate.pipeline.sensitive_data.session_store import SessionStore
 
diff --git a/tests/pipeline/sensitive_data/test_session_store.py b/tests/pipeline/sensitive_data/test_session_store.py
index b9ab64fe..e90b953e 100644
--- a/tests/pipeline/sensitive_data/test_session_store.py
+++ b/tests/pipeline/sensitive_data/test_session_store.py
@@ -1,5 +1,5 @@
-import uuid
 import pytest
+
 from codegate.pipeline.sensitive_data.session_store import SessionStore
 
 
diff --git a/tests/test_server.py b/tests/test_server.py
index aa549810..bcf55e7e 100644
--- a/tests/test_server.py
+++ b/tests/test_server.py
@@ -14,7 +14,6 @@
 
 from codegate import __version__
 from codegate.pipeline.factory import PipelineFactory
-from codegate.pipeline.sensitive_data.manager import SensitiveDataManager
 from codegate.providers.registry import ProviderRegistry
 from codegate.server import init_app
 from src.codegate.cli import UvicornServer, cli