From 0e045e0bb774e4060073ccb52ab64695f68ec40b Mon Sep 17 00:00:00 2001
From: Eddie Richter <eddie.richter@amd.com>
Date: Thu, 18 Sep 2025 16:24:03 -0600
Subject: [PATCH] Setting the response model so the cost can be calculated

---
 litellm/llms/lemonade/chat/transformation.py | 3 +++
 1 file changed, 3 insertions(+)

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