scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-29 06:33:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 200

fix(openrouter): annotate untyped arguments to eliminate basedpyright breaches

Browse source
This commit is contained in:
yyouretoast 2026-10-16 18:54:31 +03:00
parent 5ce51cf5f0
commit 69b006cf01
3 changed files with 12 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/openrouter/chat/transformation.py
View file

@ -68,8 +68,8 @@ class OpenrouterConfig(OpenAIGPTConfig):
resolved_base: Final = resolve_openrouter_api_base(
api_base=api_base,
region=self.region,
litellm_params=litellm_params,
optional_params=optional_params,
litellm_params=litellm_params, # pyright: ignore[reportUnknownArgumentType] # overrides untyped OpenAIGPTConfig signature
optional_params=optional_params, # pyright: ignore[reportUnknownArgumentType] # overrides untyped OpenAIGPTConfig signature
).rstrip("/")
endpoint: Final = "chat/completions"
if endpoint in resolved_base:

10
litellm/llms/openrouter/data_residency.py
View file

@ -39,8 +39,8 @@ _OPENROUTER_REGIONAL_HOSTNAMES: Final[Mapping[str, str]] = MappingProxyType(
_DEFAULT_OPENROUTER_API_BASE: Final[str] = "https://openrouter.ai/api/v1"
def _extract_region_value(params: object) -> str | None:
if not isinstance(params, Mapping):
def _extract_region_value(params: Mapping[str, object] | None) -> str | None:
if params is None:
return None
for key in ("region", "region_name", "data_residency"):
val: object = params.get(key)
@ -51,9 +51,9 @@ def _extract_region_value(params: object) -> str | None:
def resolve_openrouter_api_base(
api_base: str | None = None,
region: object = None,
litellm_params: object = None,
optional_params: object = None,
region: str | None = None,
litellm_params: Mapping[str, object] | None = None,
optional_params: Mapping[str, object] | None = None,
) -> str:
"""
Resolve the OpenRouter API base URL based on custom api_base, explicit region, or environment.

8
litellm/main.py
View file

@ -3417,8 +3417,9 @@ def _complete_openrouter(ctx: CompletionDispatchContext) -> _CompletionDispatchR
api_base=api_base,
region=getattr(ctx.provider_config, "region", None) or getattr(litellm.OpenrouterConfig, "region", None),
litellm_params=litellm_params,
optional_params=optional_params,
optional_params=optional_params, # pyright: ignore[reportUnknownArgumentType] # untyped optional_params in dispatch context
)
litellm_params["api_base"] = api_base
litellm_params["data_residency"] = infer_openrouter_data_residency("openrouter", api_base)
@ -6791,9 +6792,10 @@ def embedding(
api_base = resolve_openrouter_api_base( # rebind-ok: regional endpoint resolution for openrouter
api_base=api_base,
region=getattr(litellm.OpenrouterConfig, "region", None),
litellm_params=litellm_params_dict,
optional_params=optional_params,
litellm_params=litellm_params_dict, # pyright: ignore[reportUnknownArgumentType] # untyped litellm_params in embedding
optional_params=optional_params, # pyright: ignore[reportUnknownArgumentType] # untyped optional_params in embedding
)
litellm_params_dict["api_base"] = api_base
litellm_params_dict["data_residency"] = infer_openrouter_data_residency("openrouter", api_base)