Merge pull request #154 from ollama/done-reason

add done reason to generate type responses
ollama · May 10, 2024 · cb81f52 · cb81f52
2 parents 5dc857e + eaab477
commit cb81f52
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ollama/_types.py b/ollama/_types.py
@@ -19,6 +19,9 @@ class BaseGenerateResponse(TypedDict):
   done: bool
   'True if response is complete, otherwise False. Useful for streaming to detect the final response.'
 
+  done_reason: str
+  'Reason for completion. Only present when done is True.'
+
   total_duration: int
   'Total duration in nanoseconds.'