black and ruff

test/test_llm_session_mocked.py

@@ -1,8 +1,7 @@
-import pytest
 from unittest.mock import MagicMock, patch
+
 from photools3.llm_session import LlmSession
-from photools3.photo import Photo
-from test.test_photo import PATHS_DNG_EXAMPLES
+
 
 def test_llm_session_mocked(test_photo):
     # We want to mock OpenAI client's chat.completions.create
@@ -10,22 +9,24 @@ def test_llm_session_mocked(test_photo):
         # Setup the mock response structure
         mock_client = MagicMock()
         mocked_openai.return_value = mock_client
-        
+
         # Mock the models list to return a proper object with an ID string
         # This prevents Pydantic ValidationError when LlmSession tries to use model_id
         mock_model = MagicMock()
         mock_model.id = "gpt-4o"
         mock_client.models.list.return_value.data = [mock_model]
-        
+
         # Create a response object: response.choices[0].message.content
         mock_response = MagicMock()
         mock_choice = MagicMock()
         mock_message = MagicMock()
-        
-        mock_message.content = '{"caption": "A beautiful sunset", "tags": ["sunset", "nature"]}'
+
+        mock_message.content = (
+            '{"caption": "A beautiful sunset", "tags": ["sunset", "nature"]}'
+        )
         mock_choice.message = mock_message
         mock_response.choices = [mock_choice]
-        
+
         # Configure the client to return our mock response
         mock_client.chat.completions.create.return_value = mock_response
 
@@ -41,6 +42,6 @@ def test_llm_session_mocked(test_photo):
         assert result["caption"] == "A beautiful sunset"
         assert "sunset" in result["tags"]
         assert "nature" in result["tags"]
-        
+
         # Verify that the mock was called
         assert mock_client.chat.completions.create.called