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

(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)
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"
if response_ms:
_response_ms = response_ms
@ -343,7 +343,7 @@ class EmbeddingResponse(OpenAIObject):
_response_ms = None
data = []
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):
# Define custom behavior for the 'in' operator