From e8ff4d5eca447eb3e6aa1a3ff47140e993a52664 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 22 Nov 2023 13:50:25 -0800 Subject: [PATCH] (feat) clean out junk params from litellm embedding --- litellm/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/utils.py b/litellm/utils.py index 5d4ef74b21..073c1ee194 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -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