'commit'

dsLightRag/Liblib/LiblibUtil.py

@@ -1,10 +1,13 @@
 import hmac
+import os
+import tempfile
 from hashlib import sha1
 import base64
 import time
 import uuid
 import requests
 import Config.Config
+from Util.ObsUtil import ObsUploader
 
 
 class LiblibUtil:
@@ -79,7 +82,7 @@ class LiblibUtil:
         payload = {"generateUuid": generate_uuid}
         return self.post_request(uri, payload)
 
-    def wait_for_generation_completion(self, generate_uuid, interval=5):
+    def wait_for_generation_completion(self, generate_uuid, interval=5, max_wait_time=100):
         """等待生成完成并返回最终状态"""
         while True:
             status_data = self.get_generation_status(generate_uuid)