scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-03 02:22:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 82

style(gemini): format count_tokens request body per ruff format

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-09-23 13:02:01 +00:00
parent 4e25825b91
commit 3bc391c16e
1 changed files with 13 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
litellm/llms/gemini/count_tokens/handler.py
View file

@ -141,10 +141,20 @@ class GoogleAIStudioTokenCounter:
"generateContentRequest": { # mutable-ok: httpx json body takes a plain dict
"model": f"models/{model}",
"contents": cleaned_contents,
**( # mutable-ok: httpx json body takes a plain dict
{"systemInstruction": system_instruction} if system_instruction is not None else {}
**(
{ # mutable-ok: httpx json body takes a plain dict
"systemInstruction": system_instruction,
}
if system_instruction is not None
else {} # mutable-ok: httpx json body takes a plain dict
),
**(
{ # mutable-ok: httpx json body takes a plain dict
"tools": tools,
}
if tools is not None
else {} # mutable-ok: httpx json body takes a plain dict
),
**({"tools": tools} if tools is not None else {}), # mutable-ok: httpx json body takes a plain dict
}
}
)