From de883ca5fb10390ac9ef5ab91373a14a8e8c64a5 Mon Sep 17 00:00:00 2001
From: GitPusher99 <robinyang.beijing@gmail.com>
Date: Thu, 28 Mar 2024 23:04:20 +0800
Subject: [PATCH] fix: refresh token

---
 src/lib/SunoApi.ts | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/lib/SunoApi.ts b/src/lib/SunoApi.ts
index b34acbd..92e27b9 100644
--- a/src/lib/SunoApi.ts
+++ b/src/lib/SunoApi.ts
@@ -90,6 +90,7 @@ class SunoApi {
     make_instrumental: boolean = false,
     wait_audio: boolean = false,
   ): Promise<AudioInfo[]> {
+    await this.keepAlive(false);
     const startTime = Date.now();
     const audios = this.generateSongs(prompt, false, undefined, undefined, make_instrumental, wait_audio);
     const costTime = Date.now() - startTime;
@@ -115,6 +116,7 @@ class SunoApi {
     make_instrumental: boolean = false,
     wait_audio: boolean = false,
   ): Promise<AudioInfo[]> {
+    await this.keepAlive(false);
     const startTime = Date.now();
     const audios = await this.generateSongs(prompt, true, tags, title, make_instrumental, wait_audio);
     const costTime = Date.now() - startTime;
@@ -142,6 +144,7 @@ class SunoApi {
     make_instrumental?: boolean,
     wait_audio: boolean = false
   ): Promise<AudioInfo[]> {
+    await this.keepAlive(false);
     const payload: any = {
       make_instrumental: make_instrumental == true,
       mv: "chirp-v3-0",
@@ -239,6 +242,7 @@ class SunoApi {
    * @returns A promise that resolves to an array of AudioInfo objects.
    */
   public async get(songIds?: string[]): Promise<AudioInfo[]> {
+    await this.keepAlive(false);
     let url = `${SunoApi.BASE_URL}/api/feed/`;
     if (songIds) {
       url = `${url}?ids=${songIds.join(',')}`;
@@ -269,6 +273,7 @@ class SunoApi {
   }
 
   public async get_credits(): Promise<object> {
+    await this.keepAlive(false);
     const response = await this.client.get(`${SunoApi.BASE_URL}/api/billing/info/`);
     return {
       credits_left: response.data.total_credits_left,