Skip to content

Commit 875ffd7

Browse files
undo removed llm_config
1 parent 231175c commit 875ffd7

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/agentlab/llm/llm_configs.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,13 @@
9595
max_new_tokens=64_000,
9696
temperature=1e-1,
9797
),
98+
"openai/gpt-4.1-nano-2025-04-14": OpenAIModelArgs(
99+
model_name="gpt-4.1-nano-2025-04-14",
100+
max_total_tokens=128_000,
101+
max_input_tokens=128_000,
102+
max_new_tokens=16_384,
103+
vision_support=True,
104+
),
98105
"openai/gpt-5-nano-2025-08-07": OpenAIModelArgs(
99106
model_name="gpt-5-nano-2025-08-07",
100107
max_total_tokens=400_000,
@@ -119,6 +126,7 @@
119126
temperature=1, # temperature param not supported by gpt-5
120127
vision_support=True,
121128
),
129+
# ---------------- Azure ----------------#
122130
"azure/gpt-35-turbo/gpt-35-turbo": AzureModelArgs(
123131
model_name="gpt-35-turbo",
124132
max_total_tokens=8_192,

0 commit comments

Comments
 (0)