More fixes

jakozaur · jakozaur · commit cb185187301e · 2025-08-16T19:49:01.000+02:00
diff --git a/src/categories/ai_escalation.py b/src/categories/ai_escalation.py
@@ -12,9 +12,9 @@
 import re
 from dataclasses import dataclass
 
-from utils.evaluator import EvaluationResult, VulnerabilityCategory
-from utils.live_display import get_display
-from utils.model_client import ModelResponse
+from src.utils.evaluator import EvaluationResult, VulnerabilityCategory
+from src.utils.live_display import get_display
+from src.utils.model_client import ModelResponse
 
 from .base import BaseTest, BaseTester
 from .registry import register_category
diff --git a/src/categories/attachment_ai.py b/src/categories/attachment_ai.py
@@ -12,8 +12,8 @@
 import re
 from dataclasses import dataclass
 
-from utils.evaluator import EvaluationResult, VulnerabilityCategory
-from utils.model_client import ModelResponse
+from src.utils.evaluator import EvaluationResult, VulnerabilityCategory
+from src.utils.model_client import ModelResponse
 
 from .base import BaseTest, BaseTester
 from .registry import register_category
diff --git a/src/categories/base.py b/src/categories/base.py
@@ -7,9 +7,9 @@
 from dataclasses import dataclass, field
 from typing import Any
 
-from utils.evaluator import EvaluationResult, ResponseEvaluator, VulnerabilityCategory
-from utils.live_display import get_display
-from utils.model_client import ModelResponse, OllamaClient
+from src.utils.evaluator import EvaluationResult, ResponseEvaluator, VulnerabilityCategory
+from src.utils.live_display import get_display
+from src.utils.model_client import ModelResponse, OllamaClient
 
 
 @dataclass
diff --git a/src/categories/cot_overload.py b/src/categories/cot_overload.py
@@ -13,8 +13,8 @@
 from dataclasses import dataclass
 from typing import Any
 
-from utils.evaluator import EvaluationResult, VulnerabilityCategory
-from utils.model_client import ModelResponse
+from src.utils.evaluator import EvaluationResult, VulnerabilityCategory
+from src.utils.model_client import ModelResponse
 
 from .base import BaseTest, BaseTester
 from .registry import register_category
diff --git a/src/categories/deception_adderall.py b/src/categories/deception_adderall.py
@@ -11,8 +11,8 @@
 
 from dataclasses import dataclass
 
-from utils.evaluator import EvaluationResult, VulnerabilityCategory
-from utils.model_client import ModelResponse
+from src.utils.evaluator import EvaluationResult, VulnerabilityCategory
+from src.utils.model_client import ModelResponse
 
 from .base import BaseTest, BaseTester
 from .registry import register_category
diff --git a/src/categories/deception_samples.py b/src/categories/deception_samples.py
@@ -5,7 +5,7 @@
 
 from dataclasses import dataclass
 
-from utils.evaluator import VulnerabilityCategory
+from src.utils.evaluator import VulnerabilityCategory
 
 from .base import BaseTest, BaseTester
 from .registry import register_category
diff --git a/src/categories/deception_security.py b/src/categories/deception_security.py
@@ -11,8 +11,8 @@
 
 from dataclasses import dataclass
 
-from utils.evaluator import EvaluationResult, VulnerabilityCategory
-from utils.model_client import ModelResponse
+from src.utils.evaluator import EvaluationResult, VulnerabilityCategory
+from src.utils.model_client import ModelResponse
 
 from .base import BaseTest, BaseTester
 from .registry import register_category
diff --git a/src/categories/deception_speed_pressure.py b/src/categories/deception_speed_pressure.py
@@ -12,9 +12,9 @@
 import re
 from dataclasses import dataclass
 
-from utils.evaluator import EvaluationResult, VulnerabilityCategory
-from utils.live_display import get_display
-from utils.model_client import ModelResponse
+from src.utils.evaluator import EvaluationResult, VulnerabilityCategory
+from src.utils.live_display import get_display
+from src.utils.model_client import ModelResponse
 
 from .base import BaseTest, BaseTester
 from .registry import register_category
diff --git a/src/categories/exploit.py b/src/categories/exploit.py
@@ -5,7 +5,7 @@
 
 from dataclasses import dataclass
 
-from utils.evaluator import VulnerabilityCategory
+from src.utils.evaluator import VulnerabilityCategory
 
 from .base import BaseTest, BaseTester
 from .registry import register_category
