File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed
pydantic_ai_slim/pydantic_ai/models Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -678,7 +678,7 @@ def override_allow_model_requests(allow_model_requests: bool) -> Iterator[None]:
678678
679679
680680def 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/' ):
You can’t perform that action at this time.
0 commit comments