From 9a1aebc14673a6150c7d47c3d0c2d797f697ff92 Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Tue, 1 Sep 2026 13:50:21 +0000
Subject: [PATCH] fix(registry): declare databricks deepseek cache-write rate
 at the input rate per repo convention

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 litellm/model_prices_and_context_window_backup.json             | 2 ++
 model_prices_and_context_window.json                            | 2 ++
 .../llms/databricks/test_databricks_cost_calculator.py          | 2 ++
 3 files changed, 6 insertions(+)

diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json
index fae9699541a..dd45495884d 100644
--- a/litellm/model_prices_and_context_window_backup.json
+++ b/litellm/model_prices_and_context_window_backup.json
@@ -15108,6 +15108,7 @@
         "supports_vision": true
     },
     "databricks/databricks-deepseek-v4-flash-0731": {
+        "cache_creation_input_token_cost": 1.4e-07,
         "cache_read_input_token_cost": 2.8e-08,
         "input_cost_per_token": 1.4e-07,
         "input_dbu_cost_per_token": 2e-06,
@@ -15135,6 +15136,7 @@
         "supports_vision": false
     },
     "databricks/databricks-deepseek-v4-pro-0813": {
+        "cache_creation_input_token_cost": 1.31999e-06,
         "cache_read_input_token_cost": 1.3202e-07,
         "input_cost_per_token": 1.31999e-06,
         "input_dbu_cost_per_token": 1.8857e-05,
diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json
index fae9699541a..dd45495884d 100644
--- a/model_prices_and_context_window.json
+++ b/model_prices_and_context_window.json
@@ -15108,6 +15108,7 @@
         "supports_vision": true
     },
     "databricks/databricks-deepseek-v4-flash-0731": {
+        "cache_creation_input_token_cost": 1.4e-07,
         "cache_read_input_token_cost": 2.8e-08,
         "input_cost_per_token": 1.4e-07,
         "input_dbu_cost_per_token": 2e-06,
@@ -15135,6 +15136,7 @@
         "supports_vision": false
     },
     "databricks/databricks-deepseek-v4-pro-0813": {
+        "cache_creation_input_token_cost": 1.31999e-06,
         "cache_read_input_token_cost": 1.3202e-07,
         "input_cost_per_token": 1.31999e-06,
         "input_dbu_cost_per_token": 1.8857e-05,
diff --git a/tests/test_litellm/llms/databricks/test_databricks_cost_calculator.py b/tests/test_litellm/llms/databricks/test_databricks_cost_calculator.py
index 29ad8ee4b6e..e72642f7a04 100644
--- a/tests/test_litellm/llms/databricks/test_databricks_cost_calculator.py
+++ b/tests/test_litellm/llms/databricks/test_databricks_cost_calculator.py
@@ -62,6 +62,8 @@ PUBLISHED_DBU_PER_MILLION: Final = {
     "databricks/databricks-gemini-2-5-pro": ("22.321", "178.571", "22.321", "2.232"),
     "databricks/databricks-gemini-2-5-flash": ("5.357", "44.643", "5.357", "0.536"),
     "databricks/databricks-kimi-k3": ("42.857", "214.286", "42.857", "4.286"),
+    "databricks/databricks-deepseek-v4-flash-0731": ("2.000", "4.000", "2.000", "0.400"),
+    "databricks/databricks-deepseek-v4-pro-0813": ("18.857", "56.571", "18.857", "1.886"),
     "databricks/databricks-glm-5-2": ("20.000", "62.857", "20.000", "3.714"),
 }
 PROMOTIONAL_DISCOUNT: Final = 0.80