Updated to support gpt-4o

README.md

@@ -8,6 +8,7 @@ The functionality shown here serves as an interface to the ChatGPT and DALL·E A
 
 Some of the current LLMs supported are:
 - gpt-3.5-turbo, gpt-3.5-turbo-1106, gpt-3.5-turbo-0125
+- gpt-4o, gpt-4o-2024-05-13 (GPT-4 Omni)
 - gpt-4-turbo, gpt-4-turbo-2024-04-09 (GPT-4 Turbo with Vision)
 - gpt-4, gpt-4-0613
 - dall-e-2, dall-e-3

openAIChat.m

@@ -114,7 +114,8 @@
             arguments
                 systemPrompt                       {llms.utils.mustBeTextOrEmpty} = []
                 nvp.Tools                    (1,:) {mustBeA(nvp.Tools, "openAIFunction")} = openAIFunction.empty
-                nvp.ModelName                (1,1) {mustBeMember(nvp.ModelName,["gpt-4-turbo", ...
+                nvp.ModelName                (1,1) {mustBeMember(nvp.ModelName,["gpt-4o", ...
+                                                        "gpt-4o-2024-05-13","gpt-4-turbo", ...
                                                         "gpt-4-turbo-2024-04-09","gpt-4","gpt-4-0613", ...
                                                         "gpt-3.5-turbo","gpt-3.5-turbo-0125", ...
                                                         "gpt-3.5-turbo-1106"])} = "gpt-3.5-turbo"
@@ -217,7 +218,7 @@
             end
 
             if iscell(messagesStruct{end}.content) && any(cellfun(@(x) isfield(x,"image_url"), messagesStruct{end}.content))
-                if ~ismember(this.ModelName,["gpt-4-turbo","gpt-4-turbo-2024-04-09"]) 
+                if ~ismember(this.ModelName,["gpt-4-turbo","gpt-4-turbo-2024-04-09","gpt-4o","gpt-4o-2024-05-13"]) 
                  error("llms:invalidContentTypeForModel", ...
                        llms.utils.errorMessageCatalog.getMessage("llms:invalidContentTypeForModel", "Image content", this.ModelName));
                 end

tests/topenAIChat.m

@@ -159,6 +159,11 @@ function createOpenAIChatWithOpenAIKey(testCase)
                 ApiKey=getenv("OPENAI_KEY"));
 
             testCase.verifyWarningFree(@()generate(chat, "Hello world."));
+
+            chat = openAIChat("You are a useful assistant", ...
+                ApiKey=getenv("OPENAI_KEY"), ModelName="gpt-4o");
+
+            testCase.verifyWarningFree(@()generate(chat, "Hello world."));
         end
 
     end