lint

Author: slkoo-cc

pydantic_ai_slim/pydantic_ai/models/__init__.py

@@ -9,12 +9,12 @@
 import base64
 import warnings
 from abc import ABC, abstractmethod
-from collections.abc import AsyncIterator, Iterator
+from collections.abc import AsyncIterator, Callable, Iterator
 from contextlib import asynccontextmanager, contextmanager
 from dataclasses import dataclass, field, replace
 from datetime import datetime
 from functools import cache, cached_property
-from typing import Any, Generic, Literal, TypeVar, Callable, overload
+from typing import Any, Generic, Literal, TypeVar, overload
 
 import httpx
 from typing_extensions import TypeAliasType, TypedDict
@@ -47,7 +47,7 @@
 )
 from ..output import OutputMode
 from ..profiles import DEFAULT_PROFILE, ModelProfile, ModelProfileSpec
-from ..providers import infer_provider, Provider
+from ..providers import Provider, infer_provider
 from ..settings import ModelSettings, merge_model_settings
 from ..tools import ToolDefinition
 from ..usage import RequestUsage
@@ -677,7 +677,9 @@ def override_allow_model_requests(allow_model_requests: bool) -> Iterator[None]:
         ALLOW_MODEL_REQUESTS = old_value  # pyright: ignore[reportConstantRedefinition]
 
 
-def infer_model(model: Model | KnownModelName | str, provider_generator: Callable[[str], Provider[Any]] | None = None) -> Model:  # noqa: C901
+def infer_model(
+    model: Model | KnownModelName | str, provider_generator: Callable[[str], Provider[Any]] | None = None
+) -> Model:  # noqa: C901
     """Infer the model from the name. May optionally pass a callable that setup a custom provider for the model."""
     if isinstance(model, Model):
         return model
@@ -714,7 +716,7 @@ def infer_model(model: Model | KnownModelName | str, provider_generator: Callabl
         provider_name = 'google-vertex'
 
     if provider_generator is None:
-         provider_generator = infer_provider
+        provider_generator = infer_provider
     provider = provider_generator(provider_name)
 
     model_kind = provider_name