From d949f440a89de8a29ed2e78cb4e2698f695608a1 Mon Sep 17 00:00:00 2001 From: Trung Dinh Date: Fri, 17 Jan 2025 11:42:36 +0700 Subject: [PATCH] Add support model deepseek/deepseek-chat --- pr_agent/algo/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py index 72eb7912..d4f08e5a 100644 --- a/pr_agent/algo/__init__.py +++ b/pr_agent/algo/__init__.py @@ -29,6 +29,7 @@ MAX_TOKENS = { 'claude-instant-1': 100000, 'claude-2': 100000, 'command-nightly': 4096, + 'deepseek/deepseek-chat': 128000, # 128K, 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,