Update topP parameter (#40)

SwiftBeta · Apr 25, 2024 · 96ce84e · 96ce84e
1 parent 3afefca
commit 96ce84e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/Sources/SwiftOpenAI/OpenAI/DataModels/Chat/ChatCompletionsOptionalParameters.swift b/Sources/SwiftOpenAI/OpenAI/DataModels/Chat/ChatCompletionsOptionalParameters.swift
@@ -2,15 +2,15 @@ import Foundation
 
 public struct ChatCompletionsOptionalParameters {
     public let temperature: Double?
-    public let topP: Int?
+    public let topP: Double?
     public let n: Int?
     public let stop: [String]?
     public let stream: Bool
     public let maxTokens: Int?
     public let user: String?
 
     public init(temperature: Double = 1.0,
-                topP: Int = 1,
+                topP: Double = 1.0,
                 n: Int = 1,
                 stop: [String]? = nil,
                 stream: Bool = false,

diff --git a/Sources/SwiftOpenAI/OpenAI/DataModels/Completions/CompletionsOptionalParameters.swift b/Sources/SwiftOpenAI/OpenAI/DataModels/Completions/CompletionsOptionalParameters.swift
@@ -5,7 +5,7 @@ public struct CompletionsOptionalParameters {
     public let suffix: String?
     public let maxTokens: Int?
     public let temperature: Double?
-    public let topP: Int?
+    public let topP: Double?
     public let n: Int?
     public let logprobs: Int?
     public let echo: Bool?
@@ -19,7 +19,7 @@ public struct CompletionsOptionalParameters {
                 suffix: String = "",
                 maxTokens: Int? = 16,
                 temperature: Double? = 1.0,
-                topP: Int? = 1,
+                topP: Double? = 1.0,
                 n: Int? = 1,
                 logprobs: Int? = nil,
                 echo: Bool? = false,