pr-agent/pr_agent/algo/__init__.py

MAX_TOKENS = {
    'text-embedding-ada-002': 8000,
    'gpt-3.5-turbo': 16000,
    'gpt-3.5-turbo-0125': 16000,
    'gpt-3.5-turbo-0613': 4000,
    'gpt-3.5-turbo-1106': 16000,
    'gpt-3.5-turbo-16k': 16000,
    'gpt-3.5-turbo-16k-0613': 16000,
    'gpt-4': 8000,
    'gpt-4-0613': 8000,
    'gpt-4-32k': 32000,
    'gpt-4-1106-preview': 128000,  # 128K, but may be limited by config.max_model_tokens
    'gpt-4-0125-preview': 128000,  # 128K, but may be limited by config.max_model_tokens
    'gpt-4o': 128000,  # 128K, but may be limited by config.max_model_tokens
    'gpt-4o-2024-05-13': 128000,  # 128K, but may be limited by config.max_model_tokens
    'gpt-4-turbo-preview': 128000,  # 128K, but may be limited by config.max_model_tokens
    'gpt-4-turbo-2024-04-09': 128000,  # 128K, but may be limited by config.max_model_tokens
    'gpt-4-turbo': 128000,  # 128K, but may be limited by config.max_model_tokens
    'gpt-4o-mini': 128000,  # 128K, but may be limited by config.max_model_tokens
    'gpt-4o-mini-2024-07-18': 128000,  # 128K, but may be limited by config.max_model_tokens
    'gpt-4o-2024-08-06': 128000,  # 128K, but may be limited by config.max_model_tokens
    'gpt-4o-2024-11-20': 128000,  # 128K, but may be limited by config.max_model_tokens
    'gpt-4.5-preview': 128000,  # 128K, but may be limited by config.max_model_tokens
    'gpt-4.5-preview-2025-02-27': 128000,  # 128K, but may be limited by config.max_model_tokens
    'gpt-4.1': 1047576,
    'gpt-4.1-2025-04-14': 1047576,
    'gpt-4.1-mini': 1047576,
    'gpt-4.1-mini-2025-04-14': 1047576,
    'gpt-4.1-nano': 1047576,
    'gpt-4.1-nano-2025-04-14': 1047576,
    'o1-mini': 128000,  # 128K, but may be limited by config.max_model_tokens
    'o1-mini-2024-09-12': 128000,  # 128K, but may be limited by config.max_model_tokens
    'o1-preview': 128000,  # 128K, but may be limited by config.max_model_tokens
    'o1-preview-2024-09-12': 128000,  # 128K, but may be limited by config.max_model_tokens
    'o1-2024-12-17': 204800,  # 200K, but may be limited by config.max_model_tokens
    'o1': 204800,  # 200K, but may be limited by config.max_model_tokens
    'o3-mini': 204800,  # 200K, but may be limited by config.max_model_tokens
    'o3-mini-2025-01-31': 204800,  # 200K, but may be limited by config.max_model_tokens
    'o3': 200000,  # 200K, but may be limited by config.max_model_tokens
    'o3-2025-04-16': 200000,  # 200K, but may be limited by config.max_model_tokens
    'o4-mini': 200000, # 200K, but may be limited by config.max_model_tokens
    'o4-mini-2025-04-16': 200000, # 200K, but may be limited by config.max_model_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
    '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,
    'vertex_ai/codechat-bison-32k': 32000,
    'vertex_ai/claude-3-haiku@20240307': 100000,
    'vertex_ai/claude-3-5-haiku@20241022': 100000,
    'vertex_ai/claude-3-sonnet@20240229': 100000,
    'vertex_ai/claude-3-opus@20240229': 100000,
    'vertex_ai/claude-opus-4@20250514': 200000,
    'vertex_ai/claude-3-5-sonnet@20240620': 100000,
    'vertex_ai/claude-3-5-sonnet-v2@20241022': 100000,
    'vertex_ai/claude-3-7-sonnet@20250219': 200000,
    'vertex_ai/claude-sonnet-4@20250514': 200000,
    'vertex_ai/gemini-1.5-pro': 1048576,
    'vertex_ai/gemini-2.5-pro-preview-03-25': 1048576,
    'vertex_ai/gemini-2.5-pro-preview-05-06': 1048576,
    'vertex_ai/gemini-1.5-flash': 1048576,
    'vertex_ai/gemini-2.0-flash': 1048576,
    'vertex_ai/gemini-2.5-flash-preview-04-17': 1048576,
    'vertex_ai/gemini-2.5-flash-preview-05-20': 1048576,
    'vertex_ai/gemma2': 8200,
    'gemini/gemini-1.5-pro': 1048576,
    'gemini/gemini-1.5-flash': 1048576,
    'gemini/gemini-2.0-flash': 1048576,
    'gemini/gemini-2.5-flash-preview-04-17': 1048576,
    'gemini/gemini-2.5-flash-preview-05-20': 1048576,
    'gemini/gemini-2.5-pro-preview-03-25': 1048576,
    'gemini/gemini-2.5-pro-preview-05-06': 1048576,
    'codechat-bison': 6144,
    'codechat-bison-32k': 32000,
    'anthropic.claude-instant-v1': 100000,
    'anthropic.claude-v1': 100000,
    'anthropic.claude-v2': 100000,
    'anthropic/claude-3-opus-20240229': 100000,
    'anthropic/claude-opus-4-20250514': 200000,
    'anthropic/claude-3-5-sonnet-20240620': 100000,
    'anthropic/claude-3-5-sonnet-20241022': 100000,
    'anthropic/claude-3-7-sonnet-20250219': 200000,
    'anthropic/claude-sonnet-4-20250514': 200000,
    'claude-3-7-sonnet-20250219': 200000,
    'anthropic/claude-3-5-haiku-20241022': 100000,
    'bedrock/anthropic.claude-instant-v1': 100000,
    'bedrock/anthropic.claude-v2': 100000,
    'bedrock/anthropic.claude-v2:1': 100000,
    'bedrock/anthropic.claude-3-sonnet-20240229-v1:0': 100000,
    'bedrock/anthropic.claude-opus-4-20250514-v1:0': 200000,
    'bedrock/anthropic.claude-3-haiku-20240307-v1:0': 100000,
    'bedrock/anthropic.claude-3-5-haiku-20241022-v1:0': 100000,
    'bedrock/anthropic.claude-3-5-sonnet-20240620-v1:0': 100000,
    'bedrock/anthropic.claude-3-5-sonnet-20241022-v2:0': 100000,
    'bedrock/anthropic.claude-3-7-sonnet-20250219-v1:0': 200000,
    'bedrock/anthropic.claude-sonnet-4-20250514-v1:0': 200000,
    "bedrock/us.anthropic.claude-opus-4-20250514-v1:0": 200000,
    "bedrock/us.anthropic.claude-3-5-sonnet-20241022-v2:0": 100000,
    "bedrock/us.anthropic.claude-3-7-sonnet-20250219-v1:0": 200000,
    "bedrock/us.anthropic.claude-sonnet-4-20250514-v1:0": 200000,
    'claude-3-5-sonnet': 100000,
    'groq/meta-llama/llama-4-scout-17b-16e-instruct': 131072,
    'groq/meta-llama/llama-4-maverick-17b-128e-instruct': 131072,
    'groq/llama3-8b-8192': 8192,
    'groq/llama3-70b-8192': 8192,
    'groq/llama-3.1-8b-instant': 8192,
    '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': 131072,
    'xai/grok-3-beta': 131072,
    'xai/grok-3-fast': 131072,
    'xai/grok-3-fast-beta': 131072,
    'xai/grok-3-mini': 131072,
    'xai/grok-3-mini-beta': 131072,
    'xai/grok-3-mini-fast': 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,
    "watsonx/meta-llama/llama-3-405b-instruct": 16384,
    "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,
    "mistral/mistral-small-latest": 8191,
    "mistral/mistral-medium-latest": 8191,
    "mistral/mistral-large-2407": 128000,
    "mistral/mistral-large-latest": 128000,
    "mistral/open-mistral-7b": 8191,
    "mistral/open-mixtral-8x7b": 8191,
    "mistral/open-mixtral-8x22b": 8191,
    "mistral/codestral-latest": 8191,
    "mistral/open-mistral-nemo": 128000,
    "mistral/open-mistral-nemo-2407": 128000,
    "mistral/open-codestral-mamba": 256000,
    "mistral/codestral-mamba-latest": 256000,
    "codestral/codestral-latest": 8191,
    "codestral/codestral-2405": 8191,
}

