generated from john/python-template
This commit is contained in:
@@ -99,8 +99,20 @@ class TestPipelineSuccessFlow:
|
||||
top_p=None,
|
||||
settings=None,
|
||||
provider=None,
|
||||
source_reference=None,
|
||||
requested_model=None,
|
||||
) -> TranscriptionResult:
|
||||
_ = (image_path, prompt_name, prompt_text, temperature, top_p, settings, provider)
|
||||
_ = (
|
||||
image_path,
|
||||
prompt_name,
|
||||
prompt_text,
|
||||
temperature,
|
||||
top_p,
|
||||
settings,
|
||||
provider,
|
||||
source_reference,
|
||||
requested_model,
|
||||
)
|
||||
return TranscriptionResult(
|
||||
text="Pipeline transcript",
|
||||
provider="openrouter",
|
||||
@@ -181,9 +193,11 @@ class TestPipelineSuccessFlow:
|
||||
top_p=None,
|
||||
settings=None,
|
||||
provider=None,
|
||||
source_reference=None,
|
||||
requested_model=None,
|
||||
) -> TranscriptionResult:
|
||||
page_name = Path(image_path).name
|
||||
_ = (prompt_name, prompt_text, temperature, top_p, settings, provider)
|
||||
_ = (prompt_name, prompt_text, temperature, top_p, settings, provider, source_reference, requested_model)
|
||||
return TranscriptionResult(
|
||||
text=f"Transcript for {page_name}",
|
||||
provider="openrouter",
|
||||
@@ -248,10 +262,22 @@ class TestPipelineSuccessFlow:
|
||||
top_p=None,
|
||||
settings=None,
|
||||
provider=None,
|
||||
source_reference=None,
|
||||
requested_model=None,
|
||||
) -> TranscriptionResult:
|
||||
nonlocal call_count
|
||||
call_count += 1
|
||||
_ = (image_path, prompt_name, prompt_text, temperature, top_p, settings, provider)
|
||||
_ = (
|
||||
image_path,
|
||||
prompt_name,
|
||||
prompt_text,
|
||||
temperature,
|
||||
top_p,
|
||||
settings,
|
||||
provider,
|
||||
source_reference,
|
||||
requested_model,
|
||||
)
|
||||
if call_count == 2:
|
||||
raise RuntimeError("simulated page failure")
|
||||
return TranscriptionResult(
|
||||
@@ -327,9 +353,21 @@ class TestPipelineSuccessFlow:
|
||||
top_p=None,
|
||||
settings=None,
|
||||
provider=None,
|
||||
source_reference=None,
|
||||
requested_model=None,
|
||||
) -> TranscriptionResult:
|
||||
nonlocal call_count
|
||||
_ = (image_path, prompt_name, prompt_text, temperature, top_p, settings, provider)
|
||||
_ = (
|
||||
image_path,
|
||||
prompt_name,
|
||||
prompt_text,
|
||||
temperature,
|
||||
top_p,
|
||||
settings,
|
||||
provider,
|
||||
source_reference,
|
||||
requested_model,
|
||||
)
|
||||
call_count += 1
|
||||
return TranscriptionResult(
|
||||
text="new transcript",
|
||||
@@ -383,8 +421,20 @@ class TestPipelineFailureFlow:
|
||||
top_p=None,
|
||||
settings=None,
|
||||
provider=None,
|
||||
source_reference=None,
|
||||
requested_model=None,
|
||||
) -> TranscriptionResult:
|
||||
_ = (image_path, prompt_name, prompt_text, temperature, top_p, settings, provider)
|
||||
_ = (
|
||||
image_path,
|
||||
prompt_name,
|
||||
prompt_text,
|
||||
temperature,
|
||||
top_p,
|
||||
settings,
|
||||
provider,
|
||||
source_reference,
|
||||
requested_model,
|
||||
)
|
||||
raise RuntimeError("pipeline provider failure")
|
||||
|
||||
monkeypatch.setattr(
|
||||
|
||||
Reference in New Issue
Block a user