Add prompt as an optional input to transcription params (#17)

rakyll · Mar 21, 2023 · d85e436 · d85e436
1 parent 03cd225
commit d85e436
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/audio/audio.go b/audio/audio.go
@@ -37,10 +37,14 @@ func NewClient(session *openai.Session, model string) *Client {
 }
 
 type CreateTranscriptionParams struct {
-	Model       string
-	Language    string
+	Model    string
+	Language string
+
 	Audio       io.Reader
 	AudioFormat string // such as "mp3" or "wav", etc.
+
+	Prompt string // optional
+	// TODO: Add temperature.
 }
 
 type CreateTranscriptionResponse struct {
@@ -59,6 +63,9 @@ func (c *Client) CreateTranscription(ctx context.Context, p *CreateTranscription
 	if p.Language != "" {
 		params.Set("language", p.Language)
 	}
+	if p.Prompt != "" {
+		params.Set("prompt", p.Prompt)
+	}
 	var r CreateTranscriptionResponse
 	return &r, c.s.Upload(ctx, c.CreateTranscriptionEndpoint, p.Audio, p.AudioFormat, params, &r)
 }