Action fix

src/llmguardian/__init__.py

@@ -7,13 +7,14 @@ __version__ = "1.4.0"
 __author__ = "dewitt4"
 __license__ = "Apache-2.0"
 
-from typing import List, Dict, Optional
+from typing import Dict, List, Optional
 
-# Package level imports
-from .scanners.prompt_injection_scanner import PromptInjectionScanner
 from .core.config import Config
 from .core.logger import setup_logging
 
+# Package level imports
+from .scanners.prompt_injection_scanner import PromptInjectionScanner
+
 # Initialize logging
 setup_logging()
 

src/llmguardian/agency/__init__.py

@@ -1,5 +1,5 @@
 # src/llmguardian/agency/__init__.py
-from .permission_manager import PermissionManager
 from .action_validator import ActionValidator
-from .scope_limiter import ScopeLimiter
 from .executor import SafeExecutor
+from .permission_manager import PermissionManager
+from .scope_limiter import ScopeLimiter

src/llmguardian/agency/action_validator.py

@@ -1,7 +1,8 @@
 # src/llmguardian/agency/action_validator.py
-from typing import Dict, List, Optional
 from dataclasses import dataclass
 from enum import Enum
+from typing import Dict, List, Optional
+
 from ..core.logger import SecurityLogger
 
 

src/llmguardian/agency/executor.py

@@ -1,6 +1,7 @@
 # src/llmguardian/agency/executor.py
-from typing import Dict, Any, Optional
 from dataclasses import dataclass
+from typing import Any, Dict, Optional
+
 from ..core.logger import SecurityLogger
 from .action_validator import Action, ActionValidator
 from .permission_manager import PermissionManager

src/llmguardian/agency/permission_manager.py

@@ -1,7 +1,8 @@
 # src/llmguardian/agency/permission_manager.py
-from typing import Dict, List, Optional, Set
 from dataclasses import dataclass
 from enum import Enum
+from typing import Dict, List, Optional, Set
+
 from ..core.logger import SecurityLogger
 
 

src/llmguardian/agency/scope_limiter.py

@@ -1,7 +1,8 @@
 # src/llmguardian/agency/scope_limiter.py
-from typing import Dict, List, Optional, Set
 from dataclasses import dataclass
 from enum import Enum
+from typing import Dict, List, Optional, Set
+
 from ..core.logger import SecurityLogger
 
 

src/llmguardian/api/__init__.py

@@ -1,4 +1,4 @@
 # src/llmguardian/api/__init__.py
-from .routes import router
 from .models import SecurityRequest, SecurityResponse
+from .routes import router
 from .security import SecurityMiddleware

src/llmguardian/api/app.py

@@ -1,6 +1,7 @@
 # src/llmguardian/api/app.py
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
+
 from .routes import router
 from .security import SecurityMiddleware
 

src/llmguardian/api/models.py

@@ -1,8 +1,9 @@
 # src/llmguardian/api/models.py
-from pydantic import BaseModel
-from typing import List, Optional, Dict, Any
-from enum import Enum
 from datetime import datetime
+from enum import Enum
+from typing import Any, Dict, List, Optional
+
+from pydantic import BaseModel
 
 
 class SecurityLevel(str, Enum):

src/llmguardian/api/routes.py

@@ -1,9 +1,11 @@
 # src/llmguardian/api/routes.py
-from fastapi import APIRouter, Depends, HTTPException
 from typing import List
-from .models import SecurityRequest, SecurityResponse, PrivacyRequest, VectorRequest
+
+from fastapi import APIRouter, Depends, HTTPException
+
 from ..data.privacy_guard import PrivacyGuard
 from ..vectors.vector_scanner import VectorScanner
+from .models import PrivacyRequest, SecurityRequest, SecurityResponse, VectorRequest
 from .security import verify_token
 
 router = APIRouter()

src/llmguardian/api/security.py

@@ -1,10 +1,11 @@
 # src/llmguardian/api/security.py
