update generate response

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"`
-}

server/routes.go

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