mathias2784
906707de55
Tests / test_and_lint (push) Successful in 11s
Reviewed-on: #11 Co-authored-by: Mathias Scheider <mathias2784@web.de>
21 lines
466 B
Python
21 lines
466 B
Python
import pytest
|
|
|
|
from photools3.llm_session import LlmSession
|
|
from photools3.photo import Photo
|
|
from test.test_photo import PATHS_DNG_EXAMPLES
|
|
|
|
|
|
@pytest.fixture
|
|
def test_photo():
|
|
return Photo(PATHS_DNG_EXAMPLES[0])
|
|
|
|
|
|
@pytest.fixture
|
|
def llm_session():
|
|
return LlmSession(
|
|
system_prompt=(
|
|
"You are a helpful assistent to photographer. "
|
|
'Answer in JSON format: `{"caption": "...", "tags": ["tag1", "tag2", ...]}`'
|
|
)
|
|
)
|