phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 18:24:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(fix) use isinstance to check types

Browse source
This commit is contained in:
ishaan-jaff 2024-01-30 08:31:21 -08:00
parent b9a117fa1d
commit f941c57688
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/llms/bedrock.py
View file

@ -702,7 +702,7 @@ def _embedding_func_single(
encoding=None,
logging_obj=None,
):
if type(input) != str:
if isinstance(input, str) is False:
raise BedrockError(
message="Bedrock Embedding API input must be type str | List[str]",
status_code=400,
@ -800,7 +800,8 @@ def embedding(
aws_role_name=aws_role_name,
aws_session_name=aws_session_name,
)
if type(input) == str:
if isinstance(input, str):
## Embedding Call
embeddings = [
_embedding_func_single(
model,
@ -810,8 +811,8 @@ def embedding(
logging_obj=logging_obj,
)
]
elif type(input) == list:
## Embedding Call
elif isinstance(input, list):
## Embedding Call - assuming this is a List[str]
embeddings = [
_embedding_func_single(
model,