From b0609272c0b53b8ee8d4d16945356a7595bb20ae Mon Sep 17 00:00:00 2001
From: shenjack <3695888@qq.com>
Date: Fri, 25 Apr 2025 01:20:22 +0800
Subject: [PATCH] ollama: Add DeepSeek v3 max token length (#29156)

Add deepseek-v3 max token length for ollama

Release Notes:

- N/A
---
 crates/ollama/src/ollama.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/crates/ollama/src/ollama.rs b/crates/ollama/src/ollama.rs
index 640e672f7f..17a084d61d 100644
--- a/crates/ollama/src/ollama.rs
+++ b/crates/ollama/src/ollama.rs
@@ -84,7 +84,7 @@ fn get_max_tokens(name: &str) -> usize {
         "mistral" | "codestral" | "mixstral" | "llava" | "qwen2" | "qwen2.5-coder"
         | "dolphin-mixtral" => 32768,
         "llama3.1" | "llama3.2" | "llama3.3" | "phi3" | "phi3.5" | "phi4" | "command-r"
-        | "deepseek-coder-v2" | "deepseek-r1" | "yi-coder" => 128000,
+        | "deepseek-coder-v2" | "deepseek-v3" | "deepseek-r1" | "yi-coder" => 128000,
         _ => DEFAULT_TOKENS,
     }
     .clamp(1, MAXIMUM_TOKENS)