mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-06 08:16:43 +00:00
075a089d82
* use is_batch_jsonl_file * fix valid_content_type * fix transform_create_file_request * fix _transform_openai_jsonl_content_to_bedrock_jsonl_content * test_transform_openai_jsonl_content_to_bedrock_jsonl_content * fix mypy linting errors * fix BEDROCK_BATCH_MODEL * fix working sample * fix comment * fix model list * fix: use with managed batches * refactor
27 lines
899 B
Python
27 lines
899 B
Python
from typing import Optional
|
|
|
|
from litellm.types.llms.openai import CreateFileRequest
|
|
from litellm.types.utils import ExtractedFileData
|
|
|
|
|
|
class FilesAPIUtils:
|
|
"""
|
|
Utils for files API interface on litellm
|
|
"""
|
|
@staticmethod
|
|
def is_batch_jsonl_file(create_file_data: CreateFileRequest, extracted_file_data: ExtractedFileData) -> bool:
|
|
"""
|
|
Check if the file is a batch jsonl file
|
|
"""
|
|
return (
|
|
create_file_data.get("purpose") == "batch"
|
|
and FilesAPIUtils.valid_content_type(extracted_file_data.get("content_type"))
|
|
and extracted_file_data.get("content") is not None
|
|
)
|
|
|
|
@staticmethod
|
|
def valid_content_type(content_type: Optional[str]) -> bool:
|
|
"""
|
|
Check if the content type is valid
|
|
"""
|
|
return content_type in set(["application/jsonl", "application/octet-stream"])
|