diff --git a/litellm/llms/lemonade/chat/transformation.py b/litellm/llms/lemonade/chat/transformation.py index 5e675066b0a..a8707cce571 100644 --- a/litellm/llms/lemonade/chat/transformation.py +++ b/litellm/llms/lemonade/chat/transformation.py @@ -106,5 +106,8 @@ class LemonadeChatConfig(OpenAILikeChatConfig): api_key=api_key, ) + # Storing lemonade in the model response for easier cost calculation later + setattr(model_response, "model", "lemonade/" + model) + return model_response \ No newline at end of file