USER_MESSAGE_ONLY_MODELS = [
    "deepseek/deepseek-reasoner",
    "o1-mini",
    "o1-mini-2024-09-12",
    "o1-preview"
]

NO_SUPPORT_TEMPERATURE_MODELS = [
    "deepseek/deepseek-reasoner",
    "o1-mini",
    "o1-mini-2024-09-12",
    "o1",
    "o1-2024-12-17",
    "o3-mini",
    "o3-mini-2025-01-31",
    "o1-preview",
    "o3",
    "o3-2025-04-16",
    "o4-mini",
    "o4-mini-2025-04-16",
]

SUPPORT_REASONING_EFFORT_MODELS = [
    "o3-mini",
    "o3-mini-2025-01-31",
    "o3",
    "o3-2025-04-16",
    "o4-mini",
    "o4-mini-2025-04-16",
]

CLAUDE_EXTENDED_THINKING_MODELS = [
    "anthropic/claude-3-7-sonnet-20250219",
    "claude-3-7-sonnet-20250219"
]
Initial commit - PR-Agent OSS release 2023-07-06 00:21:08 +03:00			`MAX_TOKENS = {`
issue tool 2023-09-05 08:40:05 +03:00			`'text-embedding-ada-002': 8000,`
chore: update GPT3.5 models 2024-04-22 20:25:32 -04:00			`'gpt-3.5-turbo': 16000,`
			`'gpt-3.5-turbo-0125': 16000,`
