mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-06 08:16:43 +00:00
Setting the response model so the cost can be calculated
This commit is contained in:
parent
eb71611a97
commit
0e045e0bb7
1 changed files with 3 additions and 0 deletions
|
|
@ -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
|
||||
|
||||
Loading…
Add table
Reference in a new issue