add done_reason to chat and generate responses

ollama · May 9, 2024 · 2537e92 · 2537e92
1 parent a0b47fe
commit 2537e92
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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