bbchops/transcription
1
0
Fork 0
generated from john/python-template
Code Issues 4 Pull Requests Actions Packages Projects Releases Wiki Activity

V4.5 Complete - Enhanced trancription context, added option to restranscribe source under different models.

Browse Source
This commit is contained in:
Jim Lancaster
2026-08-16 09:06:56 -05:00
parent bdb1b31b0a
commit 7054cd8af9
28 changed files with 2883 additions and 1485 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/integration/test_pipeline_flow.py
+17 -9
View File
@@ -1,9 +1,11 @@
"""Integration tests for end-to-end upload and worker pipeline behavior."""
import io
from pathlib import Path
from uuid import uuid4
import pytest
from PIL import Image
from transcription.config import Settings
from transcription.db.models import Document
@@ -18,6 +20,12 @@ from transcription.services.store import create_job_for_document
from transcription.services.workflows import advance_job
def _jpeg_bytes(color: str = "white") -> bytes:
output = io.BytesIO()
Image.new("RGB", (2, 2), color=color).save(output, format="JPEG")
return output.getvalue()
def _build_services(default_session_factory) -> ServiceBundle:
services = ServiceBundle()
object.__setattr__(
@@ -56,7 +64,7 @@ class TestPipelineSuccessFlow:
)
upload_result = await create_document_job(
filename="pipeline.jpg",
file_bytes=b"pipeline-bytes",
file_bytes=_jpeg_bytes(),
session=async_session,
settings=settings,
)
@@ -141,9 +149,9 @@ class TestPipelineSuccessFlow:
create_result = await create_job_for_document(
document_id=document.id,
source_files=[
("page-01.jpg", b"one"),
("page-02.jpg", b"two"),
("page-03.jpg", b"three"),
("page-01.jpg", _jpeg_bytes("white")),
("page-02.jpg", _jpeg_bytes("gray")),
("page-03.jpg", _jpeg_bytes("black")),
],
session=async_session,
settings=settings,
@@ -206,8 +214,8 @@ class TestPipelineSuccessFlow:
create_result = await create_job_for_document(
document_id=document.id,
source_files=[
("page-01.jpg", b"one"),
("page-02.jpg", b"two"),
("page-01.jpg", _jpeg_bytes("white")),
("page-02.jpg", _jpeg_bytes("gray")),
],
session=async_session,
settings=settings,
@@ -272,8 +280,8 @@ class TestPipelineSuccessFlow:
create_result = await create_job_for_document(
document_id=document.id,
source_files=[
("page-01.jpg", b"one"),
("page-02.jpg", b"two"),
("page-01.jpg", _jpeg_bytes("white")),
("page-02.jpg", _jpeg_bytes("gray")),
],
session=async_session,
settings=settings,
@@ -346,7 +354,7 @@ class TestPipelineFailureFlow:
settings = Settings(openrouter_api_key="test-key", upload_dir=tmp_path)
upload_result = await create_document_job(
filename="pipeline.jpg",
file_bytes=b"pipeline-bytes",
file_bytes=_jpeg_bytes(),
session=async_session,
settings=settings,
)