Initial commit - PR-Agent OSS release 2023-07-06 00:21:08 +03:00			`'gpt-3.5-turbo-0613': 4000,`
chore: update GPT3.5 models 2024-04-22 20:25:32 -04:00			`'gpt-3.5-turbo-1106': 16000,`
Initial commit - PR-Agent OSS release 2023-07-06 00:21:08 +03:00			`'gpt-3.5-turbo-16k': 16000,`
			`'gpt-3.5-turbo-16k-0613': 16000,`
			`'gpt-4': 8000,`
			`'gpt-4-0613': 8000,`
			`'gpt-4-32k': 32000,`
Update Python code formatting, configuration loading, and local model additions 1. Code Formatting: - Standardized Python code formatting across multiple files to align with PEP 8 guidelines. This includes adjustments to whitespace, line breaks, and inline comments. 2. Configuration Loader Enhancements: - Enhanced the `get_settings` function in `config_loader.py` to provide more robust handling of settings retrieval. Added detailed documentation to improve code maintainability and clarity. 3. Model Addition in __init__.py: - Added a new model "ollama/llama3" with a token limit to the MAX_TOKENS dictionary in `__init__.py` to support new AI capabilities and configurations. 2024-06-03 23:58:31 +08:00			`'gpt-4-1106-preview': 128000, # 128K, but may be limited by config.max_model_tokens`
moving the 'improve' command to turbo mode, with auto_extended=true 2024-02-01 09:46:04 +02:00			`'gpt-4-0125-preview': 128000, # 128K, but may be limited by config.max_model_tokens`
Add gpt-4o models 2024-05-14 08:24:34 +07:00			`'gpt-4o': 128000, # 128K, but may be limited by config.max_model_tokens`
			`'gpt-4o-2024-05-13': 128000, # 128K, but may be limited by config.max_model_tokens`
feat(pr_agent): add support for gpt-4-turbo-preview model and update default settings 2024-03-26 14:47:05 +09:00			`'gpt-4-turbo-preview': 128000, # 128K, but may be limited by config.max_model_tokens`
add new models 2024-04-10 14:44:38 +09:00			`'gpt-4-turbo-2024-04-09': 128000, # 128K, but may be limited by config.max_model_tokens`
			`'gpt-4-turbo': 128000, # 128K, but may be limited by config.max_model_tokens`
