From 4eaa70440a247cee2767bd16e7dc830da559105a Mon Sep 17 00:00:00 2001 From: Sujith Date: Tue, 14 Jul 2026 15:49:46 +0530 Subject: [PATCH] fix(main): use PEP 604 unions for new store and prompt_cache_key params --- litellm/main.py | 8 ++++---- litellm/utils.py | 4 ++-- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/litellm/main.py b/litellm/main.py index 81b02082abd..3b13e04bbe3 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -435,8 +435,8 @@ async def acompletion( verbosity: Optional[Literal["low", "medium", "high"]] = None, safety_identifier: Optional[str] = None, service_tier: Optional[str] = None, - store: Optional[bool] = None, - prompt_cache_key: Optional[str] = None, + store: bool | None = None, + prompt_cache_key: str | None = None, # set api_base, api_version, api_key base_url: Optional[str] = None, api_version: Optional[str] = None, @@ -4832,8 +4832,8 @@ def completion( # type: ignore extra_headers: Optional[dict] = None, safety_identifier: Optional[str] = None, service_tier: Optional[str] = None, - store: Optional[bool] = None, - prompt_cache_key: Optional[str] = None, + store: bool | None = None, + prompt_cache_key: str | None = None, # soon to be deprecated params by OpenAI functions: Optional[List] = None, function_call: Optional[str] = None, diff --git a/litellm/utils.py b/litellm/utils.py index b16ecdf88be..bc5b2447761 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -3791,8 +3791,8 @@ def get_optional_params( thinking: Optional[AnthropicThinkingParam] = None, web_search_options: Optional[OpenAIWebSearchOptions] = None, safety_identifier: Optional[str] = None, - store: Optional[bool] = None, - prompt_cache_key: Optional[str] = None, + store: bool | None = None, + prompt_cache_key: str | None = None, base_model: Optional[str] = None, **kwargs, ):