refactor: changes type from Double to Int in context (#19)

kevinhermawan · May 7, 2024 · d97be57 · d97be57
1 parent 046951a
commit d97be57
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/Sources/OllamaKit/RequestData/OKGenerateRequestData.swift b/Sources/OllamaKit/RequestData/OKGenerateRequestData.swift
@@ -23,8 +23,8 @@ public struct OKGenerateRequestData: Encodable {
     /// An optional string specifying the system message.
     public var system: String?
 
-    /// An optional array of doubles representing contextual information.
-    public var context: [Double]?
+    /// An optional array of integers representing contextual information.
+    public var context: [Int]?
 
     /// Optional ``OKCompletionOptions`` providing additional configuration for the generation request.
     public var options: OKCompletionOptions?

diff --git a/Sources/OllamaKit/Responses/OKGenerateResponse.swift b/Sources/OllamaKit/Responses/OKGenerateResponse.swift
@@ -18,8 +18,8 @@ public struct OKGenerateResponse: OKCompletionResponse, Decodable {
     /// A string containing the generated content.
     public let response: String
 
-    /// An optional array of doubles representing contextual information used in the generation.
-    public let context: [Double]?
+    /// An optional array of integers representing contextual information used in the generation.
+    public let context: [Int]?
 
     /// A boolean indicating whether the generation process is complete.
     public let done: Bool