'commit'

dsLightRag/Test/TestQWen3Image.py

@@ -34,12 +34,11 @@ def test_get_config():
         return False, None
 
 
-def test_generate_image(prompt, n=1, size='1328*1328', save_local=True):
+def test_generate_image(prompt,  size='1328*1328', save_local=True):
     """测试生成图片接口
     
     Args:
         prompt: 图片描述提示词
-        n: 生成图片数量
         size: 图片尺寸
         save_local: 是否保存到本地
         
@@ -51,13 +50,12 @@ def test_generate_image(prompt, n=1, size='1328*1328', save_local=True):
         headers = {"Content-Type": "application/json"}
         data = {
             "prompt": prompt,
-            "n": n,
             "size": size,
             "save_local": save_local
         }
         
         logger.info(f"调用生成图片接口: {url}")
-        logger.info(f"请求参数: prompt={prompt[:50]}..., n={n}, size={size}, save_local={save_local}")
+        logger.info(f"请求参数: prompt={prompt[:50]}...,  size={size}, save_local={save_local}")
         
         # 记录开始时间
         start_time = time.time()
@@ -109,7 +107,6 @@ def main():
     basic_prompt = "一只可爱的小猫在草地上玩耍，阳光明媚，高清细节"
     generate_success, generate_data = test_generate_image(
         prompt=basic_prompt,
-        n=1,
         size="1328*1328",
         save_local=True
     )
@@ -117,13 +114,11 @@ def main():
     # 3. 测试生成图片接口 - 不同参数
     if config_success:
         supported_sizes = config_data["data"].get("supported_sizes", ["1328*1328"])
-        max_images = min(2, config_data["data"].get("max_images_per_request", 1))  # 为了测试效率，最多请求2张
-        
-        logger.info(f"\n3. 测试生成图片接口 - 不同参数(size={supported_sizes[0]}, n={max_images})")
+
+        logger.info(f"\n3. 测试生成图片接口 - 不同参数(size={supported_sizes[0]})")
         different_prompt = "美丽的山水风景画，中国水墨画风格"
         test_generate_image(
             prompt=different_prompt,
-            n=max_images,
             size=supported_sizes[0],
             save_local=True
         )
@@ -132,8 +127,7 @@ def main():
     
     # 输出测试结果摘要
     success_count = sum([config_success, generate_success])
-    total_count = 2  # 基本测试的接口数量
-    logger.info(f"测试结果: {success_count}/{total_count} 接口测试成功")
+    logger.info(f"测试结果: {success_count} 接口测试成功")
 
 
 if __name__ == "__main__":