Merge pull request #15 from tomdai/add-enum-chat-role

Add enum `Chat.Role`…
MacPaw · Mar 10, 2023 · 6337cba · 6337cba
2 parents d0493cf + 35d97bf
commit 6337cba
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 {