update generate response

dd960d1d · Michael Yang · Jeffrey Morgan · 2c80eddd · dd960d1d · dd960d1d
Commit dd960d1d authored Jul 06, 2023 by Michael Yang Committed by Jeffrey Morgan Jul 06, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 14 deletions

api/types.go api/types.go +0 -8

server/routes.go server/routes.go +2 -6

No files found.
--- a/api/types.go
+++ b/api/types.go
@@ -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"`
-}
--- a/server/routes.go
+++ b/server/routes.go
@@ -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)