fix: context must be an array of integers (#10)

kevinhermawan · Jan 15, 2024 · 1873022 · 1873022
1 parent 1b98b7c
commit 1873022
Showing 1 changed file with 2 additions and 2 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?