highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-03-26 17:51:48 +01:00
Code Issues Packages Projects Releases Wiki Activity

update generate response

Browse Source
This commit is contained in:
Michael Yang 2023-07-06 12:27:50 -07:00 committed by Jeffrey Morgan
parent 2c80eddd71
commit dd960d1d5e
2 changed files with 2 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/types.go
View File

@ -36,11 +36,3 @@ type GenerateRequest struct {
type GenerateResponse struct {
Response string `json:"response"`
}
type TokenResponse struct {
Choices []TokenResponseChoice `json:"choices"`
}
type TokenResponseChoice struct {
Text string `json:"text"`
}

8
server/routes.go
View File

@ -78,12 +78,8 @@ func generate(c *gin.Context) {
return false
}
resp := api.TokenResponse{
Choices: []api.TokenResponseChoice{
{
Text: token,
},
},
resp := api.GenerateResponse{
Response: token,
}
bts, err := json.Marshal(resp)