🎷 [UPDATE] Tool choice added.

Author: burakcolakappcent

Sources/ACMOpenAI/Requests/Chat/ACMOAIChatRequest.swift

@@ -38,7 +38,7 @@ public extension ACMOAIChatRequest {
         public var user: String?
 
         /// Init function for creating request
-        public init(model: String, messages: [ACMChatMessageModel], tools: [ACMToolsModel]? = nil, 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], tools: [ACMToolsModel]? = nil, tool_choice: ACMToolChoiceModel? = nil, 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
@@ -52,6 +52,7 @@ public extension ACMOAIChatRequest {
             self.logit_bias = logit_bias
             self.user = user
             self.tools = tools
+            self.tool_choice = tool_choice
         }
     }
 }

Sources/ACMOpenAI/Requests/Chat/ACMToolsModel.swift

@@ -20,12 +20,16 @@ public struct ACMToolsModel: Codable {
 public struct ACMToolsFunctionModel: Codable {
     public let description: String?
     public let name: String
-    public let params: String?
+    public let parameters: String?
 
     public init(description: String? = nil, name: String, params: [String: Any]? = nil) {
         self.description = description
         self.name = name
-        self.params = String(describing: params)
+        if let params {
+            self.parameters = String(describing: params)
+        } else {
+            self.parameters = nil
+        }
     }
 }
 
@@ -45,8 +49,17 @@ public struct ACMToolChoiceModel: Codable {
 public struct ACMToolChoiceObjectModel: Codable {
     public var type: String
     public var function: ACMToolChoiceObjectFunctionModel
+
+    public init(type: String, function: ACMToolChoiceObjectFunctionModel) {
+        self.type = type
+        self.function = function
+    }
 }
 
 public struct ACMToolChoiceObjectFunctionModel: Codable {
     public var name: String
+
+    public init(name: String) {
+        self.name = name
+    }
 }