From 57501ad20ece909c278000fea870c2c66fce1d5c Mon Sep 17 00:00:00 2001
From: Kaustubh Maske Patil <37668193+nikochiko@users.noreply.github.com>
Date: Wed, 29 Oct 2025 12:42:23 +0530
Subject: [PATCH] deprecate deepseek r1, add v3.1

---
 daras_ai_v2/language_model.py | 12 ++++++++++++
 scripts/init_llm_pricing.py   | 10 ++++++++++
 2 files changed, 22 insertions(+)

diff --git a/daras_ai_v2/language_model.py b/daras_ai_v2/language_model.py
index 935638604..fd8d87875 100644
--- a/daras_ai_v2/language_model.py
+++ b/daras_ai_v2/language_model.py
@@ -443,6 +443,18 @@ class LargeLanguageModels(Enum):
         context_window=128_000,
         max_output_tokens=8192,
         supports_json=True,
+        is_deprecated=True,
+        redirect_to="deepseek_v3p1",
+    )
+
+    deepseek_v3p1 = LLMSpec(
+        label="DeepSeek V3.1 • DeepSeek",
+        model_id="accounts/fireworks/models/deepseek-v3p1",
+        llm_api=LLMApis.fireworks,
+        context_window=163_800,
+        max_output_tokens=20_500,
+        supports_json=True,
+        is_thinking_model=True,
     )
 
     # https://console.groq.com/docs/models
diff --git a/scripts/init_llm_pricing.py b/scripts/init_llm_pricing.py
index a06224ea3..5d74d1240 100644
--- a/scripts/init_llm_pricing.py
+++ b/scripts/init_llm_pricing.py
@@ -1131,6 +1131,16 @@ def run():
         pricing_url="https://fireworks.ai/pricing",
     )
 
+    llm_pricing_create(
+        model_id="accounts/fireworks/models/deepseek-v3p1",
+        model_name=LargeLanguageModels.deepseek_v3p1.name,
+        unit_cost_input=0.56,
+        unit_cost_output=1.68,
+        unit_quantity=10**6,
+        provider=ModelProvider.fireworks,
+        pricing_url="https://fireworks.ai/pricing",
+    )
+
     llm_pricing_create(
         model_id="accounts/fireworks/models/mistral-small-24b-instruct-2501",
         model_name=LargeLanguageModels.mistral_small_24b_instruct.name,