Skip to content

Commit 13a3f6e

Browse files
committed
feat: added new models (gpt o series, and gemini 1.5 pro)
1 parent bacee51 commit 13a3f6e

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

Diff for: llmware/model_configs.py

+12
Original file line numberDiff line numberDiff line change
@@ -500,6 +500,18 @@
500500
{"model_name": "gpt-4o-mini-2024-07-18", "display_name": "gpt-4o-mini-2024-07-18", "model_family": "OpenAIGenModel",
501501
"model_category": "generative-api", "model_location": "api", "context_window": 128000},
502502

503+
#gpt-o1
504+
{"model_name": "o1", "display_name": "gpt-o1", "model_family": "OpenAIGenModel", "model_category": "generative-api", "model_location": "api", "context_window": 200000},
505+
506+
#gpt-o1-pro
507+
{"model_name": "o1-pro", "display_name": "gpt-o1-pro", "model_family": "OpenAIGenModel", "model_category": "generative-api", "model_location": "api", "context_window": 200000},
508+
509+
#o3-mini-high
510+
{"model_name": "o3-mini", "display_name": "gpt-o3-mini-high", "reasoning_effort": "high", "model_family": "OpenAIGenModel", "model_category": "generative-api", "model_location": "api", "context_window": 200000},
511+
512+
# Google gemini 1.5 pro
513+
{"model_name": "gemini-1.5-pro", "display_name": "Gemini 1.5 Pro", "model_family": "GoogleGenModel", "model_category": "generative-api", "model_location": "api", "context_window": 128000},
514+
503515
# add api-based llmware custom model
504516
{"model_name": "llmware-inference-server", "display_name": "LLMWare-GPT", "model_family": "LLMWareModel",
505517
"model_category": "generative-api", "model_location": "api", "context_window": 2048},

0 commit comments

Comments
 (0)