Setting the response model so the cost can be calculated

2025-09-18 16:24:03 -06:00 · 2025-09-18 16:24:03 -06:00 · 0e045e0bb7
commit 0e045e0bb7
parent eb71611a97
1 changed files with 3 additions and 0 deletions
--- 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