scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

style: black-format batch_embed_content_transformation.py

Browse source 
CI flagged this file as needing reformat after the merge resolution.
This commit is contained in:
Chesars 2026-04-25 17:30:56 -03:00
parent 2255f7651f
commit 91a4d34990
1 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
litellm/llms/vertex_ai/gemini_embeddings/batch_embed_content_transformation.py
View file

@ -130,7 +130,9 @@ def _is_multimodal_input(input: EmbeddingInput) -> bool:
for element in input:
if isinstance(element, list):
if any(_is_multimodal_element(sub) for sub in element if isinstance(sub, str)):
if any(
_is_multimodal_element(sub) for sub in element if isinstance(sub, str)
):
return True
elif isinstance(element, str) and _is_multimodal_element(element):
return True
@ -363,13 +365,16 @@ def process_response(
for e in input_list:
if isinstance(e, list):
text_elements.extend(
sub for sub in e
sub
for sub in e
if isinstance(sub, str) and not _is_multimodal_element(sub)
)
elif isinstance(e, str) and not _is_multimodal_element(e):
text_elements.append(e)
if text_elements:
input_text = get_formatted_prompt(data={"input": text_elements}, call_type="embedding")
input_text = get_formatted_prompt(
data={"input": text_elements}, call_type="embedding"
)
prompt_tokens = token_counter(model=model, text=input_text)
else:
prompt_tokens = 0