src/photools3/cli.py

@@ -69,6 +69,7 @@ def _run_keywords_extraction(file_path: Path) -> Any:
     photo = Photo(file_path)
     return session.call([photo])
 
+
 def _run_qa(file_path: Path) -> Any:
     """Handles the core logic of loading the photo and calling the LLM."""
     session = LlmSessionQuality()

src/photools3/llm_session.py

@@ -138,6 +138,7 @@ class LlmSessionCaptionsAndKeywords(LlmSession):
             model_id=model_id,
         )
 
+
 class LlmSessionQuality(LlmSession):
 
     def __init__(self, temperature: float = settings.llm_default_temperature):
@@ -163,7 +164,7 @@ class LlmSessionQuality(LlmSession):
                 "  - `exposure`: severly under- or over-exposed.\n"
                 "  - `style`: photo has severe style issues (beyond paersonal taste)\n"
             ),
-                temperature=temperature
+            temperature=temperature,
         )
 
     def call(self, photos: list[Photo], model_id=None):
@@ -172,4 +173,3 @@ class LlmSessionQuality(LlmSession):
             photos=photos,
             model_id=model_id,
         )
-        

src/photools3/models.py

@@ -102,7 +102,7 @@ class LlmResponseCaptionsAndKeywords(BaseModel):
             "A list of keywords that properly tag the photo. "
             "Allowed keywords: "
             + str(load_valid_keywords())
-            .replace("'", "\"")
+            .replace("'", '"')
             .replace("{", "[")
             .replace("}", "]")
         )

test/test_llm_session_mocked.py

@@ -61,7 +61,9 @@ def test_llm_session_captions_and_keywords_mocked(test_photo):
         mock_choice = MagicMock()
         mock_message = MagicMock()
 
-        mock_message.content = '{"caption": "A mocked caption", "tags": ["architektur"]}'
+        mock_message.content = (
+            '{"caption": "A mocked caption", "tags": ["architektur"]}'
+        )
         mock_choice.message = mock_message
         mock_response.choices = [mock_choice]