diff --git a/src/interfaces.ts b/src/interfaces.ts
index 1acefd0..3baea05 100644
--- a/src/interfaces.ts
+++ b/src/interfaces.ts
@@ -124,6 +124,7 @@ export interface GenerateResponse {
   created_at: Date
   response: string
   done: boolean
+  done_reason: string
   context: number[]
   total_duration: number
   load_duration: number
@@ -138,6 +139,7 @@ export interface ChatResponse {
   created_at: Date
   message: Message
   done: boolean
+  done_reason: string
   total_duration: number
   load_duration: number
   prompt_eval_count: number