Add optional 'additionalProperties' param

MacPaw · Aug 30, 2024 · c5bbd80 · c5bbd80
1 parent 3c970ea
commit c5bbd80
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/Sources/OpenAI/Public/Models/ChatQuery.swift b/Sources/OpenAI/Public/Models/ChatQuery.swift
@@ -703,6 +703,7 @@ public struct ChatQuery: Equatable, Codable, Streamable {
                 public let multipleOf: Int?
                 public let minimum: Int?
                 public let maximum: Int?
+                public let additionalProperties: Bool?
 
                 public init(
                     type: Self.JSONType,
@@ -713,7 +714,8 @@ public struct ChatQuery: Equatable, Codable, Streamable {
                     enum: [String]? = nil,
                     multipleOf: Int? = nil,
                     minimum: Int? = nil,
-                    maximum: Int? = nil
+                    maximum: Int? = nil,
+                    additionalProperties: Bool? = nil
                 ) {
                     self.type = type
                     self.properties = properties
@@ -724,6 +726,7 @@ public struct ChatQuery: Equatable, Codable, Streamable {
                     self.multipleOf = multipleOf
                     self.minimum = minimum
                     self.maximum = maximum
+                    self.additionalProperties = additionalProperties
                 }
 
                 public struct Property: Codable, Equatable {

diff --git a/Tests/OpenAITests/OpenAITestsDecoder.swift b/Tests/OpenAITests/OpenAITestsDecoder.swift
@@ -264,7 +264,8 @@ class OpenAITestsDecoder: XCTestCase {
                           "location": .init(type: .string, description: "The city and state, e.g. San Francisco, CA"),
                           "unit": .init(type: .string, enum: ["celsius", "fahrenheit"])
                         ],
-                        required: ["location"]
+                        required: ["location"],
+                        additionalProperties: false
                       ),
                     strict: true
                 ))
@@ -293,7 +294,8 @@ class OpenAITestsDecoder: XCTestCase {
                     },
                     "unit": { "type": "string", "enum": ["celsius", "fahrenheit"] }
                   },
-                  "required": ["location"]
+                  "required": ["location"],
+                  "additionalProperties": false
                 },
                 "strict": true
               },