Add support model gpt-4o-mini 2024-07-23 07:41:04 +07:00			`'gpt-4o-mini': 128000, # 128K, but may be limited by config.max_model_tokens`
Add support model gpt-4o-mini-2024-07-18 2024-07-23 07:43:47 +07:00			`'gpt-4o-mini-2024-07-18': 128000, # 128K, but may be limited by config.max_model_tokens`
Add support model gpt-4o-2024-08-06 2024-08-07 07:28:51 +07:00			`'gpt-4o-2024-08-06': 128000, # 128K, but may be limited by config.max_model_tokens`
Add support model gpt-4o-2024-11-20 2024-11-21 22:10:27 +07:00			`'gpt-4o-2024-11-20': 128000, # 128K, but may be limited by config.max_model_tokens`
Add support of OpenAI GPT-4.5 Preview model Reference: - https://openai.com/index/introducing-gpt-4-5/ - https://platform.openai.com/docs/models/gpt-4.5-preview 2025-04-04 05:13:15 +08:00			`'gpt-4.5-preview': 128000, # 128K, but may be limited by config.max_model_tokens`
			`'gpt-4.5-preview-2025-02-27': 128000, # 128K, but may be limited by config.max_model_tokens`
Add support of OpenAI GPT-4.1 model family Reference: - https://openai.com/index/gpt-4-1/ - https://platform.openai.com/docs/models/gpt-4.1 2025-04-15 01:54:35 +08:00			`'gpt-4.1': 1047576,`
			`'gpt-4.1-2025-04-14': 1047576,`
			`'gpt-4.1-mini': 1047576,`
			`'gpt-4.1-mini-2025-04-14': 1047576,`
			`'gpt-4.1-nano': 1047576,`
			`'gpt-4.1-nano-2025-04-14': 1047576,`
Add o1-mini models 2024-09-13 08:22:09 +07:00			`'o1-mini': 128000, # 128K, but may be limited by config.max_model_tokens`
			`'o1-mini-2024-09-12': 128000, # 128K, but may be limited by config.max_model_tokens`
Add o1-preview models 2024-09-13 08:23:18 +07:00			`'o1-preview': 128000, # 128K, but may be limited by config.max_model_tokens`
			`'o1-preview-2024-09-12': 128000, # 128K, but may be limited by config.max_model_tokens`
Fix context windows token for model o1-2024-12-17 2024-12-19 23:11:45 +07:00			`'o1-2024-12-17': 204800, # 200K, but may be limited by config.max_model_tokens`
Supported model gpt-o1 2024-12-22 10:33:26 +07:00			`'o1': 204800, # 200K, but may be limited by config.max_model_tokens`
Add support o3-mini model 2025-02-02 18:02:53 +07:00			`'o3-mini': 204800, # 200K, but may be limited by config.max_model_tokens`
			`'o3-mini-2025-01-31': 204800, # 200K, but may be limited by config.max_model_tokens`
Add OpenAI o3 & 4o-mini reasoning models Reference: - https://platform.openai.com/docs/models/o3 - https://platform.openai.com/docs/models/o4-mini - https://openai.com/index/introducing-o3-and-o4-mini/ 2025-04-17 02:19:42 +08:00			`'o3': 200000, # 200K, but may be limited by config.max_model_tokens`
			`'o3-2025-04-16': 200000, # 200K, but may be limited by config.max_model_tokens`
			`'o4-mini': 200000, # 200K, but may be limited by config.max_model_tokens`
			`'o4-mini-2025-04-16': 200000, # 200K, but may be limited by config.max_model_tokens`
bug fixes and updates 2023-08-03 16:05:46 -07:00			`'claude-instant-1': 100000,`
			`'claude-2': 100000,`
			`'command-nightly': 4096,`
Add support model deepseek/deepseek-chat 2025-01-17 11:42:36 +07:00			`'deepseek/deepseek-chat': 128000, # 128K, but may be limited by config.max_model_tokens`
Add support model deepseek/deepseek-reasoner 2025-01-22 20:32:18 +07:00			`'deepseek/deepseek-reasoner': 64000, # 64K, but may be limited by config.max_model_tokens`
bug fixes and updates 2023-08-03 16:05:46 -07:00			`'replicate/llama-2-70b-chat:2c1608e18606fad2812020dc541930f2d0495ce32eee50074220b87300bc16e1': 4096,`
Support Google's Vertex AI 2023-11-07 09:13:08 +00:00			`'meta-llama/Llama-2-7b-chat-hf': 4096,`
			`'vertex_ai/codechat-bison': 6144,`
			`'vertex_ai/codechat-bison-32k': 32000,`
