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) OpenAI embedding

Browse source
This commit is contained in:
ishaan-jaff 2023-11-29 08:44:46 -08:00
parent 3c254cc555
commit 2bbd9c063d
1 changed files with 1 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/utils.py
View file

@ -3335,16 +3335,8 @@ def convert_to_model_response_object(response_object: Optional[dict]=None, model
if "object" in response_object: if "object" in response_object:
model_response_object.object = response_object["object"] model_response_object.object = response_object["object"]
embedding_data = []
for idx, embedding in enumerate(response_object["data"]):
embedding_obj = Embedding(
embedding=embedding.get("embedding", None),
index = embedding.get("index", idx),
object=embedding.get("object", "embedding")
)
embedding_data.append(embedding_obj)
model_response_object.data = embedding_data model_response_object.data = response_object["data"]
if "usage" in response_object and response_object["usage"] is not None: 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 model_response_object.usage.completion_tokens = response_object["usage"].get("completion_tokens", 0) # type: ignore