Ruff

Author: dicksontsai

tests/test_client.py

@@ -20,7 +20,9 @@ async def _test():
             ) as mock_process:
                 # Mock the async generator
                 async def mock_generator():
-                    yield AssistantMessage(content=[TextBlock(text="4")], model="claude-opus-4-1-20250805")
+                    yield AssistantMessage(
+                        content=[TextBlock(text="4")], model="claude-opus-4-1-20250805"
+                    )
 
                 mock_process.return_value = mock_generator()
 
@@ -43,7 +45,10 @@ async def _test():
             ) as mock_process:
 
                 async def mock_generator():
-                    yield AssistantMessage(content=[TextBlock(text="Hello!")], model="claude-opus-4-1-20250805")
+                    yield AssistantMessage(
+                        content=[TextBlock(text="Hello!")],
+                        model="claude-opus-4-1-20250805",
+                    )
 
                 mock_process.return_value = mock_generator()
 

tests/test_types.py

@@ -32,7 +32,9 @@ def test_assistant_message_with_text(self):
     def test_assistant_message_with_thinking(self):
         """Test creating an AssistantMessage with thinking content."""
         thinking_block = ThinkingBlock(thinking="I'm thinking...", signature="sig-123")
-        msg = AssistantMessage(content=[thinking_block], model="claude-opus-4-1-20250805")
+        msg = AssistantMessage(
+            content=[thinking_block], model="claude-opus-4-1-20250805"
+        )
         assert len(msg.content) == 1
         assert msg.content[0].thinking == "I'm thinking..."
         assert msg.content[0].signature == "sig-123"