Support models: Anthropic Calude 3 on Vertex AI 2024-06-13 14:34:14 +09:00			`'vertex_ai/claude-3-haiku@20240307': 100000,`
Add support claude-3-5-haiku-20241022 model 2024-11-06 07:52:58 +07:00			`'vertex_ai/claude-3-5-haiku@20241022': 100000,`
Support models: Anthropic Calude 3 on Vertex AI 2024-06-13 14:34:14 +09:00			`'vertex_ai/claude-3-sonnet@20240229': 100000,`
			`'vertex_ai/claude-3-opus@20240229': 100000,`
feat: add support for Claude 4 2025-05-23 14:16:44 +09:00			`'vertex_ai/claude-opus-4@20250514': 200000,`
feat: Support Anthropic Calude 3.5 Sonnet on Vertex AI 2024-06-25 17:32:17 +09:00			`'vertex_ai/claude-3-5-sonnet@20240620': 100000,`
Add model vertex_ai/claude-3-5-sonnet-v2@20241022 2024-10-23 07:11:33 +07:00			`'vertex_ai/claude-3-5-sonnet-v2@20241022': 100000,`
Update pr_agent/algo/__init__.py Co-authored-by: qodo-merge-pro-for-open-source[bot] <189517486+qodo-merge-pro-for-open-source[bot]@users.noreply.github.com> 2025-02-25 16:51:39 +09:00			`'vertex_ai/claude-3-7-sonnet@20250219': 200000,`
feat: add support for Claude 4 2025-05-23 14:16:44 +09:00			`'vertex_ai/claude-sonnet-4@20250514': 200000,`
remove extra space 2024-06-24 14:01:33 +02:00			`'vertex_ai/gemini-1.5-pro': 1048576,`
Update model references from o3-mini to o4-mini and add Gemini models 2025-04-19 09:26:35 +03:00			`'vertex_ai/gemini-2.5-pro-preview-03-25': 1048576,`
Add Gemini-2.5-pro-preview-05-06 model and update litellm dependency 2025-05-16 16:32:45 +03:00			`'vertex_ai/gemini-2.5-pro-preview-05-06': 1048576,`
add additional google models support 2024-07-09 14:29:50 +02:00			`'vertex_ai/gemini-1.5-flash': 1048576,`
Support generally available `gemini-2.0-flash` Signed-off-by: Yu Ishikawa <yu-iskw@users.noreply.github.com> 2025-02-17 08:40:05 +09:00			`'vertex_ai/gemini-2.0-flash': 1048576,`
Update model references from o3-mini to o4-mini and add Gemini models 2025-04-19 09:26:35 +03:00			`'vertex_ai/gemini-2.5-flash-preview-04-17': 1048576,`
Complete Gemini 2.5 Flash model info 2025-05-24 23:52:18 +08:00			`'vertex_ai/gemini-2.5-flash-preview-05-20': 1048576,`
add additional google models support 2024-07-09 14:29:50 +02:00			`'vertex_ai/gemma2': 8200,`
Support Google AI Studio Signed-off-by: Yu Ishikawa <yu-iskw@users.noreply.github.com> 2024-10-29 08:00:16 +09:00			`'gemini/gemini-1.5-pro': 1048576,`
			`'gemini/gemini-1.5-flash': 1048576,`
Support generally available `gemini-2.0-flash` Signed-off-by: Yu Ishikawa <yu-iskw@users.noreply.github.com> 2025-02-17 08:40:05 +09:00			`'gemini/gemini-2.0-flash': 1048576,`
Complete Gemini 2.5 Flash model info 2025-05-24 23:52:18 +08:00			`'gemini/gemini-2.5-flash-preview-04-17': 1048576,`
			`'gemini/gemini-2.5-flash-preview-05-20': 1048576,`
add gemini 2.5 pro preview model token limit 2025-04-08 20:41:59 +05:30			`'gemini/gemini-2.5-pro-preview-03-25': 1048576,`
Add Gemini-2.5-pro-preview-05-06 model and update litellm dependency 2025-05-16 16:32:45 +03:00			`'gemini/gemini-2.5-pro-preview-05-06': 1048576,`
Support Google's Vertex AI 2023-11-07 09:13:08 +00:00			`'codechat-bison': 6144,`
			`'codechat-bison-32k': 32000,`
