fix: cosmetic

agentic_security/logutils.py

@@ -1,25 +1,3 @@
-# import sys
-
-# from loguru import logger
-
-# # Define custom colors
-# BLUE = "#89CFF0"
-# BROWN = "#8B4513"  # Brown for DEBUG
-
-# # Define custom log level colors
-# logger.level("DEBUG", color=f"<fg {BROWN}>")
-# logger.level("INFO", color=f"<fg {BLUE}>")
-
-# # Define custom log format with aligned messages and colored levels
-# LOG_FORMAT = (
-#     "<level>{level:<8}</level> "  # Properly formatted and colored log level
-#     "<level>{message:<100}</level> "  # Left-aligned message for readability
-#     "<cyan>{file.name}</cyan>:<cyan>{line}</cyan>"  # File name and line number in cyan
-# )
-
-# # Remove default handlers and add a new one with custom formatting
-# logger.remove()
-# logger.add(sys.stdout, format=LOG_FORMAT, level="DEBUG", colorize=True)
 import logging
 import logging.config
 import time

agentic_security/misc/banner.py

@@ -1,11 +1,8 @@
+from importlib.metadata import version
+
 from pyfiglet import Figlet, FontNotFound
 from termcolor import colored
 
-try:
-    from importlib.metadata import version
-except ImportError:
-    from importlib_metadata import version
-
 
 def generate_banner(
     title: str = "Agentic Security",
@@ -14,7 +11,7 @@ def generate_banner(
     tagline: str = "Proactive Threat Detection & Automated Security Protocols",
     author: str = "Developed by: [Security Team]",
     website: str = "Website: https://github.com/msoedov/agentic_security",
-    warning: str | None = "",  # Using Optional for warning since it might be None
+    warning: str = "",
 ) -> str:
     """Generate a visually enhanced banner with dynamic width and borders."""
     # Define the text elements
@@ -80,10 +77,8 @@ def generate_banner(
 
 
 def init_banner():
-    return
-    ver = version("agentic_security")
     try:
-        print(generate_banner(version=ver))
+        print(generate_banner(version=version("agentic_security")))
     except Exception:
         # UnicodeEncodeError with codec on some systems
         pass

agentic_security/probe_actor/fuzzer.py

@@ -47,7 +47,9 @@ from agentic_security.probe_data import audio_generator, image_generator, msj_da
 from agentic_security.probe_data.data import prepare_prompts, create_probe_dataset
 
 #: Maximum number of characters from a prompt to include in scan results.
-MAX_PROMPT_LENGTH = settings_var("fuzzer.max_prompt_lenght", 2048)
+MAX_PROMPT_LENGTH = settings_var(
+    "fuzzer.max_prompt_length", settings_var("fuzzer.max_prompt_lenght", 2048)
+)
 #: Multiplier applied to the user-specified budget to derive the internal token limit.
 BUDGET_MULTIPLIER = settings_var("fuzzer.budget_multiplier", 100000000)
 #: Number of initial random points for the Bayesian optimizer before fitting a model.