Roy Han 9 mesi fa
parent
commit
e4d35198a2
2 ha cambiato i file con 13 aggiunte e 1 eliminazioni
  1. 2 0
      api/types.go
  2. 11 1
      server/routes.go

+ 2 - 0
api/types.go

@@ -82,6 +82,8 @@ type GenerateRequest struct {
 	Options map[string]interface{} `json:"options"`
 
 	Audio string `json:"audio,omitempty"`
+
+	Transcribe bool `json:"transcribe,omitempty"`
 }
 
 // ChatRequest describes a request sent by [Client.Chat].

+ 11 - 1
server/routes.go

@@ -287,7 +287,17 @@ func (s *Server) GenerateHandler(c *gin.Context) {
 			return
 		}
 
-		req.Prompt = w.Text
+		if req.Transcribe {
+			c.JSON(http.StatusOK, api.GenerateResponse{
+				Model:      req.Model,
+				CreatedAt:  time.Now().UTC(),
+				Response:   w.Text,
+				Done:       true,
+				DoneReason: "stop",
+			})
+		}
+
+		req.Prompt += w.Text
 	}
 
 	r, m, opts, err := s.scheduleRunner(c.Request.Context(), req.Model, caps, req.Options, req.KeepAlive)