feat: adds doneReason to chat and generation responses (#33)

kevinhermawan · Jul 30, 2024 · 5f36880 · 5f36880
1 parent 85a7c55
commit 5f36880
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/Sources/OllamaKit/Responses/OKChatResponse.swift b/Sources/OllamaKit/Responses/OKChatResponse.swift
@@ -21,6 +21,9 @@ public struct OKChatResponse: OKCompletionResponse, Decodable {
     /// A boolean indicating whether the chat session is complete.
     public let done: Bool
 
+    /// An optional string indicating the cause for the completion of the chat session.
+    public let doneReason: String?
+
     /// An optional integer representing the total duration of processing the request.
     public let totalDuration: Int?
 

diff --git a/Sources/OllamaKit/Responses/OKGenerateResponse.swift b/Sources/OllamaKit/Responses/OKGenerateResponse.swift
@@ -24,6 +24,9 @@ public struct OKGenerateResponse: OKCompletionResponse, Decodable {
     /// A boolean indicating whether the generation process is complete.
     public let done: Bool
 
+    /// An optional string indicating the cause for the completion of the generation.
+    public let doneReason: String?
+
     /// An optional integer representing the total duration of processing the request.
     public let totalDuration: Int?