root 3 săptămâni în urmă
părinte
comite
29551ac454
1 a modificat fișierele cu 25 adăugiri și 28 ștergeri
  1. 25 28
      ai-slibra-assistant/util/request_utils.py

+ 25 - 28
ai-slibra-assistant/util/request_utils.py

@@ -73,41 +73,38 @@ def get_voice(content, is_local, silence_duration):
             "silence_duration":silence_duration
         }
     }
-    retry = 1
-    while retry <=3:
-        try:
-            resp = requests.post(api_url, json.dumps(request_json), headers=header)
-            rfile = f"{rid}.wav"
-            path, file = f"{VOICE_DIR}/{rfile}", f"../voice/{rfile}"
-            if "data" in resp.json():
-                data = resp.json()["data"]
-                file_to_save = open(file, "wb")
-                file_to_save.write(base64.b64decode(data))
-                if is_local:
+    try:
+        resp = requests.post(api_url, json.dumps(request_json), headers=header)
+        rfile = f"{rid}.wav"
+        path, file = f"{VOICE_DIR}/{rfile}", f"../voice/{rfile}"
+        if "data" in resp.json():
+            data = resp.json()["data"]
+            file_to_save = open(file, "wb")
+            file_to_save.write(base64.b64decode(data))
+            if is_local:
+                if os.path.exists(file):
+                    logger.info(f"voice local file ::session_id={rid}, res={file}")
+                    return path
+            else:
+                files = {'file': open(file, 'rb')}
+                response = requests.post(DEFAULT_FILE_UPLOAD_URL, files=files)
+                if response.ok:
+                    result = json.loads(response.text)
+                    url = result.get('data')
                     if os.path.exists(file):
-                        logger.info(f"voice local file ::session_id={rid}, res={file}")
-                        return path
-                else:
-                    files = {'file': open(file, 'rb')}
-                    response = requests.post(DEFAULT_FILE_UPLOAD_URL, files=files)
-                    if response.ok:
-                        result = json.loads(response.text)
-                        url = result.get('data')
-                        if os.path.exists(file):
-                            os.system(f"rm -fr {file}")
-                        logger.info(f"voice upload_plot::session_id={rid}, res={url}")
-                        return url
-        except Exception as e:
-            retry +=1
-            time.sleep(2)
-            logger.info(f"voice generate 错误{e} retry:{retry}")
+                        os.system(f"rm -fr {file}")
+                    logger.info(f"voice upload_plot::session_id={rid}, res={url}")
+                    return url
+    except Exception as e:
+        logger.info(f"voice generate 错误{e}")
+
 
 @timetic
 def voice_service(content, local=False, silence_duration="125"):
     encode_b = content.encode("utf-8")
     key = base64.b64encode(encode_b)
     name = "voice_url"
-    retry,num =1,1
+    retry,num =3,1
     while num <= retry:
         try:
             url = r.hget(name, key)