Fix code formatting (black and isort)

Author: sbenodiz

custom_components/ai_agent_ha/agent.py

@@ -59,10 +59,20 @@ def sanitize_for_logging(data: Any, mask: str = "***REDACTED***") -> Any:
     """
     # Sensitive field patterns (case-insensitive)
     sensitive_patterns = {
-        'token', 'key', 'password', 'secret', 'credential',
-        'auth', 'authorization', 'api_key', 'apikey',
-        'llama_token', 'openai_token', 'gemini_token',
-        'anthropic_token', 'openrouter_token'
+        "token",
+        "key",
+        "password",
+        "secret",
+        "credential",
+        "auth",
+        "authorization",
+        "api_key",
+        "apikey",
+        "llama_token",
+        "openai_token",
+        "gemini_token",
+        "anthropic_token",
+        "openrouter_token",
     }
 
     if isinstance(data, dict):
@@ -190,7 +200,10 @@ async def get_response(self, messages, **kwargs):
                     )
                     _LOGGER.debug("Local API response status: %d", resp.status)
                     # Sanitize headers to avoid logging any auth tokens
-                    _LOGGER.debug("Local API response headers: %s", sanitize_for_logging(dict(resp.headers)))
+                    _LOGGER.debug(
+                        "Local API response headers: %s",
+                        sanitize_for_logging(dict(resp.headers)),
+                    )
 
                     # Try to parse as JSON
                     try:
@@ -2216,7 +2229,9 @@ async def process_query(
 
             _LOGGER.debug(f"Processing query with provider: {provider}")
             # Log sanitized config (masks all tokens/keys for security)
-            _LOGGER.debug(f"Using config: {json.dumps(sanitize_for_logging(config), default=str)}")
+            _LOGGER.debug(
+                f"Using config: {json.dumps(sanitize_for_logging(config), default=str)}"
+            )
 
             selected_provider = provider or config.get("ai_provider", "llama")
             models_config = config.get("models", {})

tests/test_ai_agent_ha/test_sanitization.py

@@ -1,7 +1,9 @@
 """Tests for the sanitization utility function."""
-import pytest
-import sys
+
 import os
+import sys
+
+import pytest
 
 # Add the parent directory to the path for direct imports
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", ".."))
@@ -28,18 +30,30 @@ def sanitize_for_logging(data: Any, mask: str = "***REDACTED***") -> Any:
         """
         # Sensitive field patterns (case-insensitive)
         sensitive_patterns = {
-            'token', 'key', 'password', 'secret', 'credential',
-            'auth', 'authorization', 'api_key', 'apikey',
-            'llama_token', 'openai_token', 'gemini_token',
-            'anthropic_token', 'openrouter_token'
+            "token",
+            "key",
+            "password",
+            "secret",
+            "credential",
+            "auth",
+            "authorization",
+            "api_key",
+            "apikey",
+            "llama_token",
+            "openai_token",
+            "gemini_token",
+            "anthropic_token",
+            "openrouter_token",
         }
 
         if isinstance(data, dict):
             sanitized = {}
             for key, value in data.items():
                 # Check if key matches any sensitive pattern
                 key_lower = str(key).lower()
-                is_sensitive = any(pattern in key_lower for pattern in sensitive_patterns)
+                is_sensitive = any(
+                    pattern in key_lower for pattern in sensitive_patterns
+                )
 
                 if is_sensitive:
                     sanitized[key] = mask
@@ -82,7 +96,7 @@ def test_sanitize_multiple_tokens(self):
             "gemini_token": "gm-xyz789",
             "anthropic_token": "sk-ant-456",
             "ai_provider": "openai",
-            "model": "gpt-4"
+            "model": "gpt-4",
         }
         result = sanitize_for_logging(data)
 
@@ -119,11 +133,8 @@ def test_sanitize_authorization(self):
     def test_sanitize_nested_dict(self):
         """Test sanitization of nested dictionaries."""
         data = {
-            "config": {
-                "openai_token": "sk-abc123",
-                "model": "gpt-4"
-            },
-            "ai_provider": "openai"
+            "config": {"openai_token": "sk-abc123", "model": "gpt-4"},
+            "ai_provider": "openai",
         }
         result = sanitize_for_logging(data)
 
@@ -135,7 +146,7 @@ def test_sanitize_list_of_dicts(self):
         """Test sanitization of lists containing dictionaries."""
         data = [
             {"api_key": "key1", "name": "config1"},
-            {"api_key": "key2", "name": "config2"}
+            {"api_key": "key2", "name": "config2"},
         ]
         result = sanitize_for_logging(data)
 
@@ -146,10 +157,7 @@ def test_sanitize_list_of_dicts(self):
 
     def test_sanitize_tuple(self):
         """Test sanitization of tuples."""
-        data = (
-            {"token": "secret", "name": "test"},
-            "plain_value"
-        )
+        data = ({"token": "secret", "name": "test"}, "plain_value")
         result = sanitize_for_logging(data)
 
         assert isinstance(result, tuple)
@@ -163,7 +171,7 @@ def test_sanitize_case_insensitive(self):
             "Token": "secret1",
             "API_KEY": "secret2",
             "Password": "secret3",
-            "AUTHORIZATION": "secret4"
+            "AUTHORIZATION": "secret4",
         }
         result = sanitize_for_logging(data)
 
@@ -177,7 +185,7 @@ def test_sanitize_partial_match(self):
         data = {
             "my_token": "secret",
             "custom_api_key": "secret2",
-            "db_password": "secret3"
+            "db_password": "secret3",
         }
         result = sanitize_for_logging(data)
 
@@ -213,18 +221,17 @@ def test_sanitize_deeply_nested(self):
                 "level2": {
                     "level3": {
                         "api_key": "secret",
-                        "config": {
-                            "token": "secret2",
-                            "safe_value": "ok"
-                        }
+                        "config": {"token": "secret2", "safe_value": "ok"},
                     }
                 }
             }
         }
         result = sanitize_for_logging(data)
 
         assert result["level1"]["level2"]["level3"]["api_key"] == "***REDACTED***"
-        assert result["level1"]["level2"]["level3"]["config"]["token"] == "***REDACTED***"
+        assert (
+            result["level1"]["level2"]["level3"]["config"]["token"] == "***REDACTED***"
+        )
         assert result["level1"]["level2"]["level3"]["config"]["safe_value"] == "ok"
 
     def test_sanitize_real_config(self):
@@ -240,8 +247,8 @@ def test_sanitize_real_config(self):
             "models": {
                 "openai": "gpt-4",
                 "gemini": "gemini-pro",
-                "anthropic": "claude-3-sonnet"
-            }
+                "anthropic": "claude-3-sonnet",
+            },
         }
         result = sanitize_for_logging(config)
 
@@ -265,7 +272,7 @@ def test_sanitize_http_headers(self):
             "Content-Type": "application/json",
             "Authorization": "Bearer secret-token",
             "X-API-Key": "api-key-123",
-            "User-Agent": "HomeAssistant/1.0"
+            "User-Agent": "HomeAssistant/1.0",
         }
         result = sanitize_for_logging(headers)