Skip to content

Commit a86f8c2

Browse files
committed
feature : more detailed model class map
help from Grok4 Some models' names do not start with the providers'
1 parent ab347e7 commit a86f8c2

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

entrypoint.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -166,10 +166,15 @@ def get_config_class_dict() -> Dict[str, type]:
166166
"""
167167
return {
168168
'claude': ClaudeConfig,
169+
'claude-sonnet-4-20250514': ClaudeConfig, # Add specific model
169170
'gemini': GeminiConfig,
171+
'gemini-2.5-flash': GeminiConfig,
170172
'grok': GrokConfig,
173+
'grok-code-fast': GrokConfig,
171174
'nvidia_nim': NvidiaNIMConfig,
175+
'google/gemma-2-9b-it': NvidiaNIMConfig, # Add specific model
172176
'perplexity': PerplexityConfig,
177+
'sonar': PerplexityConfig # Add specific model
173178
}
174179

175180

0 commit comments

Comments
 (0)