generated from john/python-template
This commit is contained in:
@@ -81,8 +81,20 @@ class TestWorkflowReliability:
|
||||
top_p=None,
|
||||
settings=None,
|
||||
provider=None,
|
||||
source_reference=None,
|
||||
requested_model=None,
|
||||
):
|
||||
_ = (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 TimeoutError("simulated provider timeout")
|
||||
|
||||
monkeypatch.setattr("transcription.services.workflows.transcribe_document_image", _never_returns)
|
||||
@@ -146,9 +158,9 @@ class TestWorkflowReliability:
|
||||
budget_seconds = 0.20
|
||||
real_build = workflows_module.build_provider_input
|
||||
|
||||
def _slow_build(source_arg):
|
||||
def _slow_build(source_arg, **kwargs):
|
||||
time.sleep(setup_seconds)
|
||||
return real_build(source_arg)
|
||||
return real_build(source_arg, **kwargs)
|
||||
|
||||
async def _never_returns(*args, **kwargs):
|
||||
_ = (args, kwargs)
|
||||
|
||||
Reference in New Issue
Block a user