Merge pull request #16 from prompt-security/make_debug_level_configurable_only_on_commandline

Make debug level configurable only on command line

Author: vitaly-ps

ps_fuzz/app_config.py

@@ -4,7 +4,6 @@
 import colorama
 from .util import wrap_text
 from .results_table import print_table
-from .ps_logging import setup_logging
 import logging
 logger = logging.getLogger(__name__)
 
@@ -17,8 +16,7 @@ class AppConfig:
         'num_attempts': 3,
         'num_threads': 4,
         'attack_temperature': 0.6,
-        'debug_level': 1,
-        'system_prompt': ''
+        'system_prompt': '',
     }
 
     def __init__(self, config_state_file: str):
@@ -131,17 +129,6 @@ def num_threads(self, value: int):
         self.config_state['num_threads'] = value
         self.save()
 
-    @property
-    def debug_level(self) -> int:
-        return self.config_state['debug_level']
-
-    @debug_level.setter
-    def debug_level(self, value: int):
-        if not (0 <= value <= 2): raise ValueError("Debug level must be between 0 and 2")
-        self.config_state['debug_level'] = value
-        setup_logging(value) # Update logging level based on configured debug level
-        self.save()
-
     @property
     def system_prompt(self) -> str:
         return self.config_state['system_prompt']

ps_fuzz/cli.py

@@ -5,6 +5,7 @@
 from dotenv import load_dotenv
 dotenv_path = os.path.join(os.getcwd(), '.env')
 load_dotenv(dotenv_path)
+from .ps_logging import setup_logging
 from .chat_clients import *
 from .client_config import ClientConfig
 from .attack_config import AttackConfig
@@ -46,6 +47,9 @@ def main():
             print(f"  {BRIGHT}{test_name}{RESET}: {test_description}")
         sys.exit(0)
 
+    # Setup debug level (default 1)
+    setup_logging(args.debug_level if args.debug_level is not None else 1)
+
     # Load application config from file (if exists)
     app_config = AppConfig(APP_CONFIG_FILE)
 

ps_fuzz/interactive_mode.py

@@ -53,7 +53,6 @@ def show(cls, state: AppConfig):
             ['Fuzzer Configuration', None, FuzzerOptions],
             ['Target LLM Configuration', None, TargetLLMOptions],
             ['Attack LLM Configuration', None, AttackLLMOptions],
-            ['Debug Level', None, DebugOptions],
             ['Show all configuration', show_all_config, MainMenu],
             ['Exit', None, None],
         ]
@@ -99,22 +98,6 @@ def show(cls, state: AppConfig):
         state.num_attempts = int(result['num_attempts'])
         return MainMenu
 
-class DebugOptions:
-    @classmethod
-    def show(cls, state: AppConfig):
-        print("Debug Options: Review and modify the debug log level")
-        print("----------------------------------------------------")
-        result = inquirer.prompt([
-            inquirer.Text('debug_level',
-                message="Debug Level (0-2) (0=warnings and errors; 1 = normal; 2 = verbose)",
-                default=str(state.debug_level),
-                validate=lambda _, x: x.isdigit() and int(x) > 0
-            ),
-        ])
-        if result is None: return  # Handle prompt cancellation concisely
-        state.debug_level = int(result['debug_level'])
-        return MainMenu
-
 class TargetLLMOptions:
     @classmethod
     def show(cls, state: AppConfig):