Merge pull request #22699 from BerriAI/fix/mypy-count-tokens-signature

Fix mypy override errors in count_tokens signatures
This commit is contained in:
Julio Quinteros Pro
2026-03-03 18:28:14 -03:00
committed by GitHub
2 changed files with 4 additions and 2 deletions
+2 -1
View File
@@ -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
+2 -1
View File
@@ -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