Merge pull request #39 from AppcentMobile/feature/stream-support

Feature/stream support

Author: burakcolakappcent

Package.resolved

@@ -5,14 +5,17 @@ import PackageDescription
 
 let package = Package(
     name: "ACMOpenAI",
+    platforms: [
+        .iOS(.v13), // Set the minimum iOS version here
+    ],
     products: [
         .library(
             name: "ACMOpenAI",
             targets: ["ACMOpenAI"]
         ),
     ],
     dependencies: [
-        .package(url: "https://github.com/AppcentMobile/ACMNetworking.git", from: "1.1.3"),
+        .package(url: "https://github.com/AppcentMobile/ACMNetworking.git", branch: "main"),
     ],
     targets: [
         .target(

Package.swift

@@ -7,5 +7,5 @@ public struct ChatCallback {}
 
 public extension ChatCallback {
     /// Create chat typealias
-    typealias Create = ((ACMOAIChatResponse.Create?) -> Void)?
+    typealias Create = (([ACMOAIChatResponse.Create?]) -> Void)?
 }

Sources/ACMOpenAI/Callbacks/Chat/ChatCallback.swift

@@ -3,7 +3,7 @@
 //
 
 /// Constants for holding open ai data
-struct ACMOAIConstants {
+enum ACMOAIConstants {
     static let organizationHeader = "OpenAI-Organization"
     static let contentTypeKey = "Content-Type"
     static let contentTypeValue = "application/json"

Sources/ACMOpenAI/Constants/ACMOAIConstants.swift

@@ -60,10 +60,20 @@ public extension ACMOAIChatManager {
             to = to.add(param: ACMBodyModel(key: "user", value: user))
         }
 
-        network.request(to: to.build()) { (response: ACMOAIChatResponse.Create) in
-            onSuccess?(response)
-        } onError: { error in
-            onError?(error)
+        let buildEndpoint = to.build()
+
+        if request.stream == true {
+            network.request(to: buildEndpoint) { (response: [ACMOAIChatResponse.Create]) in
+                onSuccess?(response)
+            } onError: { error in
+                onError?(error)
+            }
+        } else {
+            network.request(to: buildEndpoint) { (response: ACMOAIChatResponse.Create) in
+                onSuccess?([response])
+            } onError: { error in
+                onError?(error)
+            }
         }
     }
 }

Sources/ACMOpenAI/Manager/Chat/ACMOAIChatManager.swift

@@ -0,0 +1,24 @@
+//
+//  ACMChatMessageModel.swift
+//
+//
+//  Created by burak on 18.12.2023.
+//
+
+import UIKit
+
+public final class ACMChatMessageModel: Codable {
+    public let role: String?
+    public let content: String?
+
+    public init(from decoder: Decoder) throws {
+        let container = try decoder.container(keyedBy: CodingKeys.self)
+        role = try container.decodeIfPresent(String.self, forKey: .role)
+        content = try container.decodeIfPresent(String.self, forKey: .content)
+    }
+
+    public init(role: String?, content: String?) {
+        self.role = role
+        self.content = content
+    }
+}

Sources/ACMOpenAI/Requests/Chat/ACMChatMessageModel.swift

@@ -11,7 +11,7 @@ public extension ACMOAIChatRequest {
         /// model: required*
         public var model: String
         /// messages: required*
-        public var messages: [[String: String]]
+        public var messages: [ACMChatMessageModel]
         /// temperature: optional
         public var temperature: Double?
         /// top_p: optional
@@ -34,7 +34,7 @@ public extension ACMOAIChatRequest {
         public var user: String?
 
         /// Init function for creating request
-        public init(model: String, messages: [[String: String]], temperature: Double? = nil, top_p: Double? = nil, n: Int? = nil, stream: Bool? = nil, stop: [String]? = nil, max_tokens: Int? = nil, presence_penalty: Double? = nil, frequency_penalty: Double? = nil, logit_bias: [String: Int]? = nil, user: String? = nil) {
+        public init(model: String, messages: [ACMChatMessageModel], temperature: Double? = nil, top_p: Double? = nil, n: Int? = nil, stream: Bool? = nil, stop: [String]? = nil, max_tokens: Int? = nil, presence_penalty: Double? = nil, frequency_penalty: Double? = nil, logit_bias: [String: Int]? = nil, user: String? = nil) {
             self.model = model
             self.messages = messages
             self.temperature = temperature