Add more models

alexshapalov · Oct 21, 2024 · 3a39719 · 3a39719
1 parent a55aec5
commit 3a39719
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/rubyai/configuration.rb b/lib/rubyai/configuration.rb
@@ -9,7 +9,11 @@ class Configuration
       "gpt-4" => "gpt-4",
       "gpt-4-32k" => "gpt-4-32k",
       "gpt-3.5-turbo" => "gpt-3.5-turbo",
-      "text-davinci-003" => "text-davinci-003"
+      "text-davinci-003" => "text-davinci-003",
+      "gpt-4o-mini" => "gpt-4o-mini",
+      "o1-preview" => "o1-preview",
+      "o1-mini" => "o1-mini",
+      "gpt-4-turbo" => "gpt-4-turbo"
     }
 
     DEFAULT_MODEL = "gpt-3.5-turbo"