mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-03 02:22:24 +00:00
style: apply black formatting to embedding.py
This commit is contained in:
parent
91b06f23e1
commit
d410787dca
1 changed files with 2 additions and 2 deletions
|
|
@ -227,7 +227,7 @@ class BedrockEmbedding(BaseAWSLLM):
|
|||
elif model == "amazon.titan-embed-g1-text-02":
|
||||
returned_response = AmazonTitanG1Config()._transform_response(
|
||||
response_list=response_list, model=model
|
||||
)
|
||||
)
|
||||
elif provider == "twelvelabs":
|
||||
returned_response = (
|
||||
TwelveLabsMarengoEmbeddingConfig()._transform_response(
|
||||
|
|
@ -472,7 +472,7 @@ class BedrockEmbedding(BaseAWSLLM):
|
|||
elif model == "amazon.titan-embed-g1-text-02":
|
||||
transformed_request = AmazonTitanG1Config()._transform_request(
|
||||
input=i, inference_params=inference_params
|
||||
)
|
||||
)
|
||||
else:
|
||||
raise Exception(
|
||||
"Unmapped model. Received={}. Expected={}".format(
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue