'commit'

dsLightRag/JiMeng/Async_SubmitTask.py

@@ -1,30 +0,0 @@
-from __future__ import print_function
-from JiMeng.Kit.JmCommon import JmCommon
-from volcengine.visual.VisualService import VisualService
-
-if __name__ == '__main__':
-    visual_service = VisualService()
-
-    visual_service.set_ak(JmCommon.ak)
-    visual_service.set_sk(JmCommon.sk)
-
-    # 请求Body(查看接口文档请求参数-请求示例，将请求参数内容复制到此)
-    form = {
-        "req_key": "jimeng_high_aes_general_v21_L",
-        # "prompt": "少女，光影，瘦，白皙，干净，美丽",
-        "prompt": "雨天街头，一位头戴黄色安全帽、身穿荧光黄安全背心与棕色外套、搭配蓝色牛仔裤和棕色工装靴的工人，背着一个小女孩。工人脖子上侧挂一个小熊玩偶包，拖住小女孩臀部的手同时挂着一袋小熊饼干。小女孩双手搂着工人的脖子。小女孩穿红色雨衣，戴红色雨衣帽。两个人露出微笑。背景是居民楼和树木，地面潮湿，氛围温馨。",
-        "width": 1024,
-        "height": 1024,
-        "seed": -1,
-        "ddim_steps": 20,
-        "scale": 7.0,
-        "return_url": True,
-        "logo_info": {
-            "add_logo": True,
-            "position": 2,
-            "language": 0,
-            "opacity": 1
-        }
-    }
-    resp = visual_service.cv_submit_task(form)
-    print(resp)

dsLightRag/JiMeng/Async_text2img_xl_sft.py

@@ -1,30 +0,0 @@
-# uv pip install volcengine
-from volcengine.visual.VisualService import VisualService
-from JiMeng.Kit.JmCommon import JmCommon
-
-if __name__ == '__main__':
-    visual_service = VisualService()
-
-    visual_service.set_ak(JmCommon.ak)
-    visual_service.set_sk(JmCommon.sk)
-
-    # https://www.volcengine.com/docs/6791/1330203
-    form = {
-        "req_key": "t2i_xl_sft",
-        # "prompt": "少女，光影，瘦，白皙，干净，美丽",
-        "prompt": "雨天街头，一位头戴黄色安全帽、身穿荧光黄安全背心与棕色外套、搭配蓝色牛仔裤和棕色工装靴的工人，背着一个小女孩。工人脖子上侧挂一个小熊玩偶包，拖住小女孩臀部的手同时挂着一袋小熊饼干。小女孩双手搂着工人的脖子。小女孩穿红色雨衣，戴红色雨衣帽。两个人露出微笑。背景是居民楼和树木，地面潮湿，氛围温馨。",
-        "width": 1024,
-        "height": 1024,
-        "seed": -1,
-        "ddim_steps": 20,
-        "scale": 7.0,
-        "return_url": True,
-        "logo_info": {
-            "add_logo": True,
-            "position": 2,
-            "language": 0,
-            "opacity": 1
-        }
-    }
-    resp = visual_service.text2img_xl_sft(form)
-    print(resp)

dsLightRag/JiMeng/DoubaoShiTu.py

@@ -0,0 +1,58 @@
+import os
+import json
+import requests
+
+from JiMeng.Kit.VolcanoConst import VOLCANO_API_KEY
+
+# API请求URL
+url = 'https://ark.cn-beijing.volces.com/api/v3/chat/completions'
+
+# 请求头
+headers = {
+    'Content-Type': 'application/json',
+    'Authorization': f'Bearer {VOLCANO_API_KEY}'
+}
+
+# 请求体数据
+payload = {
+    'model': 'doubao-1-5-vision-pro-32k-250115',
+    'messages': [
+        {
+            'content': [
+                {
+                    'image_url': {
+                        'url': 'https://dsideal.obs.cn-north-1.myhuaweicloud.com/HuangHai/Backup/Text2Img.jpg'
+                    },
+                    'type': 'image_url'
+                },
+                {
+                    'text': '图片主要讲了什么?',
+                    'type': 'text'
+                }
+            ],
+            'role': 'user'
+        }
+    ]
+}
+
+# 发送POST请求
+try:
+    response = requests.post(url, headers=headers, data=json.dumps(payload), timeout=30)
+
+    # 检查响应状态码
+    response.raise_for_status()
+
+    # 解析响应数据
+    result = response.json()
+    print('请求成功:')
+    print(json.dumps(result, ensure_ascii=False, indent=2))
+
+except requests.exceptions.RequestException as e:
+    print(f'请求发生错误: {e}')
+    if hasattr(e, 'response') and e.response is not None:
+        print(f'错误响应状态码: {e.response.status_code}')
+        print(f'错误响应内容: {e.response.text}')
+
+except json.JSONDecodeError:
+    print('响应解析失败: 返回的内容不是有效的JSON格式')
+    print(f'响应内容: {response.text}')

dsLightRag/JiMeng/JmImg2Video.py

@@ -42,7 +42,7 @@ def main():
     try:
         # 玩法参考: https://www.volcengine.com/docs/85621/1544774
         image_urls = [
-            "https://dsideal.obs.myhuaweicloud.com/HuangHai/%E5%A4%87%E4%BB%BD/%E5%B0%8F%E4%B9%94%E5%A4%B4%E5%83%8F.jpg"
+            "https://dsideal.obs.cn-north-1.myhuaweicloud.com/HuangHai/Backup/Text2Img.jpg"
         ]
         prompt = ""
 

dsLightRag/JiMeng/Kit/VolcanoConst.py

@@ -0,0 +1,2 @@
+# 火山API_KEY
+VOLCANO_API_KEY="f6150e6c-422a-4265-8b63-4d941b271220"