-from fastapi import HTTPException, Security
-from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
-import jwt
 from datetime import datetime, timedelta
 from typing import Optional
 
+import jwt
+from fastapi import HTTPException, Security
+from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
+
 security = HTTPBearer()
 
 

src/llmguardian/cli/cli_interface.py

@@ -3,21 +3,22 @@ LLMGuardian CLI Interface
 Command-line interface for the LLMGuardian security tool.
 """
 
-import click
 import json
 import logging
-from typing import Optional, Dict
 from pathlib import Path
-from rich.console import Console
-from rich.table import Table
-from rich.panel import Panel
-from rich import print as rprint
-from rich.logging import RichHandler
+from typing import Dict, Optional
+
+import click
 from prompt_injection_scanner import (
-    PromptInjectionScanner,
     InjectionPattern,
     InjectionType,
+    PromptInjectionScanner,
 )
+from rich import print as rprint
+from rich.console import Console
+from rich.logging import RichHandler
+from rich.panel import Panel
+from rich.table import Table
 
 # Set up logging with rich
 logging.basicConfig(

src/llmguardian/core/__init__.py

@@ -2,9 +2,9 @@
 core/__init__.py - Core module initialization for LLMGuardian
 """
 
-from typing import Dict, Any, Optional
 import logging
 from pathlib import Path
+from typing import Any, Dict, Optional
 
 # Version information
 __version__ = "1.0.0"
@@ -12,29 +12,29 @@ __author__ = "dewitt4"
 __license__ = "Apache-2.0"
 
 # Core components
-from .config import Config, SecurityConfig, APIConfig, LoggingConfig, MonitoringConfig
+from .config import APIConfig, Config, LoggingConfig, MonitoringConfig, SecurityConfig
 from .exceptions import (
-    LLMGuardianError,
-    SecurityError,
-    ValidationError,
     ConfigurationError,
+    LLMGuardianError,
     PromptInjectionError,
     RateLimitError,
+    SecurityError,
+    ValidationError,
 )
-from .logger import SecurityLogger, AuditLogger
+from .logger import AuditLogger, SecurityLogger
 from .rate_limiter import (
-    RateLimiter,
     RateLimit,
+    RateLimiter,
     RateLimitType,
     TokenBucket,
     create_rate_limiter,
 )
 from .security import (
-    SecurityService,
     SecurityContext,
-    SecurityPolicy,
     SecurityMetrics,
     SecurityMonitor,
+    SecurityPolicy,
+    SecurityService,
 )
 
 # Initialize logging

src/llmguardian/core/config.py

@@ -2,19 +2,21 @@
 core/config.py - Configuration management for LLMGuardian
 """
 
-import os
-import yaml
 import json
-from pathlib import Path
-from typing import Dict, Any, Optional, List
-from dataclasses import dataclass, asdict, field
 import logging
-from enum import Enum
+import os
 import threading
+from dataclasses import asdict, dataclass, field
+from enum import Enum
+from pathlib import Path
+from typing import Any, Dict, List, Optional
+
+import yaml
+
 from .exceptions import (
     ConfigLoadError,
-    ConfigValidationError,
     ConfigurationNotFoundError,
+    ConfigValidationError,
 )
 from .logger import SecurityLogger
 

src/llmguardian/core/events.py

@@ -2,13 +2,14 @@
 core/events.py - Event handling system for LLMGuardian
 """
 
-from typing import Dict, List, Callable, Any, Optional
-from datetime import datetime
 import threading
 from dataclasses import dataclass
+from datetime import datetime
 from enum import Enum
-from .logger import SecurityLogger
+from typing import Any, Callable, Dict, List, Optional
+
 from .exceptions import LLMGuardianError
+from .logger import SecurityLogger
 
 
 class EventType(Enum):

src/llmguardian/core/exceptions.py

