scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-10 08:28:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

refactor(token_counter): read function and custom tool call text through one lookup

Browse source
This commit is contained in:
John Karam 2026-10-29 13:19:34 +00:00
parent bcbf2b7a3d
commit fb5414393e
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
litellm/litellm_core_utils/token_counter.py
View file

@ -571,11 +571,12 @@ def _count_function_call_tokens(
total = 0
for tool_call in value:
if "function" in tool_call:
total += count_function(str(tool_call["function"].get("arguments", "")))
kind, field = "function", "arguments"
elif "custom" in tool_call:
total += count_function(str(tool_call["custom"].get("input", "")))
kind, field = "custom", "input"
else:
raise ValueError(f"Unsupported tool call {tool_call} must contain a function or custom key")
total += count_function(str(tool_call[kind].get(field, "")))
return total
if key == "function_call":
if not isinstance(value, Mapping):