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

(fix) bedrock - embedding - support str input

Browse source
This commit is contained in:
ishaan-jaff 2024-01-11 23:02:12 +05:30
parent a876748bf5
commit a9d812eb8d
1 changed files with 22 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/llms/bedrock.py
View file

@ -737,7 +737,17 @@ def embedding(
aws_region_name=aws_region_name, aws_region_name=aws_region_name,
aws_bedrock_runtime_endpoint=aws_bedrock_runtime_endpoint, aws_bedrock_runtime_endpoint=aws_bedrock_runtime_endpoint,
) )
if type(input) == str:
embeddings = [
_embedding_func_single(
model,
input,
optional_params=optional_params,
client=client,
logging_obj=logging_obj,
)
]
else:
## Embedding Call ## Embedding Call
embeddings = [ embeddings = [
_embedding_func_single( _embedding_func_single(