@@ -2,11 +2,11 @@
 core/exceptions.py - Custom exceptions for LLMGuardian
 """
 
-from typing import Dict, Any, Optional
-from dataclasses import dataclass
-import traceback
 import logging
+import traceback
+from dataclasses import dataclass
 from datetime import datetime
+from typing import Any, Dict, Optional
 
 
 @dataclass

src/llmguardian/core/logger.py

@@ -2,12 +2,12 @@
 core/logger.py - Logging configuration for LLMGuardian
 """
 
+import json
 import logging
 import logging.handlers
-import json
 from datetime import datetime
 from pathlib import Path
-from typing import Optional, Dict, Any
+from typing import Any, Dict, Optional
 
 
 class SecurityLogger:

src/llmguardian/core/monitoring.py

@@ -2,14 +2,15 @@
 core/monitoring.py - Monitoring system for LLMGuardian
 """
 
-from datetime import datetime, timedelta
-from typing import Dict, List, Optional, Any
-from dataclasses import dataclass
+import json
+import statistics
 import threading
 import time
-import json
 from collections import deque
-import statistics
+from dataclasses import dataclass
+from datetime import datetime, timedelta
+from typing import Any, Dict, List, Optional
+
 from .logger import SecurityLogger
 
 

src/llmguardian/core/rate_limiter.py

@@ -2,18 +2,20 @@
 core/rate_limiter.py - Rate limiting implementation for LLMGuardian
 """
 
-import time
+import json
 import os
-import psutil
-from datetime import datetime, timedelta
-from typing import Dict, Optional, List, Tuple, Any
 import threading
+import time
 from dataclasses import dataclass
+from datetime import datetime, timedelta
 from enum import Enum
-import json
-from .logger import SecurityLogger
-from .exceptions import RateLimitError
+from typing import Any, Dict, List, Optional, Tuple
+
+import psutil
+
 from .events import EventManager, EventType
+from .exceptions import RateLimitError
+from .logger import SecurityLogger
 
 
 class RateLimitType(Enum):
