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

(feat) clean out junk params from litellm embedding

Browse source
This commit is contained in:
ishaan-jaff 2023-11-22 13:50:25 -08:00
parent b0801f61e6
commit e8ff4d5eca
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/utils.py
View file

@ -335,7 +335,7 @@ class ModelResponse(OpenAIObject):
setattr(self, key, value) setattr(self, key, value)
class EmbeddingResponse(OpenAIObject): class EmbeddingResponse(OpenAIObject):
def __init__(self, id=None, choices=None, created=None, model=None, usage=None, stream=False, response_ms=None): def __init__(self, model=None, usage=None, stream=False, response_ms=None):
object = "list" object = "list"
if response_ms: if response_ms:
_response_ms = response_ms _response_ms = response_ms
@ -343,7 +343,7 @@ class EmbeddingResponse(OpenAIObject):
_response_ms = None _response_ms = None
data = [] data = []
model = model model = model
super().__init__(id=id, choices=choices, created=created, model=model, object=object, data=data, usage=usage) super().__init__(model=model, object=object, data=data, usage=usage)
def __contains__(self, key): def __contains__(self, key):
# Define custom behavior for the 'in' operator # Define custom behavior for the 'in' operator