generated from john/python-template
This commit is contained in:
@@ -99,6 +99,7 @@ class TestPipelineSuccessFlow:
|
||||
provider=None,
|
||||
source_reference=None,
|
||||
requested_model=None,
|
||||
evidence_capture=None,
|
||||
) -> TranscriptionResult:
|
||||
_ = (
|
||||
image_path,
|
||||
@@ -110,6 +111,7 @@ class TestPipelineSuccessFlow:
|
||||
provider,
|
||||
source_reference,
|
||||
requested_model,
|
||||
evidence_capture,
|
||||
)
|
||||
return TranscriptionResult(
|
||||
text="Pipeline transcript",
|
||||
@@ -191,9 +193,20 @@ class TestPipelineSuccessFlow:
|
||||
provider=None,
|
||||
source_reference=None,
|
||||
requested_model=None,
|
||||
evidence_capture=None,
|
||||
) -> TranscriptionResult:
|
||||
page_name = Path(image_path).name
|
||||
_ = (prompt_name, prompt_text, temperature, top_p, settings, provider, source_reference, requested_model)
|
||||
_ = (
|
||||
prompt_name,
|
||||
prompt_text,
|
||||
temperature,
|
||||
top_p,
|
||||
settings,
|
||||
provider,
|
||||
source_reference,
|
||||
requested_model,
|
||||
evidence_capture,
|
||||
)
|
||||
return TranscriptionResult(
|
||||
text=f"Transcript for {page_name}",
|
||||
provider="openrouter",
|
||||
@@ -260,6 +273,7 @@ class TestPipelineSuccessFlow:
|
||||
provider=None,
|
||||
source_reference=None,
|
||||
requested_model=None,
|
||||
evidence_capture=None,
|
||||
) -> TranscriptionResult:
|
||||
nonlocal call_count
|
||||
call_count += 1
|
||||
@@ -273,6 +287,7 @@ class TestPipelineSuccessFlow:
|
||||
provider,
|
||||
source_reference,
|
||||
requested_model,
|
||||
evidence_capture,
|
||||
)
|
||||
if call_count == 2:
|
||||
raise RuntimeError("simulated page failure")
|
||||
@@ -351,6 +366,7 @@ class TestPipelineSuccessFlow:
|
||||
provider=None,
|
||||
source_reference=None,
|
||||
requested_model=None,
|
||||
evidence_capture=None,
|
||||
) -> TranscriptionResult:
|
||||
nonlocal call_count
|
||||
_ = (
|
||||
@@ -363,6 +379,7 @@ class TestPipelineSuccessFlow:
|
||||
provider,
|
||||
source_reference,
|
||||
requested_model,
|
||||
evidence_capture,
|
||||
)
|
||||
call_count += 1
|
||||
return TranscriptionResult(
|
||||
@@ -419,6 +436,7 @@ class TestPipelineFailureFlow:
|
||||
provider=None,
|
||||
source_reference=None,
|
||||
requested_model=None,
|
||||
evidence_capture=None,
|
||||
) -> TranscriptionResult:
|
||||
_ = (
|
||||
image_path,
|
||||
@@ -430,6 +448,7 @@ class TestPipelineFailureFlow:
|
||||
provider,
|
||||
source_reference,
|
||||
requested_model,
|
||||
evidence_capture,
|
||||
)
|
||||
raise RuntimeError("pipeline provider failure")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user