fix: kling query

Author: timerring

docs/zh/models/kling/api/query-api.md

@@ -13,10 +13,19 @@ updatedAt: 2025-05-08
 
 ## 请求
 
+> [!TIP]
+> `{api_url}` 为你实际使用的 API 节点，请根据实际情况填写。例如：
+> - `www.dmxapi.cn`
+> - `www.dmxapi.com`
+> - `ssvip.dmxapi.com`
+
+
 - 请求方式: GET
 
 - 请求地址: `API对应请求接口/{task_id}`
-- 例如 `https://{api_url}/kling/v1/images/generations/{task_id}`
+- 例如针对 [生成图像 API](./generate-image.md)：
+  - `https://{api_url}/kling/v1/images/generations/{task_id}`
+
 
 ## 请求参数
 
@@ -48,4 +57,4 @@ updatedAt: 2025-05-08
 > 深色背景为可以修改的参数，非必选参数已经注释，可以按照自己的需求启用。
 
 
-<<< @/zh/snippets/query-api.py{5-6,21-25,44}
+<<< @/zh/snippets/query-api.py{5-6,22,42}

docs/zh/snippets/query-api.py

@@ -18,11 +18,8 @@ def query_kling_image_url(task_id):
         成功时: 返回生成图像的URL地址
         任务未完成或失败: 返回None
     """
-    action = "images"
-    action2 = "generations"
-
     # 根据请求接口，构建完整的查询路径，包含task_id参数
-    query_path = f"/kling/v1/{action}/{action2}/{task_id}"
+    query_path = f"/kling/v1/images/generations/{task_id}"
     # 构建请求头，包含鉴权消息
     headers = {
     'Authorization': f'Bearer {DMX_API_TOKEN}'
@@ -35,7 +32,8 @@ def query_kling_image_url(task_id):
     # print(json_data)
     # 检查任务状态，如果成功则返回第一张图像的URL，否则返回None
     if json_data['data']['task_status'] == "succeed":
-        return json_data['data']['task_result']['images'][0]['url']
+        image_urls = [image['url'] for image in json_data['data']['task_result']['images']]
+        return image_urls
     else: 
         return None