Add enum Chat.Role…

So that `Chat` has a safer initializer `init(role: Role, content: String)` than `init(role: String, content: String)`
MacPaw · Mar 10, 2023 · 35d97bf · 35d97bf
1 parent d0493cf
commit 35d97bf
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/Sources/OpenAI/OpenAI.swift b/Sources/OpenAI/OpenAI.swift
@@ -147,10 +147,20 @@ public extension OpenAI {
         public let role: String
         public let content: String
 
+        public enum Role: String {
+            case system
+            case assistant
+            case user
+        }
+
         public init(role: String, content: String) {
             self.role = role
             self.content = content
         }
+
+        public init(role: Role, content: String) {
+            self.init(role: role.rawValue, content: content)
+        }
     }
 
     struct ChatQuery: Codable {