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

fix linting error

Browse source
This commit is contained in:
Ishaan Jaffer 2025-10-04 11:40:47 -07:00
parent 7a41c09529
commit cdfb53dd81
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/llms/bedrock/embed/twelvelabs_marengo_transformation.py
View file

@ -6,9 +6,10 @@ Why separate file? Make it easy to see how transformation works
Docs - https://docs.aws.amazon.com/bedrock/latest/userguide/model-parameters-marengo.html
"""
from typing import List, Optional, Union
from typing import List, Optional, Union, cast
from litellm.types.llms.bedrock import (
TWELVELABS_EMBEDDING_INPUT_TYPES,
TwelveLabsAsyncInvokeRequest,
TwelveLabsMarengoEmbeddingRequest,
TwelveLabsOutputDataConfig,
@ -90,7 +91,10 @@ class TwelveLabsMarengoEmbeddingConfig:
- S3 URLs for all media types (async-invoke only)
"""
# Get input_type or default to "text"
input_type = inference_params.get("inputType") or inference_params.get("input_type") or "text"
input_type = cast(
TWELVELABS_EMBEDDING_INPUT_TYPES,
inference_params.get("inputType") or inference_params.get("input_type") or "text"
)
# Validate that async-invoke is used for video/audio
if input_type in ["video", "audio"] and not async_invoke_route: