fix: i2i t2i

Author: timerring

docs/zh/models/kling/kling-image-to-image.md

@@ -1,5 +1,5 @@
 ---
-title: 可灵 Kling 图生图
+title: 可灵 Kling 图生图案例
 gitChangelog: false
 updatedAt: 2025-05-08
 ---
@@ -24,7 +24,7 @@ updatedAt: 2025-05-08
 > 深色背景为可以修改的参数，非必选参数已经注释，可以按照自己的需求启用。
 
 
-<<< @/zh/snippets/kling-image-to-image.py{164-165,172-182}
+<<< @/zh/snippets/kling-image-to-image.py{165-166,173-183}
 
 
 ## 返回结果

docs/zh/models/kling/kling-text-to-image.md

@@ -1,5 +1,5 @@
 ---
-title: 可灵 Kling 文生图
+title: 可灵 Kling 文生图案例
 gitChangelog: false
 updatedAt: 2025-05-08
 ---
@@ -24,7 +24,7 @@ updatedAt: 2025-05-08
 > 深色背景为可以修改的参数，非必选参数已经注释，可以按照自己的需求启用。
 
 
-<<< @/zh/snippets/kling-text-to-image.py{120-121,128-135}
+<<< @/zh/snippets/kling-text-to-image.py{122-123,130-137}
 
 
 ## 返回结果
@@ -43,7 +43,7 @@ updatedAt: 2025-05-08
 
 ```mermaid
 graph TD
-    A[客户端初始化 KlingTextToImage] -->|提供 API 凭证| B[调用 generate_image 方法]
+    A[初始化 KlingTextToImage] -->|提供 API 凭证| B[调用 generate_image 方法]
     B -->|传入提示词和参数| C[_kling_generate_image 方法]
     C -->|构建请求体| D[发送 POST 请求到 API]
     D -->|返回 task_id| E[开始轮询任务状态]

docs/zh/snippets/kling-image-to-image.py

@@ -15,6 +15,7 @@ def __init__(self, api_token, api_url):
         self.api_token = api_token
         # 初始化 HTTP 连接
         self.conn = http.client.HTTPSConnection(self.api_url)
+        self.endpoint = "/kling/v1/images/generations"
         # 设置请求头
         self.headers = {
             'Authorization': f'Bearer {self.api_token}',
@@ -67,7 +68,7 @@ def _kling_generate_image(self, model_name, prompt, image, image_reference,
         }
 
         # 发送 POST 请求，提交图像生成任务
-        self.conn.request("POST", "/kling/v1/images/generations", json.dumps(payload), self.headers)
+        self.conn.request("POST", self.endpoint, json.dumps(payload), self.headers)
         # 获取响应
         res = self.conn.getresponse()
         # 读取响应内容并解析为 JSON
@@ -89,7 +90,7 @@ def _query_kling_image_url(self, task_id):
             image_url: 图像 url，任务未完成时返回 None
         """
         # 构建查询路径
-        query_path = f"/kling/v1/images/generations/{task_id}"
+        query_path = f"{self.endpoint}/{task_id}"
 
         # 发送 GET 请求，查询图像生成任务状态
         self.conn.request("GET", query_path, None, self.headers)

docs/zh/snippets/kling-text-to-image.py

@@ -14,6 +14,7 @@ def __init__(self, api_token, api_url):
         self.api_token = api_token
         # 初始化 HTTP 连接
         self.conn = http.client.HTTPSConnection(self.api_url)
+        self.endpoint = "/kling/v1/images/generations"
         # 设置请求头
         self.headers = {
             'Authorization': f'Bearer {self.api_token}',
@@ -46,14 +47,18 @@ def _kling_generate_image(self, model_name, prompt, negative_prompt, output_form
         })
 
         # 发送 POST 请求，提交图像生成任务
-        self.conn.request("POST", "/kling/v1/images/generations?=null", payload, self.headers)
+        self.conn.request("POST", self.endpoint, payload, self.headers)
         # 获取响应
         res = self.conn.getresponse()
         # 读取响应内容并解析为 JSON
         json_data = json.loads(res.read().decode("utf-8"))
         # print(json_data)
-        # 从 json 文件中返回任务 ID
-        return json_data['data']['task_id']
+        if 'code' in json_data and json_data['code'] == 0:
+            # 成功则返回提交的任务 id
+            return json_data['data']['task_id']
+        else:
+            # 失败则返回错误信息
+            raise Exception(f"API调用失败：{json_data['message']}")
 
     def _query_kling_image_url(self, task_id):
         """使用查询接口获取生成图像 url
@@ -64,10 +69,7 @@ def _query_kling_image_url(self, task_id):
             image_url: 图像 url
         """
         # 构建查询路径
-        action = "images"
-        action2 = "generations"
-
-        query_path = f"/kling/v1/{action}/{action2}/{task_id}"
+        query_path = f"{self.endpoint}/{task_id}"
 
         # 发送 GET 请求，查询图像生成任务状态
         self.conn.request("GET", query_path, None, self.headers)
@@ -125,7 +127,7 @@ def generate_image(self, model_name, prompt, negative_prompt="", output_format="
 
     # 生成图像
     image_url = kling_text_to_image.generate_image(
-        model_name="kling-v1-5", # [必选]模型名称 可选择 kling-v1-5 或 kling-v1
+        model_name="kling-v1-5", # [必选]模型名称 可选择 kling-v1-5 或 kling-v1 或 kling-v2
         prompt="生成一张袋鼠的照片，手里拿着一个写着'DMXAPI'的牌子", # [必选]文本提示词
         # negative_prompt="", # 负向文本提示词
         # output_format="png", # 输出格式：png 或 jpg