Adds error message management on generation

This commit is contained in:
Carlos Valencia 2024-06-04 17:04:55 -05:00
parent bb55059a4a
commit 299e94f592

View File

@ -23,6 +23,7 @@ export interface AudioInfo {
type?: string; type?: string;
tags?: string; // Genre of music. tags?: string; // Genre of music.
duration?: string; // Duration of the audio duration?: string; // Duration of the audio
error_message?: string; // Error message if any
} }
class SunoApi { class SunoApi {
@ -227,7 +228,10 @@ class SunoApi {
const allCompleted = response.every( const allCompleted = response.every(
audio => audio.status === 'streaming' || audio.status === 'complete' audio => audio.status === 'streaming' || audio.status === 'complete'
); );
if (allCompleted) { const allError = response.every(
audio => audio.status === 'error'
);
if (allCompleted || allError) {
return response; return response;
} }
lastResponse = response; lastResponse = response;
@ -358,6 +362,7 @@ class SunoApi {
type: audio.metadata.type, type: audio.metadata.type,
tags: audio.metadata.tags, tags: audio.metadata.tags,
duration: audio.metadata.duration_formatted, duration: audio.metadata.duration_formatted,
error_message: audio.metadata.error_message,
})); }));
} }