diff --git a/litellm/types/integrations/anthropic_cache_control_hook.py b/litellm/types/integrations/anthropic_cache_control_hook.py index 6e859f10187..83e5a9e7f01 100644 --- a/litellm/types/integrations/anthropic_cache_control_hook.py +++ b/litellm/types/integrations/anthropic_cache_control_hook.py @@ -16,4 +16,13 @@ class CacheControlMessageInjectionPoint(TypedDict): control: Optional[ChatCompletionCachedContent] -CacheControlInjectionPoint = CacheControlMessageInjectionPoint +class CacheControlToolConfigInjectionPoint(TypedDict): + """Type for tool_config-level injection points (Bedrock).""" + + location: Literal["tool_config"] + + +CacheControlInjectionPoint = Union[ + CacheControlMessageInjectionPoint, + CacheControlToolConfigInjectionPoint, +]