Add tests for LlmSession and LlmSessionCaptionsAndKeywords in both integration and mocked scenarios

2026-10-02 11:20:16 +02:00
parent 906707de55
commit ff86f08ae7
9 changed files with 994 additions and 29 deletions
@@ -12,9 +12,11 @@ def test_photo():

@pytest.fixture
 def llm_session():
-    return LlmSession(
-        system_prompt=(
-            "You are a helpful assistent to photographer. "
-            'Answer in JSON format: `{"caption": "...", "tags": ["tag1", "tag2", ...]}`'
-        )
-    )
+    return LlmSession(system_prompt="You are a helpful assistant.")
+
+
+@pytest.fixture
+def llm_session_captions_and_keywords():
+    from photools3.llm_session import LlmSessionCaptionsAndKeywords
+
+    return LlmSessionCaptionsAndKeywords()