Optimize signtures

Author: lukehinds

src/codegate/cli.py

@@ -305,16 +305,16 @@ def serve(
         ca = CertificateAuthority.get_instance()
         ca.ensure_certificates_exist()
 
-        # Set up event loop
-        loop = asyncio.new_event_loop()
-        asyncio.set_event_loop(loop)
-
         # Initialize secrets manager and pipeline factory
         secrets_manager = SecretsManager()
         pipeline_factory = PipelineFactory(secrets_manager)
 
         app = init_app(pipeline_factory)
 
+        # Set up event loop
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+
         # Run the server
         try:
             loop.run_until_complete(run_servers(cfg, app))
@@ -331,6 +331,7 @@ def serve(
         logger.exception("Unexpected error occurred")
         click.echo(f"Error: {e}", err=True)
         sys.exit(1)
+        
 
 
 async def run_servers(cfg: Config, app) -> None:

src/codegate/pipeline/secrets/secrets.py

@@ -26,6 +26,9 @@ class CodegateSecrets(PipelineStep):
     def __init__(self):
         """Initialize the CodegateSecrets pipeline step."""
         super().__init__()
+        # Initialize signatures eagerly
+        CodegateSignatures.initialize("signatures.yaml")
+        CodegateSignatures._ensure_signatures_loaded()
 
     @property
     def name(self) -> str:

src/codegate/pipeline/secrets/signatures.py

@@ -61,9 +61,11 @@ def initialize(cls, yaml_path: str) -> None:
             raise FileNotFoundError(f"Signatures file not found: {yaml_path}")
 
         with cls._instance_lock:
-            cls._yaml_path = yaml_path
-            cls._signatures_loaded = False
-            logger.debug(f"SecretFinder initialized with {yaml_path}")
+            # Only initialize if not already initialized with this path
+            if cls._yaml_path != yaml_path:
+                cls._yaml_path = yaml_path
+                cls._signatures_loaded = False
+                logger.debug(f"SecretFinder initialized with {yaml_path}")
 
     @classmethod
     def _preprocess_yaml(cls, content: str) -> str:

src/codegate/server.py

@@ -4,7 +4,6 @@
 from codegate import __description__, __version__
 from codegate.dashboard.dashboard import dashboard_router
 from codegate.pipeline.factory import PipelineFactory
-from codegate.pipeline.secrets.signatures import CodegateSignatures
 from codegate.providers.anthropic.provider import AnthropicProvider
 from codegate.providers.llamacpp.provider import LlamaCppProvider
 from codegate.providers.ollama.provider import OllamaProvider
@@ -31,9 +30,6 @@ def init_app(pipeline_factory: PipelineFactory) -> FastAPI:
     # Create provider registry
     registry = ProviderRegistry(app)
 
-    # Initialize SignaturesFinder
-    CodegateSignatures.initialize("signatures.yaml")
-
     # Register all known providers
     registry.add_provider(
         "openai",