Add creation transcription method that returns string

exyte · Sep 10, 2024 · 38e8ce3 · 38e8ce3
1 parent d925c29
commit 38e8ce3
Showing 1 changed file with 36 additions and 0 deletions.
diff --git a/Sources/ExyteOpenAI/OpenAI+Audio.swift b/Sources/ExyteOpenAI/OpenAI+Audio.swift
@@ -41,13 +41,41 @@ public extension OpenAI {
             .eraseToAnyPublisher()
     }
 
+    func createTranscription(from payload: CreateTranscriptionPayload) -> AnyPublisher<String, OpenAIError> {
+        audioProvider.requestPublisher(for: .createTranscription(payload: payload))
+            .flatMap {
+                guard let stringData = String(data: $0.data, encoding: .utf8) else {
+                    return Fail<String, OpenAIError>(error: .requestCreationFailed)
+                        .eraseToAnyPublisher()
+                }
+                return Just<String>(stringData)
+                    .setFailureType(to: OpenAIError.self)
+                    .eraseToAnyPublisher()
+            }
+            .eraseToAnyPublisher()
+    }
+
     func createTranslation(from payload: CreateTranslationPayload) -> AnyPublisher<Translation, OpenAIError> {
         audioProvider.requestPublisher(for: .createTranslation(payload: payload))
             .map { $0.data }
             .map(to: Translation.self, decoder: OpenAI.defaultDecoder)
             .eraseToAnyPublisher()
     }
 
+    func createTranslation(from payload: CreateTranslationPayload) -> AnyPublisher<String, OpenAIError> {
+        audioProvider.requestPublisher(for: .createTranslation(payload: payload))
+            .flatMap {
+                guard let stringData = String(data: $0.data, encoding: .utf8) else {
+                    return Fail<String, OpenAIError>(error: .requestCreationFailed)
+                        .eraseToAnyPublisher()
+                }
+                return Just<String>(stringData)
+                    .setFailureType(to: OpenAIError.self)
+                    .eraseToAnyPublisher()
+            }
+            .eraseToAnyPublisher()
+    }
+
 }
 
 // MARK: - Concurrency
@@ -62,8 +90,16 @@ public extension OpenAI {
         try await createTranscription(from: payload).async()
     }
 
+    func createTranscription(from payload: CreateTranscriptionPayload) async throws -> String {
+        try await createTranscription(from: payload).async()
+    }
+
     func createTranslation(from payload: CreateTranslationPayload) async throws -> Translation {
         try await createTranslation(from: payload).async()
     }
 
+    func createTranslation(from payload: CreateTranslationPayload) async throws -> String {
+        try await createTranslation(from: payload).async()
+    }
+
 }