diff --git a/features/steps/provider_registry_steps.py b/features/steps/provider_registry_steps.py index 1ed18b531..a3a26907d 100644 --- a/features/steps/provider_registry_steps.py +++ b/features/steps/provider_registry_steps.py @@ -728,6 +728,11 @@ def step_impl_result_provider_type_anthropic(context: Any) -> None: assert context.result == _ensure_provider_type(context).ANTHROPIC +@then("the provider registry result should be ProviderType.GEMINI") +def step_impl_result_provider_type_gemini(context: Any) -> None: + assert context.result == _ensure_provider_type(context).GEMINI + + @then('the provider registry result should be "{value}"') def step_impl_result_string(context: Any, value: str) -> None: assert context.result == value, f"Expected {value!r}, got {context.result!r}"