Add tests for LlmSession and LlmSessionCaptionsAndKeywords in both integration and mocked scenarios (#12)
Tests / test_and_lint (push) Successful in 13s
Tests / test_and_lint (push) Successful in 13s
Reviewed-on: #12 Co-authored-by: Mathias Scheider <mathias2784@web.de>
This commit was merged in pull request #12.
This commit is contained in:
+8
-6
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user