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

fix(utils.py): fix linting issues

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-25 14:11:40 -08:00
parent 8970f12780
commit a070bee5be
1 changed files with 10 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/utils.py
View file

@ -354,11 +354,18 @@ class Embedding(OpenAIObject):
setattr(self, key, value)
class EmbeddingResponse(OpenAIObject):
model: Optional[str] = None
"""The model used for embedding."""
data: Optional[Embedding] = None
"""The actual embedding value"""
object: str
"""The object type, which is always "embedding" """
usage: Optional[Usage] = None
"""Usage statistics for the completion request."""
"""Usage statistics for the embedding request."""
def __init__(self, model=None, usage=None, stream=False, response_ms=None, data=None):
object = "list"
if response_ms:
@ -3317,8 +3324,8 @@ def convert_to_model_response_object(response_object: Optional[dict]=None, model
model_response_object.object = response_object["object"]
data = []
for idx, embedding in enumerate(response_object["data"]):
embedding_obj = Embedding(
for idx, embedding in enumerate(response_object["data"]): # type: ignore
embedding_obj = Embedding( # type: ignore
embedding=embedding.get("embedding", None),
index = embedding.get("index", idx),
object=embedding.get("object", "embedding")