@@ -327,8 +329,8 @@ def create_rate_limiter(
 
 if __name__ == "__main__":
     # Example usage
-    from .logger import setup_logging
     from .events import create_event_manager
+    from .logger import setup_logging
 
     security_logger, _ = setup_logging()
     event_manager = create_event_manager(security_logger)

src/llmguardian/core/scanners/prompt_injection_scanner.py

@@ -2,16 +2,17 @@
 core/scanners/prompt_injection_scanner.py - Prompt injection detection for LLMGuardian
 """
 
-import re
-from dataclasses import dataclass
-from enum import Enum
-from typing import List, Optional, Dict, Set, Pattern
 import json
 import logging
+import re
+from dataclasses import dataclass
 from datetime import datetime
+from enum import Enum
+from typing import Dict, List, Optional, Pattern, Set
+
+from ..config import Config
 from ..exceptions import PromptInjectionError
 from ..logger import SecurityLogger
-from ..config import Config
 
 
 class InjectionType(Enum):

src/llmguardian/core/security.py

@@ -5,12 +5,14 @@ core/security.py - Core security services for LLMGuardian
 import hashlib
 import hmac
 import secrets
-from typing import Optional, Dict, Any, List
 from dataclasses import dataclass
 from datetime import datetime, timedelta
+from typing import Any, Dict, List, Optional
+
 import jwt
+
 from .config import Config
-from .logger import SecurityLogger, AuditLogger
+from .logger import AuditLogger, SecurityLogger
 
 
 @dataclass

src/llmguardian/core/validation.py

@@ -2,10 +2,11 @@
 core/validation.py - Input/Output validation for LLMGuardian
 """
 
+import json
 import re
-from typing import Dict, Any, List, Optional, Tuple
 from dataclasses import dataclass
-import json
+from typing import Any, Dict, List, Optional, Tuple
+
 from .logger import SecurityLogger
 
 

src/llmguardian/dashboard/app.py

@@ -1,26 +1,27 @@
 # src/llmguardian/dashboard/app.py
 
-import streamlit as st
-import plotly.express as px
-import plotly.graph_objects as go
-import pandas as pd
-import numpy as np
-from datetime import datetime, timedelta
-from typing import Dict, List, Any, Optional
-import sys
 import os
+import sys
+from datetime import datetime, timedelta
 from pathlib import Path
+from typing import Any, Dict, List, Optional
+
+import numpy as np
+import pandas as pd
+import plotly.express as px
+import plotly.graph_objects as go
+import streamlit as st
 
 # Add parent directory to path for imports
 sys.path.insert(0, str(Path(__file__).parent.parent.parent))
 
 try:
     from llmguardian.core.config import Config
+    from llmguardian.core.logger import setup_logging
     from llmguardian.data.privacy_guard import PrivacyGuard
-    from llmguardian.monitors.usage_monitor import UsageMonitor
     from llmguardian.monitors.threat_detector import ThreatDetector, ThreatLevel
+    from llmguardian.monitors.usage_monitor import UsageMonitor
     from llmguardian.scanners.prompt_injection_scanner import PromptInjectionScanner
-    from llmguardian.core.logger import setup_logging
 except ImportError:
     # Fallback for demo mode
     Config = None

src/llmguardian/data/leak_detector.py

@@ -2,15 +2,16 @@
 data/leak_detector.py - Data leakage detection and prevention
 """
 
+import hashlib
 import re
-from typing import Dict, List, Optional, Any, Set
+from collections import defaultdict
 from dataclasses import dataclass
 from datetime import datetime
 from enum import Enum
-import hashlib
-from collections import defaultdict
-from ..core.logger import SecurityLogger
+from typing import Any, Dict, List, Optional, Set
+
 from ..core.exceptions import SecurityError
+from ..core.logger import SecurityLogger
 
 
 class LeakageType(Enum):

src/llmguardian/data/poison_detector.py

@@ -2,16 +2,18 @@
 data/poison_detector.py - Detection and prevention of data poisoning attacks
 """
 
-import numpy as np
-from typing import Dict, List, Optional, Any, Set, Tuple
+import hashlib
+import json
+from collections import defaultdict
 from dataclasses import dataclass
 from datetime import datetime
 from enum import Enum
-from collections import defaultdict
-import json
-import hashlib
-from ..core.logger import SecurityLogger
+from typing import Any, Dict, List, Optional, Set, Tuple
+
+import numpy as np
+
 from ..core.exceptions import SecurityError
+from ..core.logger import SecurityLogger
 
 
 class PoisonType(Enum):

src/llmguardian/data/privacy_guard.py

@@ -2,19 +2,21 @@
 data/privacy_guard.py - Privacy protection and enforcement
 """
 
-# Add these imports at the top
-from typing import Dict, List, Optional, Any, Set, Union
-from dataclasses import dataclass, field
-from datetime import datetime
-from enum import Enum
-import re
 import hashlib
 import json
+import re
 import threading
 import time
 from collections import defaultdict
-from ..core.logger import SecurityLogger
+from dataclasses import dataclass, field
+from datetime import datetime
+from enum import Enum
+
+# Add these imports at the top
+from typing import Any, Dict, List, Optional, Set, Union
+
 from ..core.exceptions import SecurityError
+from ..core.logger import SecurityLogger
 
 
 class PrivacyLevel(Enum):

src/llmguardian/defenders/__init__.py

@@ -2,11 +2,11 @@
 defenders/__init__.py - Security defenders initialization
 """
 
+from .content_filter import ContentFilter
+from .context_validator import ContextValidator
 from .input_sanitizer import InputSanitizer
 from .output_validator import OutputValidator
 from .token_validator import TokenValidator
-from .content_filter import ContentFilter
-from .context_validator import ContextValidator
 
 __all__ = [
     "InputSanitizer",

src/llmguardian/defenders/content_filter.py

@@ -3,11 +3,12 @@ defenders/content_filter.py - Content filtering and moderation
 """
 
 import re
-from typing import Dict, List, Optional, Any, Set
 from dataclasses import dataclass
 from enum import Enum
-from ..core.logger import SecurityLogger
+from typing import Any, Dict, List, Optional, Set
+
 from ..core.exceptions import ValidationError
+from ..core.logger import SecurityLogger
 
 
 class ContentCategory(Enum):

src/llmguardian/defenders/context_validator.py

@@ -2,12 +2,13 @@
 defenders/context_validator.py - Context validation for LLM interactions
 """
 
-from typing import Dict, Optional, List, Any
+import hashlib
 from dataclasses import dataclass
 from datetime import datetime
-import hashlib
-from ..core.logger import SecurityLogger
+from typing import Any, Dict, List, Optional
+
 from ..core.exceptions import ValidationError
+from ..core.logger import SecurityLogger
 
 
 @dataclass

src/llmguardian/defenders/input_sanitizer.py

@@ -3,10 +3,11 @@ defenders/input_sanitizer.py - Input sanitization for LLM inputs
 """
 
 import re
-from typing import Dict, Any, List, Optional
 from dataclasses import dataclass
-from ..core.logger import SecurityLogger
+from typing import Any, Dict, List, Optional
+
 from ..core.exceptions import ValidationError
+from ..core.logger import SecurityLogger
 
 
 @dataclass

src/llmguardian/defenders/output_validator.py

@@ -3,10 +3,11 @@ defenders/output_validator.py - Output validation and sanitization
 """
 
 import re
-from typing import Dict, List, Optional, Set, Any
 from dataclasses import dataclass
-from ..core.logger import SecurityLogger
+from typing import Any, Dict, List, Optional, Set
+
 from ..core.exceptions import ValidationError
+from ..core.logger import SecurityLogger
 
 
 @dataclass

src/llmguardian/defenders/test_context_validator.py

@@ -2,10 +2,12 @@
 tests/defenders/test_context_validator.py - Tests for context validation
 """
 
-import pytest
 from datetime import datetime, timedelta
-from llmguardian.defenders.context_validator import ContextValidator, ValidationResult
+
+import pytest
+
 from llmguardian.core.exceptions import ValidationError
+from llmguardian.defenders.context_validator import ContextValidator, ValidationResult
 
 
 @pytest.fixture

src/llmguardian/defenders/token_validator.py

@@ -2,13 +2,15 @@
 defenders/token_validator.py - Token and credential validation
 """
 
-from typing import Dict, Optional, Any, List
-from dataclasses import dataclass
 import re
-import jwt
+from dataclasses import dataclass
 from datetime import datetime, timedelta
-from ..core.logger import SecurityLogger
+from typing import Any, Dict, List, Optional
+
+import jwt
+
 from ..core.exceptions import TokenValidationError
+from ..core.logger import SecurityLogger
 
 
 @dataclass

src/llmguardian/monitors/__init__.py

@@ -2,11 +2,11 @@
 monitors/__init__.py - Monitoring system initialization
 """
 
-from .usage_monitor import UsageMonitor
+from .audit_monitor import AuditMonitor
 from .behavior_monitor import BehaviorMonitor
-from .threat_detector import ThreatDetector
 from .performance_monitor import PerformanceMonitor
-from .audit_monitor import AuditMonitor
+from .threat_detector import ThreatDetector
+from .usage_monitor import UsageMonitor
 
 __all__ = [
     "UsageMonitor",

src/llmguardian/monitors/audit_monitor.py

@@ -3,15 +3,16 @@ monitors/audit_monitor.py - Audit trail and compliance monitoring
 """
 
 import json
-from typing import Dict, List, Optional, Any, Set
+import threading
+from collections import defaultdict
 from dataclasses import dataclass
 from datetime import datetime, timedelta
 from enum import Enum
-import threading
 from pathlib import Path
-from collections import defaultdict
-from ..core.logger import SecurityLogger
+from typing import Any, Dict, List, Optional, Set
+
 from ..core.exceptions import MonitoringError
+from ..core.logger import SecurityLogger
 
 
 class AuditEventType(Enum):

src/llmguardian/monitors/behavior_monitor.py

@@ -2,11 +2,12 @@
 monitors/behavior_monitor.py - LLM behavior monitoring
 """
 
-from typing import Dict, List, Optional, Any
 from dataclasses import dataclass
 from datetime import datetime
-from ..core.logger import SecurityLogger
+from typing import Any, Dict, List, Optional
+
 from ..core.exceptions import MonitoringError
+from ..core.logger import SecurityLogger
 
 
 @dataclass

src/llmguardian/monitors/performance_monitor.py

@@ -2,15 +2,16 @@
 monitors/performance_monitor.py - LLM performance monitoring
 """
 
-import time
 import threading
-from typing import Dict, List, Optional, Any
+import time
+from collections import deque
 from dataclasses import dataclass
 from datetime import datetime, timedelta
 from statistics import mean, median, stdev
-from collections import deque
-from ..core.logger import SecurityLogger
+from typing import Any, Dict, List, Optional
+
 from ..core.exceptions import MonitoringError
+from ..core.logger import SecurityLogger
 
 
 @dataclass

src/llmguardian/monitors/threat_detector.py

@@ -2,14 +2,15 @@
 monitors/threat_detector.py - Real-time threat detection for LLM applications
 """
 
-from typing import Dict, List, Optional, Set, Any
+import threading
+from collections import defaultdict
 from dataclasses import dataclass
 from datetime import datetime, timedelta
 from enum import Enum
-import threading
-from collections import defaultdict
-from ..core.logger import SecurityLogger
+from typing import Any, Dict, List, Optional, Set
+
 from ..core.exceptions import MonitoringError
+from ..core.logger import SecurityLogger
 
 
 class ThreatLevel(Enum):

src/llmguardian/monitors/usage_monitor.py

@@ -2,14 +2,16 @@
 monitors/usage_monitor.py - Resource usage monitoring
 """
 
-import time
-import psutil
 import threading
-from typing import Dict, List, Optional
+import time
 from dataclasses import dataclass
 from datetime import datetime
-from ..core.logger import SecurityLogger
+from typing import Dict, List, Optional
+
+import psutil
+
 from ..core.exceptions import MonitoringError
+from ..core.logger import SecurityLogger
 
 
 @dataclass

src/llmguardian/scanners/prompt_injection_scanner.py

@@ -3,12 +3,12 @@ LLMGuardian Prompt Injection Scanner
 Core module for detecting and preventing prompt injection attacks in LLM applications.
 """
 
+import logging
 import re
+from abc import ABC, abstractmethod
 from dataclasses import dataclass
 from enum import Enum
-from typing import List, Optional, Dict, Tuple
-import logging
-from abc import ABC, abstractmethod
+from typing import Dict, List, Optional, Tuple
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)

src/llmguardian/vectors/__init__.py

@@ -3,8 +3,8 @@ vectors/__init__.py - Vector security initialization
 """
 
 from .embedding_validator import EmbeddingValidator
-from .vector_scanner import VectorScanner
 from .retrieval_guard import RetrievalGuard
 from .storage_validator import StorageValidator
+from .vector_scanner import VectorScanner
 
 __all__ = ["EmbeddingValidator", "VectorScanner", "RetrievalGuard", "StorageValidator"]

src/llmguardian/vectors/embedding_validator.py

@@ -2,13 +2,15 @@
 vectors/embedding_validator.py - Embedding validation and security
 """
 
-import numpy as np
-from typing import Dict, List, Optional, Any, Tuple
+import hashlib
 from dataclasses import dataclass
 from datetime import datetime
-import hashlib
-from ..core.logger import SecurityLogger
+from typing import Any, Dict, List, Optional, Tuple
+
+import numpy as np
+
 from ..core.exceptions import ValidationError
+from ..core.logger import SecurityLogger
 
 
 @dataclass

src/llmguardian/vectors/retrieval_guard.py

@@ -2,16 +2,18 @@
 vectors/retrieval_guard.py - Security for Retrieval-Augmented Generation (RAG) operations
 """
 
-import numpy as np
-from typing import Dict, List, Optional, Any, Tuple, Set
-from dataclasses import dataclass
-from datetime import datetime
-from enum import Enum
 import hashlib
 import re
 from collections import defaultdict
-from ..core.logger import SecurityLogger
+from dataclasses import dataclass
+from datetime import datetime
+from enum import Enum
+from typing import Any, Dict, List, Optional, Set, Tuple
+
+import numpy as np
+
 from ..core.exceptions import SecurityError
+from ..core.logger import SecurityLogger
 
 
 class RetrievalRisk(Enum):

src/llmguardian/vectors/storage_validator.py

@@ -2,16 +2,18 @@
 vectors/storage_validator.py - Vector storage security validation
 """
 
-import numpy as np
-from typing import Dict, List, Optional, Any, Tuple, Set
-from dataclasses import dataclass
-from datetime import datetime
-from enum import Enum
 import hashlib
 import json
 from collections import defaultdict
-from ..core.logger import SecurityLogger
+from dataclasses import dataclass
+from datetime import datetime
+from enum import Enum
+from typing import Any, Dict, List, Optional, Set, Tuple
+
+import numpy as np
+
 from ..core.exceptions import SecurityError
+from ..core.logger import SecurityLogger
 
 
 class StorageRisk(Enum):

src/llmguardian/vectors/vector_scanner.py

@@ -2,15 +2,17 @@
 vectors/vector_scanner.py - Security scanner for vector databases and operations
 """
 
-import numpy as np
-from typing import Dict, List, Optional, Any, Set
+import hashlib
+from collections import defaultdict
 from dataclasses import dataclass
 from datetime import datetime
 from enum import Enum
-import hashlib
-from collections import defaultdict
-from ..core.logger import SecurityLogger
+from typing import Any, Dict, List, Optional, Set
+
+import numpy as np
+
 from ..core.exceptions import SecurityError
+from ..core.logger import SecurityLogger
 
 
 class VectorVulnerability(Enum):

tests/conftest.py

@@ -2,13 +2,15 @@
 tests/conftest.py - Pytest configuration and shared fixtures
 """
 
-import pytest
-import os
 import json
+import os
 from pathlib import Path
-from typing import Dict, Any
-from llmguardian.core.logger import SecurityLogger
+from typing import Any, Dict
+
+import pytest
+
 from llmguardian.core.config import Config
+from llmguardian.core.logger import SecurityLogger
 
 
 @pytest.fixture(scope="session")

tests/data/test_privacy_guard.py

@@ -2,17 +2,19 @@
 tests/data/test_privacy_guard.py - Test cases for privacy protection functionality
 """
 
-import pytest
 from datetime import datetime
 from unittest.mock import Mock, patch
+
+import pytest
+
+from llmguardian.core.exceptions import SecurityError
 from llmguardian.data.privacy_guard import (
-    PrivacyGuard,
-    PrivacyRule,
-    PrivacyLevel,
     DataCategory,
     PrivacyCheck,
+    PrivacyGuard,
+    PrivacyLevel,
+    PrivacyRule,
 )
-from llmguardian.core.exceptions import SecurityError
 
 
 @pytest.fixture

tests/unit/test_prompt_injection_scanner.py

@@ -2,12 +2,13 @@
 Tests for the LLMGuardian Prompt Injection Scanner
 """
 
-import pytest
 from typing import List
+
+import pytest
 from prompt_injection_scanner import (
-    PromptInjectionScanner,
     InjectionPattern,
     InjectionType,
+    PromptInjectionScanner,
     ScanResult,
 )
 

tests/utils/test_utils.py

@@ -4,7 +4,8 @@ tests/utils/test_utils.py - Testing utilities and helpers
 
 import json
 from pathlib import Path
-from typing import Dict, Any, Optional
+from typing import Any, Dict, Optional
+
 import numpy as np