Skip to content

Commit a257eee

Browse files
update llm configs for gpt-5
1 parent 11de47b commit a257eee

File tree

1 file changed

+15
-18
lines changed

1 file changed

+15
-18
lines changed

src/agentlab/llm/llm_configs.py

Lines changed: 15 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -20,18 +20,26 @@
2020
]
2121

2222
CHAT_MODEL_ARGS_DICT = {
23+
"openai/gpt-5-2025-08-07": OpenAIModelArgs(
24+
model_name="gpt-5-2025-08-07",
25+
max_total_tokens=400_000,
26+
max_input_tokens=272_000,
27+
max_new_tokens=16_384,
28+
temperature=1, # gpt-5 supports temperature of 1 only
29+
vision_support=True,
30+
),
2331
"openai/gpt-5-nano-2025-08-07": OpenAIModelArgs(
2432
model_name="gpt-5-nano-2025-08-07",
25-
max_total_tokens=128_000,
26-
max_input_tokens=128_000,
33+
max_total_tokens=400_000,
34+
max_input_tokens=272_000,
2735
max_new_tokens=16_384,
2836
temperature=1, # gpt-5 supports temperature of 1 only
2937
vision_support=True,
3038
),
3139
"openai/gpt-5-mini-2025-08-07": OpenAIModelArgs(
3240
model_name="gpt-5-mini-2025-08-07",
33-
max_total_tokens=128_000,
34-
max_input_tokens=128_000,
41+
max_total_tokens=400_000,
42+
max_input_tokens=272_000,
3543
max_new_tokens=16_384,
3644
temperature=1, # gpt-5 supports temperature of 1 only
3745
vision_support=True,
@@ -111,20 +119,6 @@
111119
max_new_tokens=64_000,
112120
temperature=1e-1,
113121
),
114-
"openai/gpt-5-nano-2025-08-07": OpenAIModelArgs(
115-
model_name="gpt-5-nano-2025-08-07",
116-
max_total_tokens=400_000,
117-
max_input_tokens=400_000 - 4_000,
118-
max_new_tokens=4_000,
119-
temperature=1, # temperature param not supported by gpt-5
120-
),
121-
"openai/gpt-5-mini-2025-08-07": OpenAIModelArgs(
122-
model_name="gpt-5-mini-2025-08-07",
123-
max_total_tokens=400_000,
124-
max_input_tokens=400_000 - 4_000,
125-
max_new_tokens=4_000,
126-
temperature=1, # temperature param not supported by gpt-5
127-
),
128122
"azure/gpt-35-turbo/gpt-35-turbo": AzureModelArgs(
129123
model_name="gpt-35-turbo",
130124
max_total_tokens=8_192,
@@ -171,20 +165,23 @@
171165
max_total_tokens=256_000,
172166
max_input_tokens=256_000,
173167
max_new_tokens=16_384,
168+
temperature=1, # temperature param not supported by gpt-5
174169
vision_support=True,
175170
),
176171
"azure/gpt-5-mini-2025-08-07": AzureModelArgs(
177172
model_name="gpt-5-mini",
178173
max_total_tokens=256_000,
179174
max_input_tokens=256_000,
180175
max_new_tokens=16_384,
176+
temperature=1, # temperature param not supported by gpt-5
181177
vision_support=True,
182178
),
183179
"azure/gpt-5-nano-2025-08-07": AzureModelArgs(
184180
model_name="gpt-5-nano",
185181
max_total_tokens=256_000,
186182
max_input_tokens=256_000,
187183
max_new_tokens=16_384,
184+
temperature=1, # temperature param not supported by gpt-5
188185
vision_support=True,
189186
),
190187
# ---------------- Anthropic ----------------#

0 commit comments

Comments
 (0)