Skip to content

Commit bf0b6e7

Browse files
committed
fix repeated llm configs
1 parent ecb8f11 commit bf0b6e7

File tree

1 file changed

+2
-16
lines changed

1 file changed

+2
-16
lines changed

src/agentlab/llm/llm_configs.py

Lines changed: 2 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -20,22 +20,6 @@
2020
]
2121

2222
CHAT_MODEL_ARGS_DICT = {
23-
"openai/gpt-5-nano-2025-08-07": OpenAIModelArgs(
24-
model_name="gpt-5-nano-2025-08-07",
25-
max_total_tokens=128_000,
26-
max_input_tokens=128_000,
27-
max_new_tokens=16_384,
28-
temperature=1, # gpt-5 supports temperature of 1 only
29-
vision_support=True,
30-
),
31-
"openai/gpt-5-mini-2025-08-07": OpenAIModelArgs(
32-
model_name="gpt-5-mini-2025-08-07",
33-
max_total_tokens=128_000,
34-
max_input_tokens=128_000,
35-
max_new_tokens=16_384,
36-
temperature=1, # gpt-5 supports temperature of 1 only
37-
vision_support=True,
38-
),
3923
"openai/gpt-4.1-mini-2025-04-14": OpenAIModelArgs(
4024
model_name="gpt-4.1-mini-2025-04-14",
4125
max_total_tokens=128_000,
@@ -117,13 +101,15 @@
117101
max_input_tokens=400_000 - 4_000,
118102
max_new_tokens=4_000,
119103
temperature=1, # temperature param not supported by gpt-5
104+
vision_support=True,
120105
),
121106
"openai/gpt-5-mini-2025-08-07": OpenAIModelArgs(
122107
model_name="gpt-5-mini-2025-08-07",
123108
max_total_tokens=400_000,
124109
max_input_tokens=400_000 - 4_000,
125110
max_new_tokens=4_000,
126111
temperature=1, # temperature param not supported by gpt-5
112+
vision_support=True,
127113
),
128114
"azure/gpt-35-turbo/gpt-35-turbo": AzureModelArgs(
129115
model_name="gpt-35-turbo",

0 commit comments

Comments
 (0)