Llm client quality (#14)
Tests / test_and_lint (push) Successful in 14s

Reviewed-on: #14
Co-authored-by: Mathias Scheider <mathias2784@web.de>
This commit was merged in pull request #14.
This commit is contained in:
2026-10-02 21:31:21 +00:00
committed by mathias2784
parent 419fb46c3d
commit 401488fa7e
8 changed files with 220 additions and 71 deletions
+5 -5
View File
@@ -7,7 +7,7 @@ from photools3.llm_session import LlmSession
not LlmSession(system_prompt="").is_api_reachable(), reason="API not available"
)
def test_llm_session(llm_session, test_photo):
llm_session.system_prompt = "Return only JSON format: {'keywords': ['...']}"
llm_session.system_prompt = "Return only JSON format: {'text': '...'}"
result = llm_session.call(
prompt="keywords for the attached picture", photos=[test_photo]
)
@@ -21,7 +21,7 @@ def test_llm_session_captions_and_keywords(
llm_session_captions_and_keywords, test_photo
):
result = llm_session_captions_and_keywords.call(photos=[test_photo])
assert "caption" in result
assert "tags" in result
assert isinstance(result["caption"], str)
assert isinstance(result["tags"], list)
assert result.caption
assert result.tags
assert isinstance(result.caption, str)
assert isinstance(result.tags, list)