(fix) dynamo db test - new model_spend params

2024-02-17 16:00:08 -08:00 · 2024-02-17 16:00:08 -08:00 · 44202aa25b
commit 44202aa25b
parent 35af2d666b
1 changed files with 2 additions and 0 deletions
--- a/litellm/proxy/db/dynamo_db.py
+++ b/litellm/proxy/db/dynamo_db.py
@ -287,6 +287,8 @@ class DynamoDBWrapper(CustomDB):
                            or k == "config"
                            or k == "metadata"
                            or k == "permissions"
+                            or k == "model_spend"
+                            or k == "model_max_budget"
                        )
                        and v is not None
                        and isinstance(v, str)