'commit'

dsLightRag/Liblib/LibLibUtil.py

@@ -114,7 +114,7 @@ class LibLibUtil:
         print(f"生图任务超时（{max_wait_time}秒）")
         return None
 
-    def download_and_upload_to_obs(image_url, generate_uuid):
+    def download_and_upload_to_obs(slef,image_url, generate_uuid):
         """下载图片并上传到OBS"""
         try:
             # 1. 清理URL（去除可能的引号和空格）

dsLightRag/Liblib/T1.py

@@ -1,4 +1,4 @@
-from .LibLibUtil import LibLibUtil
+from LibLibUtil import LibLibUtil
 
 
 if __name__ == '__main__':
@@ -16,23 +16,3 @@ if __name__ == '__main__':
         print(f"模型下载地址：{model_info['modelUrl']}")
     else:
         print("获取模型信息失败")
-    
-    # 测试生成图片
-    print("\n===== 测试生成图片 =====")
-    template_uuid = "7d888009f81d4252a7c458c874cd017f"
-    generate_params = {
-        "checkPointId": "0ea388c7eb854be3ba3c6f65aac6bfd3",
-        "prompt": "Asian portrait,A young woman wearing a green baseball cap",
-        "width": 768,
-        "height": 1152,
-        "sampler": 20,
-        "steps": 35,
-        "cfgScale": 2.0,
-        "imgCount": 1
-    }
-    
-    image_data = liblib.generate_text_to_image(template_uuid, generate_params)
-    if image_data:
-        print(f"图片生成成功，任务UUID: {image_data.get('generateUuid')}")
-    else:
-        print("图片生成失败")