mirror of
https://github.com/tiennm99/litellm.git
synced 2026-07-31 00:24:24 +00:00
Merge pull request #22699 from BerriAI/fix/mypy-count-tokens-signature
Fix mypy override errors in count_tokens signatures
This commit is contained in:
@@ -166,7 +166,8 @@ class GoogleAIStudioTokenCounter(BaseTokenCounter):
|
||||
contents: Optional[List[Dict[str, Any]]],
|
||||
deployment: Optional[Dict[str, Any]] = None,
|
||||
request_model: str = "",
|
||||
**kwargs,
|
||||
tools: Optional[List[Dict[str, Any]]] = None,
|
||||
system: Optional[Any] = None,
|
||||
) -> Optional[TokenCountResponse]:
|
||||
import copy
|
||||
|
||||
|
||||
@@ -1030,7 +1030,8 @@ class VertexAITokenCounter(BaseTokenCounter):
|
||||
contents: Optional[List[Dict[str, Any]]],
|
||||
deployment: Optional[Dict[str, Any]] = None,
|
||||
request_model: str = "",
|
||||
**kwargs,
|
||||
tools: Optional[List[Dict[str, Any]]] = None,
|
||||
system: Optional[Any] = None,
|
||||
) -> Optional[TokenCountResponse]:
|
||||
import copy
|
||||
|
||||
|
||||
Reference in New Issue
Block a user