From 4eb9221121d357dafafeed0381fc58747d51ed46 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 18 Jan 2024 12:29:21 -0800 Subject: [PATCH] (fix) track EmbeddingResponse cost --- litellm/utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/utils.py b/litellm/utils.py index 1ff9dbe1c81..77f0c331e9b 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -8478,6 +8478,8 @@ def get_logging_payload(kwargs, response_obj, start_time, end_time): if param in payload and type(payload[param]) != Json: if type(payload[param]) == ModelResponse: payload[param] = payload[param].model_dump_json() + if type(payload[param]) == EmbeddingResponse: + payload[param] = payload[param].model_dump_json() elif type(payload[param]) == Usage: payload[param] = payload[param].model_dump_json() else: