Skip to content

Commit 703c9b4

Browse files
slkoo-ccDouweM
andauthored
Apply suggestions from code review
Co-authored-by: Douwe Maan <[email protected]>
1 parent a905a13 commit 703c9b4

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

pydantic_ai_slim/pydantic_ai/models/__init__.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -678,7 +678,7 @@ def override_allow_model_requests(allow_model_requests: bool) -> Iterator[None]:
678678

679679

680680
def infer_model( # noqa: C901
681-
model: Model | KnownModelName | str, provider_generator: Callable[[str], Provider[Any]] | None = None
681+
model: Model | KnownModelName | str, provider_factory: Callable[[str], Provider[Any]] = infer_provider
682682
) -> Model:
683683
"""Infer the model from the name. May optionally pass a callable that setup a custom provider for the model."""
684684
if isinstance(model, Model):
@@ -715,9 +715,7 @@ def infer_model( # noqa: C901
715715
)
716716
provider_name = 'google-vertex'
717717

718-
if provider_generator is None:
719-
provider_generator = infer_provider
720-
provider = provider_generator(provider_name)
718+
provider = provider_factory(provider_name)
721719

722720
model_kind = provider_name
723721
if model_kind.startswith('gateway/'):

0 commit comments

Comments
 (0)