diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py index 240f35b76eb..f86f7ac776f 100644 --- a/litellm/proxy/_types.py +++ b/litellm/proxy/_types.py @@ -3400,12 +3400,12 @@ class LiteLLM_SpendLogs_ResponseObject(LiteLLMPydanticObjectBase): class TokenCountRequest(LiteLLMPydanticObjectBase): model: str prompt: str | None = None - messages: list[dict] | None = None + messages: list[dict[Any, Any]] | None = None """ Anthropic token counting endpoint uses /messages """ - contents: list[dict] | None = None + contents: list[dict[Any, Any]] | None = None """ Google /countTokens endpoint expects contents to be a list of dicts with the following structure: """