support Amazon Bedrock 2023-11-28 20:11:40 +09:00			`'anthropic.claude-instant-v1': 100000,`
			`'anthropic.claude-v1': 100000,`
Add support for Anthropic Claude-3 model in configuration and update Usage.md 2024-03-06 08:20:08 +02:00			`'anthropic.claude-v2': 100000,`
			`'anthropic/claude-3-opus-20240229': 100000,`
feat: add support for Claude 4 2025-05-23 14:16:44 +09:00			`'anthropic/claude-opus-4-20250514': 200000,`
feat: claude 3.5 sonnet support 2024-06-21 09:30:52 +01:00			`'anthropic/claude-3-5-sonnet-20240620': 100000,`
Add model anthropic/claude-3-5-sonnet-20241022 2024-10-23 07:10:22 +07:00			`'anthropic/claude-3-5-sonnet-20241022': 100000,`
Update pr_agent/algo/__init__.py Co-authored-by: qodo-merge-pro-for-open-source[bot] <189517486+qodo-merge-pro-for-open-source[bot]@users.noreply.github.com> 2025-02-25 16:51:50 +09:00			`'anthropic/claude-3-7-sonnet-20250219': 200000,`
feat: add support for Claude 4 2025-05-23 14:16:44 +09:00			`'anthropic/claude-sonnet-4-20250514': 200000,`
Define models that support extended thinking feature 2025-03-03 18:22:31 +07:00			`'claude-3-7-sonnet-20250219': 200000,`
Add support claude-3-5-haiku-20241022 model 2024-11-06 07:52:58 +07:00			`'anthropic/claude-3-5-haiku-20241022': 100000,`
update max_tokens 2024-03-13 11:14:04 +09:00			`'bedrock/anthropic.claude-instant-v1': 100000,`
			`'bedrock/anthropic.claude-v2': 100000,`
			`'bedrock/anthropic.claude-v2:1': 100000,`
			`'bedrock/anthropic.claude-3-sonnet-20240229-v1:0': 100000,`
fix: reflect comments 2025-05-24 09:45:27 +09:00			`'bedrock/anthropic.claude-opus-4-20250514-v1:0': 200000,`
add claude-3-haiku 2024-03-14 16:58:44 +09:00			`'bedrock/anthropic.claude-3-haiku-20240307-v1:0': 100000,`
Add support model bedrock/anthropic.claude-3-5-haiku-20241022-v1:0 2024-11-07 09:12:52 +07:00			`'bedrock/anthropic.claude-3-5-haiku-20241022-v1:0': 100000,`
sonnet-3.5 2024-07-04 12:23:36 +03:00			`'bedrock/anthropic.claude-3-5-sonnet-20240620-v1:0': 100000,`
Add support bedrock/anthropic.claude-3-5-sonnet-20241022-v2:0 model 2024-10-23 08:08:04 +07:00			`'bedrock/anthropic.claude-3-5-sonnet-20241022-v2:0': 100000,`
Update pr_agent/algo/__init__.py Co-authored-by: qodo-merge-pro-for-open-source[bot] <189517486+qodo-merge-pro-for-open-source[bot]@users.noreply.github.com> 2025-02-25 16:51:55 +09:00			`'bedrock/anthropic.claude-3-7-sonnet-20250219-v1:0': 200000,`
feat: add support for Claude 4 2025-05-23 14:16:44 +09:00			`'bedrock/anthropic.claude-sonnet-4-20250514-v1:0': 200000,`
fix: reflect comments 2025-05-24 09:45:27 +09:00			`"bedrock/us.anthropic.claude-opus-4-20250514-v1:0": 200000,`
fix: improve changelog update prompt and response handling 2024-12-23 17:06:21 +02:00			`"bedrock/us.anthropic.claude-3-5-sonnet-20241022-v2:0": 100000,`
chore: bedrock/us.anthropic.claude-3-7-sonnet-20250219-v1:0 2025-03-02 08:44:23 +09:00			`"bedrock/us.anthropic.claude-3-7-sonnet-20250219-v1:0": 200000,`
fix: reflect comments 2025-05-24 09:45:27 +09:00			`"bedrock/us.anthropic.claude-sonnet-4-20250514-v1:0": 200000,`
Add tests for patch extension and update configuration for extra lines handling - Added unit tests in `test_extend_patch.py` and `test_pr_generate_extended_diff.py` to verify patch extension functionality with extra lines. - Updated `pr_processing.py` to include `patch_extra_lines_before` and `patch_extra_lines_after` settings. - Modified `configuration.toml` to adjust `patch_extra_lines_before` to 4 and `max_context_tokens` to 16000. - Enabled extra lines in `pr_code_suggestions.py`. - Added new model `claude-3-5-sonnet` to `__init__.py`. 2024-08-11 09:21:34 +03:00			`'claude-3-5-sonnet': 100000,`
Add support of Meta's Llama 4 Scout and Maverick 17b from Groq Cloud Reference: - https://ai.meta.com/blog/llama-4-multimodal-intelligence/ - https://console.groq.com/docs/models#preview-models - https://groq.com/llama-4-now-live-on-groq-build-fast-at-the-lowest-cost-without-compromise/ 2025-04-08 01:47:15 +08:00			`'groq/meta-llama/llama-4-scout-17b-16e-instruct': 131072,`
			`'groq/meta-llama/llama-4-maverick-17b-128e-instruct': 131072,`
