Setting the response model so the cost can be calculated

This commit is contained in:
Eddie Richter 2025-09-18 16:24:03 -06:00
parent eb71611a97
commit 0e045e0bb7

View file

@ -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