diff --git a/Sources/OllamaKit/RequestData/OKGenerateRequestData.swift b/Sources/OllamaKit/RequestData/OKGenerateRequestData.swift
index 6cab851..bda80f1 100644
--- 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
index 1d9afc9..fe52331 100644
--- 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