Add support model deepseek/deepseek-reasoner

2025-07-02 11:50:37 +08:00 · 2025-01-22 20:32:18 +07:00
parent e0b6db9bec
commit 63f1d449ce
1 changed files with 1 additions and 0 deletions
--- a/pr_agent/algo/init.py
+++ b/pr_agent/algo/init.py
@ -30,6 +30,7 @@ MAX_TOKENS = {
    'claude-2': 100000,
    'command-nightly': 4096,
    'deepseek/deepseek-chat': 128000,  # 128K, but may be limited by config.max_model_tokens
+    'deepseek/deepseek-reasoner': 64000,  # 64K, but may be limited by config.max_model_tokens
    'replicate/llama-2-70b-chat:2c1608e18606fad2812020dc541930f2d0495ce32eee50074220b87300bc16e1': 4096,
    'meta-llama/Llama-2-7b-chat-hf': 4096,
    'vertex_ai/codechat-bison': 6144,