Browse Source

add done_reason to chat and generate responses (#87)

Bruce MacDonald 1 year ago
parent
commit
0a5f7de6d7
1 changed files with 2 additions and 0 deletions
  1. 2 0
      src/interfaces.ts

+ 2 - 0
src/interfaces.ts

@@ -125,6 +125,7 @@ export interface GenerateResponse {
   created_at: Date
   created_at: Date
   response: string
   response: string
   done: boolean
   done: boolean
+  done_reason: string
   context: number[]
   context: number[]
   total_duration: number
   total_duration: number
   load_duration: number
   load_duration: number
@@ -139,6 +140,7 @@ export interface ChatResponse {
   created_at: Date
   created_at: Date
   message: Message
   message: Message
   done: boolean
   done: boolean
+  done_reason: string
   total_duration: number
   total_duration: number
   load_duration: number
   load_duration: number
   prompt_eval_count: number
   prompt_eval_count: number