Audio

Author: mlang

openai-servant/src/OpenAI/Api.hs

@@ -1,3 +1,4 @@
+{-# LANGUAGE DataKinds #-}
 -- | The API
 module OpenAI.Api where
 
@@ -6,9 +7,19 @@ import Servant.API
 import Servant.Auth
 import Servant.Auth.Client
 import Servant.Multipart.API
+import Data.List.NonEmpty (NonEmpty((:|)))
+import Data.ByteString.Lazy (ByteString)
 
 type OpenAIAuth = Auth '[Bearer] ()
 
+data Audio
+
+instance Accept Audio where
+  contentTypes _ = "audio/mpeg" :| []
+
+instance MimeUnrender Audio ByteString where
+  mimeUnrender _ bs = Right bs
+
 type OpenAIApi =
   "v1" :> OpenAIApiInternal
 
@@ -46,7 +57,7 @@ type EmbeddingsApi =
   OpenAIAuth :> ReqBody '[JSON] EmbeddingCreate :> Post '[JSON] EmbeddingResponse
 
 type AudioApi =
-  OpenAIAuth :> "speech" :> ReqBody '[JSON] TextToSpeechRequest :> Post '[OctetStream] TextToSpeechResponse
+  OpenAIAuth :> "speech" :> ReqBody '[JSON] TextToSpeechRequest :> Post '[Audio] TextToSpeechResponse
     :<|> OpenAIAuth :> "transcriptions" :> MultipartForm Tmp AudioTranscriptionRequest :> Post '[JSON] AudioResponseData
     :<|> OpenAIAuth :> "translations" :> MultipartForm Tmp AudioTranslationRequest :> Post '[JSON] AudioResponseData