Add GPT-4 models. Update tests

MacPaw · Mar 19, 2023 · 2609478 · 2609478
1 parent 42eb848
commit 2609478
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/Sources/OpenAI/Public/Models.swift b/Sources/OpenAI/Public/Models.swift
@@ -20,4 +20,9 @@ public extension Model {
     static let textEmbeddingAda = "text-embedding-ada-002"
     static let gpt3_5Turbo = "gpt-3.5-turbo"
     static let gpt3_5Turbo0301 = "gpt-3.5-turbo-0301"
+
+    static let gpt4 = "gpt-4"
+    static let gpt4_0134 = "gpt-4-0314"
+    static let gpt4_32k = "gpt-4-32k"
+    static let gpt4_32k_0314 = "gpt-4-32k-0314"
 }
diff --git a/Tests/OpenAITests/OpenAITests.swift b/Tests/OpenAITests/OpenAITests.swift
@@ -55,7 +55,10 @@ final class OpenAITests: XCTestCase {
     }
 
     func testChatsAsync() async throws {
-        let query = OpenAI.ChatQuery(model: .gpt3_5Turbo, messages: [.init(role: "user", content: "who are you")], temperature: 1, top_p: 1, n: 1, stream: false, stop: nil, max_tokens: 100, presence_penalty: 0, frequency_penalty: 0, logit_bias: nil, user: nil)
+        let query = OpenAI.ChatQuery(model: .gpt4, messages: [
+            .init(role: .system, content: "You are Librarian-GPT. You know everything about the books."),
+            .init(role: .user, content: "Who wrote Harry Potter?")
+        ])
         let result = try await openAI.chats(query: query)
         XCTAssertFalse(result.choices.isEmpty)
     }