From e4d35198a23efe47890583d837881cc532e993cf Mon Sep 17 00:00:00 2001 From: Roy Han Date: Mon, 5 Aug 2024 14:16:32 -0700 Subject: [PATCH] transcribe --- api/types.go | 2 ++ server/routes.go | 12 +++++++++++- 2 files changed, 13 insertions(+), 1 deletion(-) diff --git a/api/types.go b/api/types.go index 29e5a699..6789feca 100644 --- a/api/types.go +++ b/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]. diff --git a/server/routes.go b/server/routes.go index 308970bc..0e1cf52e 100644 --- a/server/routes.go +++ b/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)