transcribe

This commit is contained in:
Roy Han 2024-08-05 14:16:32 -07:00
parent 17f9dc6d08
commit e4d35198a2
2 changed files with 13 additions and 1 deletions

View File

@ -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].

View File

@ -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)