phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(utils.py): fix embedding response object conversion

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-25 14:25:06 -08:00
parent e43e1e9ab1
commit 2eb7386095
2 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/utils.py
View file

@ -3330,10 +3330,9 @@ def convert_to_model_response_object(response_object: Optional[dict]=None, model
index = embedding.get("index", idx),
object=embedding.get("object", "embedding")
)
if isinstance(embedding_obj, EmbeddingResponse):
embedding_data.append(embedding_obj)
if len(embedding_data) > 0:
model_response_object.data = embedding_data
embedding_data.append(embedding_obj)
model_response_object.data = embedding_data
if "usage" in response_object and response_object["usage"] is not None:
model_response_object.usage.completion_tokens = response_object["usage"].get("completion_tokens", 0) # type: ignore