Add GROQ.KEY support in LiteLLMAIHandler 2024-04-21 15:21:45 +09:00			`'groq/llama3-8b-8192': 8192,`
			`'groq/llama3-70b-8192': 8192,`
Update Groq models and tokens 2024-12-28 01:51:33 +09:00			`'groq/llama-3.1-8b-instant': 8192,`
Increase the max token of groq/llama-3.3-70b-versatile 2024-12-30 11:35:02 +09:00			`'groq/llama-3.3-70b-versatile': 128000,`
Add Llama 3.1 and Mixtral 8x7B for Groq 2024-07-28 02:28:42 +09:00			`'groq/mixtral-8x7b-32768': 32768,`
Update Groq models and tokens 2024-12-28 01:51:33 +09:00			`'groq/gemma2-9b-it': 8192,`
Add support of xAI and their Grok-2 model Close #1630 2025-04-08 01:36:21 +08:00			`'xai/grok-2': 131072,`
			`'xai/grok-2-1212': 131072,`
			`'xai/grok-2-latest': 131072,`
Add Grok-3 non-beta model IDs 2025-05-24 14:22:55 +08:00			`'xai/grok-3': 131072,`
Support xAI Grok-3 series models Reference: - https://docs.x.ai/docs/release-notes#april-2025 2025-04-11 00:31:15 +08:00			`'xai/grok-3-beta': 131072,`
Add Grok-3 non-beta model IDs 2025-05-24 14:22:55 +08:00			`'xai/grok-3-fast': 131072,`
Support xAI Grok-3 series models Reference: - https://docs.x.ai/docs/release-notes#april-2025 2025-04-11 00:31:15 +08:00			`'xai/grok-3-fast-beta': 131072,`
Add Grok-3 non-beta model IDs 2025-05-24 14:22:55 +08:00			`'xai/grok-3-mini': 131072,`
Support xAI Grok-3 series models Reference: - https://docs.x.ai/docs/release-notes#april-2025 2025-04-11 00:31:15 +08:00			`'xai/grok-3-mini-beta': 131072,`
Add Grok-3 non-beta model IDs 2025-05-24 14:22:55 +08:00			`'xai/grok-3-mini-fast': 131072,`
Support xAI Grok-3 series models Reference: - https://docs.x.ai/docs/release-notes#april-2025 2025-04-11 00:31:15 +08:00			`'xai/grok-3-mini-fast-beta': 131072,`
fix wrong provider name 2024-06-04 15:09:30 +08:00			`'ollama/llama3': 4096,`
Update __init__.py 2024-07-29 08:32:34 +03:00			`'watsonx/meta-llama/llama-3-8b-instruct': 4096,`
			`"watsonx/meta-llama/llama-3-70b-instruct": 4096,`
			`"watsonx/meta-llama/llama-3-405b-instruct": 16384,`
			`"watsonx/ibm/granite-13b-chat-v2": 8191,`
			`"watsonx/ibm/granite-34b-code-instruct": 8191,`
			`"watsonx/mistralai/mistral-large": 32768,`
