highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-08-03 22:53:02 +02:00
Code Issues Packages Projects Releases Wiki Activity

add done_reason to the api (#4235)

Browse Source
This commit is contained in:
Bruce MacDonald
2024-05-09 13:30:14 -07:00
committed by GitHub
parent 1580ed4c06
commit cfa84b8470
4 changed files with 44 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
api/types.go
View File

@@ -114,9 +114,10 @@ type Message struct {
// ChatResponse is the response returned by [Client.Chat]. Its fields are
// similar to [GenerateResponse].
type ChatResponse struct {
Model string `json:"model"`
CreatedAt time.Time `json:"created_at"`
Message Message `json:"message"`
Model string `json:"model"`
CreatedAt time.Time `json:"created_at"`
Message Message `json:"message"`
DoneReason string `json:"done_reason"`
Done bool `json:"done"`
@@ -309,6 +310,9 @@ type GenerateResponse struct {
// Done specifies if the response is complete.
Done bool `json:"done"`
// DoneReason is the reason the model stopped generating text.
DoneReason string `json:"done_reason"`
// Context is an encoding of the conversation used in this response; this
// can be sent in the next request to keep a conversational memory.
Context []int `json:"context,omitempty"`