fix(pc):

tests/unit/llm_providers/test_anthropic_provider.py

@@ -5,7 +5,11 @@ from unittest.mock import MagicMock, AsyncMock, patch
 from inline_snapshot import snapshot
 
 from agentic_security.llm_providers.anthropic_provider import AnthropicProvider
-from agentic_security.llm_providers.base import LLMMessage, LLMProviderError, LLMRateLimitError
+from agentic_security.llm_providers.base import (
+    LLMMessage,
+    LLMProviderError,
+    LLMRateLimitError,
+)
 
 
 class TestAnthropicProviderInit:
@@ -209,13 +213,19 @@ class TestAnthropicProviderErrors:
 
     def test_handle_rate_limit_error(self, provider):
         import anthropic
+
         with pytest.raises(LLMRateLimitError):
-            provider._handle_error(anthropic.RateLimitError("rate limited", response=MagicMock(), body={}))
+            provider._handle_error(
+                anthropic.RateLimitError("rate limited", response=MagicMock(), body={})
+            )
 
     def test_handle_api_error(self, provider):
         import anthropic
+
         with pytest.raises(LLMProviderError):
-            provider._handle_error(anthropic.APIError("api error", request=MagicMock(), body={}))
+            provider._handle_error(
+                anthropic.APIError("api error", request=MagicMock(), body={})
+            )
 
     def test_handle_generic_error(self, provider):
         with pytest.raises(LLMProviderError):

tests/unit/llm_providers/test_factory.py

@@ -9,7 +9,11 @@ from agentic_security.llm_providers.factory import (
     list_providers,
     register_provider,
 )
-from agentic_security.llm_providers.base import BaseLLMProvider, LLMProviderError, LLMResponse
+from agentic_security.llm_providers.base import (
+    BaseLLMProvider,
+    LLMProviderError,
+    LLMResponse,
+)
 
 
 class TestListProviders:
@@ -26,11 +30,13 @@ class TestListProviders:
 class TestGetProviderClass:
     def test_get_openai(self):
         from agentic_security.llm_providers.openai_provider import OpenAIProvider
+
         cls = get_provider_class("openai")
         assert cls is OpenAIProvider
 
     def test_get_anthropic(self):
         from agentic_security.llm_providers.anthropic_provider import AnthropicProvider
+
         cls = get_provider_class("anthropic")
         assert cls is AnthropicProvider
 

tests/unit/llm_providers/test_openai_provider.py

@@ -5,7 +5,11 @@ from unittest.mock import MagicMock, AsyncMock, patch
 from inline_snapshot import snapshot
 
 from agentic_security.llm_providers.openai_provider import OpenAIProvider
-from agentic_security.llm_providers.base import LLMMessage, LLMProviderError, LLMRateLimitError
+from agentic_security.llm_providers.base import (
+    LLMMessage,
+    LLMProviderError,
+    LLMRateLimitError,
+)
 
 
 class TestOpenAIProviderInit:
@@ -111,7 +115,9 @@ class TestOpenAIProviderSync:
         mock_response.usage = None
 
         with patch.object(provider, "_get_client") as mock_client:
-            mock_client.return_value.chat.completions.create.return_value = mock_response
+            mock_client.return_value.chat.completions.create.return_value = (
+                mock_response
+            )
             result = provider.sync_generate("Hello")
             assert result.content == snapshot("Response")
 
@@ -126,7 +132,9 @@ class TestOpenAIProviderSync:
         messages = [LLMMessage(role="user", content="Hi")]
 
         with patch.object(provider, "_get_client") as mock_client:
-            mock_client.return_value.chat.completions.create.return_value = mock_response
+            mock_client.return_value.chat.completions.create.return_value = (
+                mock_response
+            )
             result = provider.sync_chat(messages)
             assert result.content == snapshot("Chat response")
 
@@ -196,8 +204,11 @@ class TestOpenAIProviderErrors:
 
     def test_handle_rate_limit_error(self, provider):
         import openai
+
         with pytest.raises(LLMRateLimitError):
-            provider._handle_error(openai.RateLimitError("rate limited", response=MagicMock(), body={}))
+            provider._handle_error(
+                openai.RateLimitError("rate limited", response=MagicMock(), body={})
+            )
 
     def test_handle_generic_error(self, provider):
         with pytest.raises(LLMProviderError):