✨新增混元生图功能（未完善）

nonebot_plugin_marshoai/util_hunyuan.py

@@ -0,0 +1,31 @@
+import json
+import types
+from tencentcloud.common import credential
+from tencentcloud.common.profile.client_profile import ClientProfile
+from tencentcloud.common.profile.http_profile import HttpProfile
+from tencentcloud.common.exception.tencent_cloud_sdk_exception import TencentCloudSDKException
+from tencentcloud.hunyuan.v20230901 import hunyuan_client, models
+from .config import config
+def generate_image(prompt: str):
+    cred = credential.Credential(config.marshoai_tencent_secretid, config.marshoai_tencent_secretkey)
+    # 实例化一个http选项，可选的，没有特殊需求可以跳过
+    httpProfile = HttpProfile()
+    httpProfile.endpoint = "hunyuan.tencentcloudapi.com"
+
+    # 实例化一个client选项，可选的，没有特殊需求可以跳过
+    clientProfile = ClientProfile()
+    clientProfile.httpProfile = httpProfile
+    client = hunyuan_client.HunyuanClient(cred, "ap-guangzhou", clientProfile)
+
+    req = models.TextToImageLiteRequest()
+    params = {
+        "Prompt": prompt,
+        "RspImgType": "url",
+        "Resolution": "1080:1920"
+    }
+    req.from_json_string(json.dumps(params))
+
+    # 返回的resp是一个TextToImageLiteResponse的实例，与请求对象对应
+    resp = client.TextToImageLite(req)
+    # 输出json格式的字符串回包
+    return resp.to_json_string()