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

fix(utils.py): fix embedding response output parsing

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-25 12:06:47 -08:00
parent 62d8f9ad2a
commit dac76a4861
3 changed files with 102 additions and 53 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/llms/openai.py
View file

@ -343,11 +343,8 @@ class OpenAIChatCompletion(BaseLLM):
additional_args={"complete_input_dict": data},
original_response=response,
)
model_response.data = response.data
model_response.model = model
model_response.usage = response.usage
model_response.object = "list"
return model_response
return convert_to_model_response_object(response_object=json.loads(response.model_dump_json()), model_response_object=model_response, response_type="embedding")
except OpenAIError as e:
exception_mapping_worked = True
raise e