Skip to content

Commit fe9b647

Browse files
1 parent 0730917 commit fe9b647

File tree

2 files changed

+29
-0
lines changed

2 files changed

+29
-0
lines changed

pr_agent/algo/__init__.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,7 @@
7171
'vertex_ai/claude-opus-4@20250514': 200000,
7272
'vertex_ai/claude-opus-4-1@20250805': 200000,
7373
'vertex_ai/claude-opus-4-5@20251101': 200000,
74+
'vertex_ai/claude-opus-4-6': 200000,
7475
'vertex_ai/claude-3-5-sonnet@20240620': 100000,
7576
'vertex_ai/claude-3-5-sonnet-v2@20241022': 100000,
7677
'vertex_ai/claude-3-7-sonnet@20250219': 200000,
@@ -106,13 +107,15 @@
106107
'anthropic/claude-opus-4-20250514': 200000,
107108
'anthropic/claude-opus-4-1-20250805': 200000,
108109
'anthropic/claude-opus-4-5-20251101': 200000,
110+
'anthropic/claude-opus-4-6': 200000,
109111
'anthropic/claude-3-5-sonnet-20240620': 100000,
110112
'anthropic/claude-3-5-sonnet-20241022': 100000,
111113
'anthropic/claude-3-7-sonnet-20250219': 200000,
112114
'anthropic/claude-sonnet-4-20250514': 200000,
113115
'anthropic/claude-sonnet-4-5-20250929': 200000,
114116
'claude-opus-4-1-20250805': 200000,
115117
'claude-opus-4-5-20251101': 200000,
118+
'claude-opus-4-6': 200000,
116119
'claude-3-7-sonnet-20250219': 200000,
117120
'anthropic/claude-3-5-haiku-20241022': 100000,
118121
'anthropic/claude-haiku-4-5-20251001': 200000,
@@ -123,6 +126,7 @@
123126
'bedrock/anthropic.claude-3-sonnet-20240229-v1:0': 100000,
124127
'bedrock/anthropic.claude-opus-4-20250514-v1:0': 200000,
125128
'bedrock/anthropic.claude-opus-4-1-20250805-v1:0': 200000,
129+
'bedrock/anthropic.claude-opus-4-6-v1:0': 200000,
126130
'bedrock/anthropic.claude-3-haiku-20240307-v1:0': 100000,
127131
'bedrock/anthropic.claude-3-5-haiku-20241022-v1:0': 100000,
128132
'bedrock/anthropic.claude-haiku-4-5-20251001-v1:0': 200000,
@@ -135,6 +139,8 @@
135139
"bedrock/us.anthropic.claude-opus-4-1-20250805-v1:0": 200000,
136140
"bedrock/global.anthropic.claude-opus-4-5-20251101-v1:0": 200000,
137141
"bedrock/us.anthropic.claude-opus-4-5-20251101-v1:0": 200000,
142+
"bedrock/global.anthropic.claude-opus-4-6-v1:0": 200000,
143+
"bedrock/us.anthropic.claude-opus-4-6-v1:0": 200000,
138144
"bedrock/us.anthropic.claude-3-5-sonnet-20241022-v2:0": 100000,
139145
"bedrock/us.anthropic.claude-haiku-4-5-20251001-v1:0": 200000,
140146
"bedrock/eu.anthropic.claude-haiku-4-5-20251001-v1:0": 200000,

tests/unittest/test_get_max_tokens.py

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,3 +65,26 @@ def test_model_max_tokens_with__limit(self, monkeypatch):
6565
expected = 10000
6666

6767
assert get_max_tokens(model) == expected
68+
69+
@pytest.mark.parametrize(
70+
"model",
71+
[
72+
"anthropic/claude-opus-4-6",
73+
"claude-opus-4-6",
74+
"vertex_ai/claude-opus-4-6",
75+
"bedrock/anthropic.claude-opus-4-6-v1:0",
76+
"bedrock/global.anthropic.claude-opus-4-6-v1:0",
77+
"bedrock/us.anthropic.claude-opus-4-6-v1:0",
78+
],
79+
)
80+
def test_claude_opus_4_6_model_max_tokens(self, monkeypatch, model):
81+
fake_settings = type('', (), {
82+
'config': type('', (), {
83+
'custom_model_max_tokens': 0,
84+
'max_model_tokens': 0
85+
})()
86+
})()
87+
88+
monkeypatch.setattr(utils, "get_settings", lambda: fake_settings)
89+
90+
assert get_max_tokens(model) == 200000

0 commit comments

Comments
 (0)