From 812f9e76197137ac084aa38db89f6bca068f97a0 Mon Sep 17 00:00:00 2001 From: Alexander Myasoedov Date: Sat, 8 Mar 2025 18:02:02 +0200 Subject: [PATCH] fix(SecurityScanner): --- agentic_security/__init__.py | 4 ++-- agentic_security/__main__.py | 8 ++++---- agentic_security/lib.py | 2 +- agentic_security/logutils.py | 1 - tests/test_lib.py | 24 ++++++++++++------------ 5 files changed, 19 insertions(+), 20 deletions(-) diff --git a/agentic_security/__init__.py b/agentic_security/__init__.py index 6ee641e..4944b1e 100644 --- a/agentic_security/__init__.py +++ b/agentic_security/__init__.py @@ -1,3 +1,3 @@ -from .lib import AgenticSecurity +from .lib import SecurityScanner -__all__ = ["AgenticSecurity"] +__all__ = ["SecurityScanner"] diff --git a/agentic_security/__main__.py b/agentic_security/__main__.py index af9ec60..4af26dd 100644 --- a/agentic_security/__main__.py +++ b/agentic_security/__main__.py @@ -5,7 +5,7 @@ import fire import uvicorn from agentic_security.app import app -from agentic_security.lib import AgenticSecurity +from agentic_security.lib import SecurityScanner from agentic_security.misc.banner import init_banner @@ -32,14 +32,14 @@ class CLI: Run Agentic Security in CI mode. """ sys.path.append(os.path.dirname(".")) - AgenticSecurity().entrypoint() + SecurityScanner().entrypoint() def init(self, host: str = "0.0.0.0", port: int = 8718): """ Generate the default CI configuration file. """ sys.path.append(os.path.dirname(".")) - AgenticSecurity().generate_default_settings(host, port) + SecurityScanner().generate_default_settings(host, port) i = init @@ -48,7 +48,7 @@ class CLI: List all available security checks. """ sys.path.append(os.path.dirname(".")) - AgenticSecurity().list_checks() + SecurityScanner().list_checks() def main(): diff --git a/agentic_security/lib.py b/agentic_security/lib.py index 21880e4..958525d 100644 --- a/agentic_security/lib.py +++ b/agentic_security/lib.py @@ -23,7 +23,7 @@ YELLOW = colorama.Fore.YELLOW BLUE = colorama.Fore.BLUE -class AgenticSecurity(SettingsMixin): +class SecurityScanner(SettingsMixin): @classmethod async def async_scan( cls, diff --git a/agentic_security/logutils.py b/agentic_security/logutils.py index 9708765..ca28f3b 100644 --- a/agentic_security/logutils.py +++ b/agentic_security/logutils.py @@ -24,7 +24,6 @@ import logging import logging.config from os import getenv - LOGGER_NAME = None LOGGING_CONFIG = { diff --git a/tests/test_lib.py b/tests/test_lib.py index d096703..dac2f24 100644 --- a/tests/test_lib.py +++ b/tests/test_lib.py @@ -8,7 +8,7 @@ import time import pytest import agentic_security.test_spec_assets as test_spec_assets -from agentic_security.lib import AgenticSecurity +from agentic_security.lib import SecurityScanner def has_module(module_name): @@ -57,7 +57,7 @@ class TestLibraryLevel: maxBudget = 1000000 max_th = 0.3 datasets = make_test_registry() - result = AgenticSecurity.scan(llmSpec, maxBudget, datasets, max_th) + result = SecurityScanner.scan(llmSpec, maxBudget, datasets, max_th) assert isinstance(result, dict) print(result) assert len(result) in [0, 1] @@ -68,7 +68,7 @@ class TestLibraryLevel: maxBudget = 1000 max_th = 0.3 datasets = make_test_registry() - result = AgenticSecurity.scan( + result = SecurityScanner.scan( llmSpec, maxBudget, datasets, max_th, enableMultiStepAttack=True ) assert isinstance(result, dict) @@ -93,7 +93,7 @@ class TestLibraryLevel: "opts": {"port": 9094}, }, ] - result = AgenticSecurity.scan(llmSpec, maxBudget, datasets, max_th) + result = SecurityScanner.scan(llmSpec, maxBudget, datasets, max_th) assert isinstance(result, dict) print(result) assert len(result) in [0, 1] @@ -120,7 +120,7 @@ class TestLibraryLevel: "modality": "text", }, ] - result = AgenticSecurity.scan(llmSpec, maxBudget, datasets, max_th) + result = SecurityScanner.scan(llmSpec, maxBudget, datasets, max_th) assert isinstance(result, dict) print(result) assert len(result) in [0, 1] @@ -148,7 +148,7 @@ class TestLibraryLevel: "modality": "text", }, ] - result = AgenticSecurity.scan(llmSpec, maxBudget, datasets, max_th) + result = SecurityScanner.scan(llmSpec, maxBudget, datasets, max_th) assert isinstance(result, dict) print(result) assert len(result) in [0, 1] @@ -164,10 +164,10 @@ class TestEntrypointCI: temp_path = os.path.join(tmpdir, "custom_agesec.toml") # Override default_path to the temporary path - AgenticSecurity.default_path = temp_path + SecurityScanner.default_path = temp_path # Generate the default configuration - security = AgenticSecurity() + security = SecurityScanner() security.generate_default_settings() # Check that the config file was created at the temporary path @@ -189,17 +189,17 @@ class TestEntrypointCI: temp_path = os.path.join(tmpdir, "custom_agesec.toml") # Override default_path to the temporary path - AgenticSecurity.default_path = temp_path + SecurityScanner.default_path = temp_path # Generate the default configuration - security = AgenticSecurity() + security = SecurityScanner() security.generate_default_settings() # Load the generated configuration - AgenticSecurity.load_config(temp_path) + SecurityScanner.load_config(temp_path) # Validate loaded configuration - config = AgenticSecurity.config + config = SecurityScanner.config assert ( config["general"]["maxBudget"] == 1000000 ), "maxBudget should be 1000000"