xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-10 15:50:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add DeepInfra support

Browse Source
This commit is contained in:
muhammad-asn
2025-03-06 15:49:07 +07:00
parent eb5f38b13b
commit 4f2551e0a6
4 changed files with 23 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pr_agent/algo/__init__.py
View File

@ -86,6 +86,9 @@ MAX_TOKENS = {
"watsonx/ibm/granite-13b-chat-v2": 8191,
"watsonx/ibm/granite-34b-code-instruct": 8191,
"watsonx/mistralai/mistral-large": 32768,
"deepinfra/deepseek-ai/DeepSeek-R1-Distill-Qwen-32B": 128000,
"deepinfra/deepseek-ai/DeepSeek-R1-Distill-Llama-70B": 128000,
"deepinfra/deepseek-ai/DeepSeek-R1": 128000,
}
USER_MESSAGE_ONLY_MODELS = [