diff --git a/src/categories/exploit_v2.py b/src/categories/exploit_v2.py
@@ -6,7 +6,7 @@
 
 from dataclasses import dataclass
 
-from utils.evaluator import VulnerabilityCategory
+from src.utils.evaluator import VulnerabilityCategory
 
 from .base import BaseTest, BaseTester
 from .registry import register_category
diff --git a/src/categories/exploit_v3.py b/src/categories/exploit_v3.py
@@ -6,7 +6,7 @@
 
 from dataclasses import dataclass
 
-from utils.evaluator import VulnerabilityCategory
+from src.utils.evaluator import VulnerabilityCategory
 
 from .base import BaseTest, BaseTester
 from .registry import register_category
diff --git a/src/categories/rce_v1.py b/src/categories/rce_v1.py
@@ -10,7 +10,7 @@
 
 from dataclasses import dataclass
 
-from utils.evaluator import VulnerabilityCategory
+from src.utils.evaluator import VulnerabilityCategory
 
 from .base import BaseTest, BaseTester
 from .registry import register_category
diff --git a/src/categories/registry.py b/src/categories/registry.py
@@ -8,12 +8,12 @@
 from dataclasses import dataclass
 from typing import TYPE_CHECKING, Any
 
-from utils.evaluator import VulnerabilityCategory
+from src.utils.evaluator import VulnerabilityCategory
 
 from .base import BaseTester
 
 if TYPE_CHECKING:
-    from utils.model_client import OllamaClient
+    from src.utils.model_client import OllamaClient
 
 
 @dataclass
diff --git a/src/categories/supremacy.py b/src/categories/supremacy.py
@@ -11,8 +11,8 @@
 
 from dataclasses import dataclass
 
-from utils.evaluator import EvaluationResult, VulnerabilityCategory
-from utils.model_client import ModelResponse
+from src.utils.evaluator import EvaluationResult, VulnerabilityCategory
+from src.utils.model_client import ModelResponse
 
 from .base import BaseTest, BaseTester
 from .registry import register_category
diff --git a/src/cli/pentest.py b/src/cli/pentest.py
@@ -5,12 +5,15 @@
 """
 
 import click
-
-from categories.registry import CATEGORY_DESCRIPTIONS, TestRegistry, initialize_builtin_categories
-from utils.model_client import OllamaClient
+from src.categories.registry import (
+    CATEGORY_DESCRIPTIONS,
+    TestRegistry,
+    initialize_builtin_categories,
+)
+from src.utils.model_client import OllamaClient
 
 # Import the essential testbed functionality
-from utils.testbed_lib import (
+from src.utils.testbed_lib import (
     calculate_timeout_stats,
     ensure_directories,
     load_config,
@@ -161,7 +164,7 @@ def main(
         click.echo(f"🔄 Repeat mode: Each test will run {repeat} times")
 
     # Configure live display based on flags
-    from utils.live_display import get_display, set_display_options
+    from src.utils.live_display import get_display, set_display_options
 
     enable_live = not (quiet or no_live)  # Live preview enabled by default, disabled by flags
     set_display_options(enable_live=enable_live, quiet_mode=quiet, verbose=verbose)
diff --git a/src/generate_example.py b/src/generate_example.py
@@ -96,9 +96,9 @@ def create_reproduction_notebook() -> None:
                     "import pandas as pd\n",
                     "\n",
                     "# Import security testing utilities\n",
-                    "from utils.model_client import OllamaClient\n",
-                    "from utils.evaluator import ResponseEvaluator, VulnerabilityCategory\n",
-                    "from categories.deception_samples import DeceptionTester",
+                    "from src.utils.model_client import OllamaClient\n",
+                    "from src.utils.evaluator import ResponseEvaluator, VulnerabilityCategory\n",
+                    "from src.categories.deception_samples import DeceptionTester",
                 ],
             },
             {
diff --git a/src/ui/cli_findings.py b/src/ui/cli_findings.py
@@ -514,8 +514,12 @@ def _export_single_competition_finding(self, finding_data: dict) -> None:
                 return
 
             # Import required modules
-            from utils.evaluator import EvaluationResult, VulnerabilityCategory, VulnerabilityScore
-            from utils.findings_generator import FindingsGenerator
+            from src.utils.evaluator import (
+                EvaluationResult,
+                VulnerabilityCategory,
+                VulnerabilityScore,
+            )
+            from src.utils.findings_generator import FindingsGenerator
 
             generator = FindingsGenerator()
 