feat: add DeepInfra support 2025-03-06 15:49:07 +07:00			`"deepinfra/deepseek-ai/DeepSeek-R1-Distill-Qwen-32B": 128000,`
			`"deepinfra/deepseek-ai/DeepSeek-R1-Distill-Llama-70B": 128000,`
			`"deepinfra/deepseek-ai/DeepSeek-R1": 128000,`
feat: add support for Mistral and Codestral models 2025-04-18 12:02:55 +09:00			`"mistral/mistral-small-latest": 8191,`
			`"mistral/mistral-medium-latest": 8191,`
			`"mistral/mistral-large-2407": 128000,`
			`"mistral/mistral-large-latest": 128000,`
			`"mistral/open-mistral-7b": 8191,`
			`"mistral/open-mixtral-8x7b": 8191,`
			`"mistral/open-mixtral-8x22b": 8191,`
			`"mistral/codestral-latest": 8191,`
			`"mistral/open-mistral-nemo": 128000,`
			`"mistral/open-mistral-nemo-2407": 128000,`
			`"mistral/open-codestral-mamba": 256000,`
			`"mistral/codestral-mamba-latest": 256000,`
			`"codestral/codestral-latest": 8191,`
			`"codestral/codestral-2405": 8191,`
Initial commit - PR-Agent OSS release 2023-07-06 00:21:08 +03:00			`}`
Define USER_MESSAGE_ONLY_MODELS in the algo init file 2025-02-02 18:01:16 +07:00
			`USER_MESSAGE_ONLY_MODELS = [`
			`"deepseek/deepseek-reasoner",`
			`"o1-mini",`
Add o1-preview model to USER_MESSAGE_ONLY_MODELS and NO_SUPPORT_TEMPERATURE_MODELS 2025-02-16 15:18:52 +07:00			`"o1-mini-2024-09-12",`
			`"o1-preview"`
Define USER_MESSAGE_ONLY_MODELS in the algo init file 2025-02-02 18:01:16 +07:00			`]`
Add NO_SUPPORT_TEMPERATURE_MODELS list 2025-02-16 15:17:59 +07:00
			`NO_SUPPORT_TEMPERATURE_MODELS = [`
			`"deepseek/deepseek-reasoner",`
			`"o1-mini",`
			`"o1-mini-2024-09-12",`
			`"o1",`
			`"o1-2024-12-17",`
			`"o3-mini",`
			`"o3-mini-2025-01-31",`
Add OpenAI o3 & 4o-mini reasoning models Reference: - https://platform.openai.com/docs/models/o3 - https://platform.openai.com/docs/models/o4-mini - https://openai.com/index/introducing-o3-and-o4-mini/ 2025-04-17 02:19:42 +08:00			`"o1-preview",`
			`"o3",`
			`"o3-2025-04-16",`
			`"o4-mini",`
			`"o4-mini-2025-04-16",`
Add NO_SUPPORT_TEMPERATURE_MODELS list 2025-02-16 15:17:59 +07:00			`]`
Define SUPPORT_REASONING_EFFORT_MODELS list 2025-02-21 22:10:33 +07:00
			`SUPPORT_REASONING_EFFORT_MODELS = [`
			`"o3-mini",`
Add OpenAI o3 & 4o-mini reasoning models Reference: - https://platform.openai.com/docs/models/o3 - https://platform.openai.com/docs/models/o4-mini - https://openai.com/index/introducing-o3-and-o4-mini/ 2025-04-17 02:19:42 +08:00			`"o3-mini-2025-01-31",`
			`"o3",`
			`"o3-2025-04-16",`
			`"o4-mini",`
			`"o4-mini-2025-04-16",`
Define SUPPORT_REASONING_EFFORT_MODELS list 2025-02-21 22:10:33 +07:00			`]`
Define models that support extended thinking feature 2025-03-03 18:22:31 +07:00
			`CLAUDE_EXTENDED_THINKING_MODELS = [`
			`"anthropic/claude-3-7-sonnet-20250219",`
			`"claude-3-7-sonnet-20250219"`
			`]`