mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 11:14:04 +00:00
c7f14e936a
* remove unused imports * fix AmazonConverseConfig * fix test * fix import * ruff check fixes * test fixes * fix testing * fix imports
14 lines
469 B
Python
14 lines
469 B
Python
from typing import List
|
|
|
|
|
|
def is_tokens_or_list_of_tokens(value: List):
|
|
# Check if it's a list of integers (tokens)
|
|
if isinstance(value, list) and all(isinstance(item, int) for item in value):
|
|
return True
|
|
# Check if it's a list of lists of integers (list of tokens)
|
|
if isinstance(value, list) and all(
|
|
isinstance(item, list) and all(isinstance(i, int) for i in item)
|
|
for item in value
|
|
):
|
|
return True
|
|
return False
|