'commit'

dsLightRag/Liblib/T1.py

@@ -0,0 +1,38 @@
+from .LiblibUtil import LiblibUtil
+
+
+if __name__ == '__main__':
+    liblib = LiblibUtil()
+    
+    # 测试获取模型版本信息
+    print("===== 测试获取模型版本信息 =====")
+    version_uuid = "390e7df48aed45d4bc3f0bcbc89fa44c"
+    model_info = liblib.get_model_version_info(version_uuid)
+    
+    if model_info:
+        print(f"模型名称：{model_info['modelName']}")
+        print(f"版本名称：{model_info['versionName']}")
+        print(f"是否允许商业用途：{model_info['commercialUse']}")
+        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("图片生成失败")