generated from john/python-template
model used being carried thru
This commit is contained in:
@@ -1,5 +1,11 @@
|
||||
import pytest
|
||||
|
||||
from transcription.services.base import ServiceBase
|
||||
|
||||
|
||||
class FakeService(ServiceBase):
|
||||
"""A fake service class for testing"""
|
||||
|
||||
|
||||
class TestServiceBase:
|
||||
class TestInitialization:
|
||||
|
||||
@@ -7,8 +7,9 @@ import pytest
|
||||
|
||||
from transcription.services.transcription import transcribe_document_image
|
||||
|
||||
|
||||
HAS_OPENROUTER_KEY = bool(os.getenv("OPENROUTER_API_KEY"))
|
||||
HAS_OPENROUTER_KEY = bool(
|
||||
os.getenv("TRANSCRIPTION_OPENROUTER_API_KEY") or os.getenv("OPENROUTER_API_KEY")
|
||||
)
|
||||
|
||||
REAL_IMAGES_DIR = Path(__file__).resolve().parents[1] / "fixtures" / "images" / "real"
|
||||
ARTIFACTS_DIR = Path(__file__).resolve().parents[1] / "artifacts" / "transcriptions"
|
||||
@@ -18,7 +19,10 @@ pytestmark = [
|
||||
pytest.mark.external,
|
||||
pytest.mark.skipif(
|
||||
not HAS_OPENROUTER_KEY,
|
||||
reason="Set OPENROUTER_API_KEY to run external real-image tests.",
|
||||
reason=(
|
||||
"Set TRANSCRIPTION_OPENROUTER_API_KEY "
|
||||
"(or legacy OPENROUTER_API_KEY) to run external real-image tests."
|
||||
),
|
||||
),
|
||||
]
|
||||
|
||||
@@ -67,4 +71,4 @@ class TestRealImageExternalTranscription:
|
||||
f"{result.text}\n"
|
||||
)
|
||||
artifact_path.write_text(artifact_text, encoding="utf-8")
|
||||
assert artifact_path.exists()
|
||||
assert artifact_path.exists()
|
||||
|
||||
Reference in New Issue
Block a user