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

Merge pull request #1684 from PeterDaveHelloKitchen/Support-xAI-Grok

Browse Source 
Add support of xAI and their Grok-2 & Grok-3 model
This commit is contained in:
Tal
2025-04-11 09:53:08 +03:00
committed by GitHub
parent e4f40da35c 0ac7028bc6
commit 7f6014e064
4 changed files with 26 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pr_agent/algo/__init__.py
View File

@ -84,6 +84,13 @@ MAX_TOKENS = {
'groq/llama-3.3-70b-versatile': 128000,
'groq/mixtral-8x7b-32768': 32768,
'groq/gemma2-9b-it': 8192,
'xai/grok-2': 131072,
'xai/grok-2-1212': 131072,
'xai/grok-2-latest': 131072,
'xai/grok-3-beta': 131072,
'xai/grok-3-fast-beta': 131072,
'xai/grok-3-mini-beta': 131072,
'xai/grok-3-mini-fast-beta': 131072,
'ollama/llama3': 4096,
'watsonx/meta-llama/llama-3-8b-instruct': 4096,
"watsonx/meta-llama/llama-3-70b-instruct": 4096,