MeUtils 2024.12.13.15.39.7__py3-none-any.whl → 2024.12.19.9.30.53__py3-none-any.whl

meutils/apis/jimeng/files.py

@@ -0,0 +1,180 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : kaiyuan
+# @Time         : 2024/12/18 16:03
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+import hmac
+import zlib
+import hashlib
+
+from meutils.pipe import *
+from meutils.caches.redis_cache import cache
+
+
+def random_str(n):
+    return ''.join(random.sample('zyxwvutsrqponmlkjihgfedcba0123456789', n))
+
+
+def hash256(msg):
+    return hashlib.sha256(msg.encode('utf-8')).hexdigest()
+
+
+def hmac_hash256(key, msg):
+    if type(key) == str:
+        return hmac.new(key.encode('utf-8'), msg.encode('utf-8'), hashlib.sha256)
+    elif type(key) == hmac.HMAC:
+        return hmac.new(key.digest(), msg.encode('utf-8'), hashlib.sha256)
+
+
+def fileCRC32(file_buffer):
+    return hex(zlib.crc32(file_buffer) & 0xFFFFFFFF)[2:]
+
+
+def u(params):
+    new_params = sorted(params.items(), key=lambda x: x[0])
+    new_params = [f"{k}={v}" for k, v in new_params]
+    return "&".join(new_params)
+
+
+class JJRequest:
+    def __init__(self, e, t, api, method="GET", params=None, data=None):
+        self.t = t
+        self.e = e
+        self.api = api
+        self.method = method
+        self.params = params
+        self.data = data
+
+    def getAuthorization(self):
+        return f"AWS4-HMAC-SHA256 Credential={self.e['access_key_id']}/{self.t[0:8]}/cn-north-1/imagex/aws4_request, SignedHeaders=x-amz-date;x-amz-security-token, Signature={self.signature()}"
+
+    def signature(self):
+        r = self.getSigningKey()
+        return hmac_hash256(r, self.stringToSign()).hexdigest()
+
+    def getSigningKey(self, r="cn-north-1", n="imagex"):
+        o = hmac_hash256("AWS4" + self.e['secret_access_key'], str(self.t[0:8]))
+        i = hmac_hash256(o, str(r))
+        s = hmac_hash256(i, str(n))
+        return hmac_hash256(s, "aws4_request")
+
+    def stringToSign(self):
+        t = []
+        t.append("AWS4-HMAC-SHA256")
+        t.append(self.t)
+        t.append(self.credentialString())
+        t.append(hash256(self.canonicalString()))
+        return "\n".join(t)
+
+    def credentialString(self, region="cn-north-1", serviceName="imagex"):
+        return "/".join([self.t[0:8], region, serviceName, "aws4_request"])
+
+    def canonicalString(self):
+        e = []
+        e.append(self.method)
+        e.append("/")
+        e.append(u(self.params))
+        e.append(self.canonicalHeaders())
+        e.append(self.signedHeaders())
+        e.append(self.hexEncodedBodyHash())
+        return "\n".join(e)
+
+    def canonicalHeaders(self):
+        return f"x-amz-date:{self.t}\nx-amz-security-token:{self.e['session_token']}\n"
+
+    def signedHeaders(self):
+        return "x-amz-date;x-amz-security-token"
+
+    def hexEncodedBodyHash(self):
+        return hash256("")
+
+
+@cache(ttl=15 * 60)
+def upload(image: bytes, upload_token: dict):  # oss 跨账号不知道是否可以使用
+    # e = auth = upload_token['data']['auth'] # 豆包
+    data = upload_token['data']
+
+    service_id = data.get('service_id', '3jr8j4ixpe')  # 即梦 3jr8j4ixpe
+
+    session_token = data['session_token']
+
+    t = time.strftime("%Y%m%dT%H%M%SZ", time.gmtime())
+
+    params = {
+        "Action": "ApplyImageUpload",
+        "Version": "2018-08-01",
+        "ServiceId": service_id,
+        "s": random_str(10),
+
+        "FileExtension": ".png",  #####
+        "FileSize": len(image),
+    }
+
+    r = JJRequest(data, t, "https://imagex.bytedanceapi.com/", method="GET", params=params)
+    headers = {
+        'authorization': r.getAuthorization(),
+        'x-amz-date': t,
+        'x-amz-security-token': session_token,
+    }
+    # logger.debug(headers)
+    response = requests.get(r.api, params=params, headers=headers)
+    response.raise_for_status()
+    logger.debug(response.status_code)
+    response = response.json()
+
+    logger.debug(bjson(response))
+    if "Result" not in response:
+        return
+
+    store_info = response['Result']['UploadAddress']['StoreInfos'][0]
+    logger.debug(bjson(store_info))
+
+    oss_uri = store_info['StoreUri']
+    oss_token = store_info['Auth']
+
+    headers = {
+        "authorization": oss_token,
+        "content-length": str(len(image)),
+        "content-Type": "image/jpeg",
+        "content-crc32": fileCRC32(image),
+    }
+
+    # oss_url = f"https://{resp['Result']['UploadAddress']['UploadHosts'][0]}/{oss_uri}"
+
+    # image_resp = requests.put(  # post
+    #     oss_url,
+    #     headers=headers,
+    #     data=image,
+    # )
+    # logger.debug(image_resp.json())
+
+    # return get_url(StoreUri)
+
+    # upload_url = f"https://tos-hl-x.snssdk.com/upload/v1/{oss_uri}"
+    upload_url = f"https://{response['Result']['UploadAddress']['UploadHosts'][0]}/upload/v1/{oss_uri}"
+
+    response = requests.post(upload_url, headers=headers, data=image)
+    response.raise_for_status()
+    response = response.json()
+
+    logger.debug(response)
+
+    return oss_uri
+
+
+if __name__ == "__main__":
+    from meutils.apis.jimeng.common import get_upload_token
+
+    token = "693701c43e477b7c405cc7e2fef0ddbd"
+
+    upload_token = arun(get_upload_token(token))
+
+    with open("11.jpg", "rb") as f:
+        file = image = f.read()
+
+        upload(image, upload_token)

meutils/apis/jimeng/images.py

@@ -0,0 +1,105 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : images
+# @Time         : 2024/12/16 17:46
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+
+from meutils.pipe import *
+
+from meutils.schemas.jimeng_types import BASE_URL, MODELS_MAP, FEISHU_URL
+from meutils.schemas.image_types import ImageRequest
+from meutils.schemas.task_types import TaskResponse
+from meutils.apis.jimeng.common import create_draft_content, get_headers, check_token
+from meutils.config_utils.lark_utils import get_next_token_for_polling
+
+from fake_useragent import UserAgent
+
+ua = UserAgent()
+
+
+async def create_task(request: ImageRequest, token: Optional[str] = None):
+    token = token or await get_next_token_for_polling(FEISHU_URL, check_token)
+
+    url = "/mweb/v1/aigc_draft/generate"
+
+    headers = get_headers(url, token)
+
+    if "http" in request.prompt:  # 图生
+        request.model = "high_aes_general_v20_L:general_v2.0_L"
+
+    draft_content = await create_draft_content(request, token)
+    payload = {
+        "extend": {
+            "root_model": request.model,
+            "template_id": ""
+        },
+        "submit_id": str(uuid.uuid4()),
+        "metrics_extra": "{\"templateId\":\"\",\"generateCount\":1,\"promptSource\":\"custom\",\"templateSource\":\"\",\"lastRequestId\":\"\",\"originRequestId\":\"\"}",
+        "draft_content": json.dumps(draft_content),
+        "http_common_info": {
+            "aid": 513695
+        }
+    }
+
+    logger.debug(bjson(payload))
+
+    async with httpx.AsyncClient(base_url=BASE_URL, headers=headers, timeout=60) as client:
+        response = await client.post(url, json=payload)
+        response.raise_for_status()
+        data = response.json()
+        logger.debug(bjson(data))
+
+    task_id = data.get("data", {}).get("aigc_data", {}).get("history_record_id")
+    return TaskResponse(task_id=task_id, system_fingerprint=token)
+
+
+async def get_task(task_id, token):
+    url = "/mweb/v1/get_history_by_ids"
+    headers = get_headers(url, token)
+    payload = {
+        "history_ids": [
+            task_id
+        ]
+    }
+
+    async with httpx.AsyncClient(base_url=BASE_URL, headers=headers, timeout=60) as client:
+        response = await client.post(url, json=payload)
+        response.raise_for_status()
+        data = response.json()
+        # {'ret': '1015', 'errmsg': 'login error', 'systime': '1734524280', 'logid': '20241218201800AC3267447B287E9E6C46', 'data': None}
+        item_list = (data.get("data") or {}).get(task_id, {}).get("item_list")
+        image_data = map(lambda x: x.get("image", {}).get("large_images"), item_list)
+
+        task_data = sum(image_data, []) | xmap_(lambda x: {"url": x.get("image_url")})
+
+        return TaskResponse(
+            task_id=task_id,
+            data=task_data,
+            message=data.get("errmsg"),
+            status=task_data and "success",
+        )
+
+
+if __name__ == '__main__':
+    token = "693701c43e477b7c405cc7e2fef0ddbd"
+
+    request = ImageRequest(prompt="一个漂亮的姑娘", size="1024x1024")
+    # request = ImageRequest(prompt="https://oss.ffire.cc/files/kling_watermark.png 让她带上墨镜", size="1024x1024")
+
+    arun(create_task(request))
+
+    # task_id = "10040025470722"
+
+    # task_id = "10053536381698"
+
+    # task_id = "10079694738434"
+
+    # task_id = "10080831230210"  # 图片编辑
+
+    # task_id = "10082971040514"
+    #
+    # arun(get_task(task_id, token))

meutils/apis/jimeng/sig.py

@@ -0,0 +1,92 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : sig
+# @Time         : 2024/12/18 15:36
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+import hashlib
+import hmac
+import datetime
+
+
+def sign(key, msg):
+    return hmac.new(key, msg.encode('utf-8'), hashlib.sha256).digest()
+
+
+def getSignatureKey(key, dateStamp, regionName, serviceName):
+    kDate = sign(('AWS4' + key).encode('utf-8'), dateStamp)
+    kRegion = sign(kDate, regionName)
+    kService = sign(kRegion, serviceName)
+    kSigning = sign(kService, 'aws4_request')
+    return kSigning
+
+
+if __name__ == '__main__':
+    pass
+    from meutils.pipe import *
+    from meutils.apis.jimeng.utils import get_upload_token
+
+    # data = arun(get_upload_token())
+
+    # access_key_id = data['data']['auth']['access_key_id']
+    # secret_access_key = data['data']['auth']['secret_access_key']
+    # session_token = data['data']['auth']['session_token']
+
+    import hashlib
+    import hmac
+    import datetime
+
+
+    def sign(key, msg):
+        """计算 HMAC-SHA256"""
+        return hmac.new(key, msg.encode('utf-8'), hashlib.sha256).digest()
+
+
+    def getSignatureKey(key, dateStamp, regionName, serviceName):
+        """生成签名密钥"""
+        kDate = sign(('AWS4' + key).encode('utf-8'), dateStamp)
+        kRegion = sign(kDate, regionName)
+        kService = sign(kRegion, serviceName)
+        kSigning = sign(kService, 'aws4_request')
+        return kSigning
+
+
+    def calculate_signature(string_to_sign, secret_key, region, service, date):
+        """计算 AWS4-HMAC-SHA256 签名"""
+        # 生成签名密钥
+        signing_key = getSignatureKey(
+            secret_key,
+            date.strftime('%Y%m%d'),
+            region,
+            service
+        )
+
+        # 计算最终签名
+        signature = hmac.new(
+            signing_key,
+            string_to_sign.encode('utf-8'),
+            hashlib.sha256
+        ).hexdigest()
+
+        return signature
+
+        # 示例参数
+
+
+    secret_key = 'AKTPN2VkZWY2MWNhNjVlNGZhZThiZmVmNzEwODA3YjM1YTY'
+    region = 'cn-north-1'
+    service = 'imagex'
+    date = datetime.datetime.utcnow()
+    print(date)
+    hash_of_canonical_request = {
+        "SessionKey": "eyJhY2NvdW50VHlwZSI6IkltYWdlWCIsImFwcElkIjoiIiwiYml6VHlwZSI6IiIsImZpbGVUeXBlIjoiaW1hZ2UiLCJsZWdhbCI6IiIsInN0b3JlSW5mb3MiOiJbe1wiU3RvcmVVcmlcIjpcInRvcy1jbi1pLWE5cm5zMnJsOTgvYmQzZGNlZWIzN2Y3NDA5NjlkMTBiZjBiZjhhMzk0NmUuanBlZ1wiLFwiQXV0aFwiOlwiU3BhY2VLZXkvYTlybnMycmw5OC8xLzp2ZXJzaW9uOnYyOmV5SmhiR2NpT2lKSVV6STFOaUlzSW5SNWNDSTZJa3BYVkNKOS5leUpsZUhBaU9qRTNNelExTXpBd016Z3NJbk5wWjI1aGRIVnlaVWx1Wm04aU9uc2lZV05qWlhOelMyVjVJam9pWm1GclpWOWhZMk5sYzNOZmEyVjVJaXdpWW5WamEyVjBJam9pZEc5ekxXTnVMV2t0WVRseWJuTXljbXc1T0NJc0ltVjRjR2x5WlNJNk1UY3pORFV6TURBek9Dd2labWxzWlVsdVptOXpJanBiZXlKdmFXUkxaWGtpT2lKaVpETmtZMlZsWWpNM1pqYzBNRGsyT1dReE1HSm1NR0ptT0dFek9UUTJaUzVxY0dWbklpd2labWxzWlZSNWNHVWlPaUl4SW4xZExDSmxlSFJ5WVNJNmV5SmliRzlqYTE5dGIyUmxJam9pSWl3aVkyOXVkR1Z1ZEY5MGVYQmxYMkpzYjJOcklqb2llMXdpYldsdFpWOXdZM1JjSWpvd0xGd2liVzlrWlZ3aU9qQXNYQ0p0YVcxbFgyeHBjM1JjSWpwdWRXeHNMRndpWTI5dVpteHBZM1JmWW14dlkydGNJanBtWVd4elpYMGlMQ0psYm1OeWVYQjBYMkZzWjI4aU9pSWlMQ0psYm1OeWVYQjBYMnRsZVNJNklpSXNJbVY0ZEY5amIyNTBaVzUwWDNSNWNHVWlPaUpwYldGblpTOXFjR1ZuSWl3aWFYTmZhVzFoWjJWNElqcDBjblZsTENKemNHRmpaU0k2SW1FNWNtNXpNbkpzT1RnaWZYMTkueVFadDJoYWFVMDBuMTFZREJXZDBCbTNIYWdtVlNUN0UzNXMwSkNnOGpzMFwiLFwiVXBsb2FkSURcIjpcIjE3ZThiODUxNzczNDRlYjFiYmM1MDAzMWU3NzQ3NGUyXCIsXCJVcGxvYWRIZWFkZXJcIjpudWxsLFwiU3RvcmFnZUhlYWRlclwiOm51bGx9XSIsInVwbG9hZEhvc3QiOiJ0b3MtZC14LWxmLnNuc3Nkay5jb20iLCJ1cmkiOiJ0b3MtY24taS1hOXJuczJybDk4L2JkM2RjZWViMzdmNzQwOTY5ZDEwYmYwYmY4YTM5NDZlLmpwZWciLCJ1c2VySWQiOiIifQ=="
+    }
+    hash_of_canonical_request = json.dumps(hash_of_canonical_request)
+    string_to_sign = f'AWS4-HMAC-SHA256\n20241218T075358Z\n20241218/cn-north-1/imagex/aws4_request\n{hash_of_canonical_request}'
+
+    # 计算签名
+    signature = calculate_signature(string_to_sign, secret_key, region, service, date)
+    print(f'签名结果: {signature}')

meutils/apis/jimeng/utils.py

@@ -0,0 +1,177 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : utils
+# @Time         : 2024/12/18 11:03
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+import httpx
+
+from meutils.pipe import *
+import random
+import base64
+import time
+import uuid
+import secrets
+import string
+import hmac
+import hashlib
+
+# 常量定义
+MODEL_NAME = "doubao"
+DEFAULT_ASSISTANT_ID = "497858"
+VERSION_CODE = "20800"
+DEVICE_ID = random.random() * 999999999999999999 + 7000000000000000000
+WEB_ID = random.random() * 999999999999999999 + 7000000000000000000
+USER_ID = str(uuid.uuid4()).replace('-', '')
+MAX_RETRY_COUNT = 3
+RETRY_DELAY = 5000
+FILE_MAX_SIZE = 100 * 1024 * 1024
+
+# 伪装headers
+FAKE_HEADERS = {
+    "Accept": "*/*",
+    "Accept-Encoding": "gzip, deflate, br, zstd",
+    "Accept-language": "zh-CN,zh;q=0.9,en-US;q=0.8,en;q=0.7",
+    "Cache-control": "no-cache",
+    "Last-event-id": "undefined",
+    "Origin": "https://www.doubao.com",
+    "Pragma": "no-cache",
+    "Priority": "u=1, i",
+    "Referer": "https://www.doubao.com",
+    "Sec-Ch-Ua": '"Google Chrome";v="131", "Chromium";v="131", "Not_A Brand";v="24"',
+    "Sec-Ch-Ua-Mobile": "?0",
+    "Sec-Ch-Ua-Platform": '"Windows"',
+    "Sec-Fetch-Dest": "empty",
+    "Sec-Fetch-Mode": "cors",
+    "Sec-Fetch-Site": "same-origin",
+    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36"
+}
+
+
+async def acquire_token(refresh_token: str) -> str:
+    """
+    获取缓存中的access_token
+    目前doubao的access_token是固定的，暂无刷新功能
+
+    Args:
+        refresh_token: 用于刷新access_token的refresh_token
+
+    Returns:
+        str: access_token
+    """
+    return refresh_token
+
+
+def generate_fake_ms_token() -> str:
+    """
+    生成伪msToken
+    """
+    # 生成96字节的随机数据
+    random_bytes = secrets.token_bytes(96)
+    # 转换为base64，并替换特殊字符
+    token = base64.b64encode(random_bytes).decode('utf-8')
+    return token.replace('+', '-').replace('/', '_').rstrip('=')
+
+
+def generate_random_string(length: int) -> str:
+    """
+    生成指定长度的随机字符串
+    """
+    chars = string.ascii_letters + string.digits
+    return ''.join(random.choice(chars) for _ in range(length))
+
+
+def generate_fake_a_bogus() -> str:
+    """
+    生成伪a_bogus
+    """
+    return f"mf-{generate_random_string(34)}-{generate_random_string(6)}"
+
+
+def generate_cookie(refresh_token: str, ms_token: Optional[str] = None) -> str:
+    """
+    生成cookie
+    """
+    ms_token = ms_token or generate_fake_ms_token()
+
+    current_timestamp = int(time.time())
+    cookie_parts = [
+        f"is_staff_user=false",
+        f"store-region=cn-gd",
+        f"store-region-src=uid",
+        f"sid_guard={refresh_token}%7C{current_timestamp}%7C5184000%7CSun%2C+02-Feb-2025+04%3A17%3A20+GMT",
+        f"uid_tt={USER_ID}",
+        f"uid_tt_ss={USER_ID}",
+        f"sid_tt={refresh_token}",
+        f"sessionid={refresh_token}",
+        f"sessionid_ss={refresh_token}",
+        f"msToken={ms_token}",
+    ]
+    return "; ".join(cookie_parts)
+
+
+async def get_upload_token():  # 3600过期
+    """
+
+    {'code': 0,
+     'data': {'auth': {'access_key_id': 'AKTPYzNkMjJlNTNjMWE1NDJiN2E5MWFkOTYxMWViYzQxYTM',
+                       'current_time': '2024-12-18T11:17:22+08:00',
+                       'expired_time': '2024-12-18T12:17:22+08:00',
+                       'secret_access_key': 'HFFTkEFKf+0DVpUrYy2yMzvgnkxLMU6+qydnGUSaDmd0vSRedpIi0qmeWSVElyOU',
+                       'session_token': 'STS2eyJMVEFjY2Vzc0tleUlkIjoiQUtMVFlUZGhPR0ptWVRNNFl6ZG1OR1JoWVRoaE0yWTJPVFl5TW1SbU0yRmhNREEiLCJBY2Nlc3NLZXlJZCI6IkFLVFBZek5rTWpKbE5UTmpNV0UxTkRKaU4yRTVNV0ZrT1RZeE1XVmlZelF4WVRNIiwiU2lnbmVkU2VjcmV0QWNjZXNzS2V5IjoiTC9WZTVSMmt4N3dsY1kvS0E5alp1WVlpSlVFM0ZjdHMzQ2Q5QjJZMVE3NlRnUDVONWViMmpKQkRQMUdyUEtqeXNYNXRKVkJPdExvVjVNOGFyY24wQ2ZtdUZRRWMxMG8xMSs3UHdKdGY0LzQ9IiwiRXhwaXJlZFRpbWUiOjE3MzQ0OTU0NDIsIlBvbGljeVN0cmluZyI6IntcIlN0YXRlbWVudFwiOlt7XCJFZmZlY3RcIjpcIkFsbG93XCIsXCJBY3Rpb25cIjpbXCJJbWFnZVg6QXBwbHlJbWFnZVVwbG9hZFwiLFwiSW1hZ2VYOkNvbW1pdEltYWdlVXBsb2FkXCJdLFwiUmVzb3VyY2VcIjpbXCJ0cm46SW1hZ2VYOio6KjpTZXJ2aWNlSWQvYTlybnMycmw5OFwiXX0se1wiRWZmZWN0XCI6XCJBbGxvd1wiLFwiQWN0aW9uXCI6W1wiUFNNXCJdLFwiUmVzb3VyY2VcIjpbXCJmbG93LmFsaWNlLnJlc291cmNlX2FwaVwiXX1dfSIsIlNpZ25hdHVyZSI6ImI2MGUxNDZkZTU0Njg2NTdlYzVlZmFjZjJlOWNlOWE5YTdhY2UwNTFlZTdkYTJjZTRmNjdiYmRiM2U4MDQ3N2IifQ=='},
+              'service_id': 'a9rns2rl98',
+              'upload_host': 'imagex.bytedanceapi.com',
+              'upload_path_prefix': 'bot-chat-image'},
+     'msg': ''}
+
+    :return:
+    """
+    cookie = generate_cookie('de2215a7bb8e442774cf388f03fac84c')
+    url = "https://www.doubao.com/alice/upload/auth_token"
+
+    headers = {
+        'priority': 'u=1, i',
+        'Cookie': cookie,
+        'User-Agent': 'Apifox/1.0.0 (https://apifox.com)',
+    }
+    payload = {
+        "scene": "bot_chat",
+        "data_type": "image"
+    }
+
+    async with httpx.AsyncClient() as client:
+        response = await client.post(url, headers=headers, json=payload)
+        response.raise_for_status()
+
+        return response.json()
+
+
+def hmac_hash256(key, msg):
+    if type(key) == str:
+        return hmac.new(key.encode('utf-8'), msg.encode('utf-8'), hashlib.sha256)
+    elif type(key) == hmac.HMAC:
+        return hmac.new(key.digest(), msg.encode('utf-8'), hashlib.sha256)
+
+
+def get_signing_key(secret_access_key, r="cn-north-1", n="imagex"):
+    dt = str(datetime.datetime.now())[:10].replace('-', '')
+    o = hmac_hash256("AWS4" + secret_access_key, dt)
+    i = hmac_hash256(o, str(r))
+    s = hmac_hash256(i, str(n))
+    return hmac_hash256(s, "aws4_request")
+
+def signature(secret_access_key):
+    r = get_signing_key(secret_access_key)
+    return hmac_hash256(r, self.stringToSign()).hexdigest()
+
+# ccd4fef2cca1a114e776badad7f4b6e73f305a4dbb09e68f336759ddb6ac0025
+
+if __name__ == '__main__':
+    # generate_cookie("")
+
+    # arun(get_upload_token())
+
+    print(get_signing_key('xW9YbDhTlWsXdaN7O2g1lfcyePxf5kJyg/r2mwSZG/iuSmbvVgToO6LVCLmUjVJ3'))

meutils/apis/siliconflow/videos.py

@@ -7,13 +7,16 @@
 # @WeChat       : meutils
 # @Software     : PyCharm
 # @Description  :
+import asyncio
 
 from meutils.pipe import *
 from meutils.caches.redis_cache import cache
+from meutils.io.files_utils import to_url, to_url_fal
 from meutils.llm.check_utils import check_token_for_siliconflow
 from meutils.schemas.task_types import TaskResponse
 from meutils.schemas.siliconflow_types import BASE_URL, VideoRequest
 from meutils.config_utils.lark_utils import get_next_token_for_polling
+from meutils.apis.translator import deeplx
 
 from openai import OpenAI, AsyncOpenAI
 
@@ -21,38 +24,54 @@ FEISHU_URL_FREE = "https://xchatllm.feishu.cn/sheets/Bmjtst2f6hfMqFttbhLcdfRJnNf
 
 check_token = partial(check_token_for_siliconflow, threshold=0.01)
 
+MODELS_MAP = {
+    "hunyuan-video": "tencent/HunyuanVideo",
+    "hunyuanvideo": "tencent/HunyuanVideo",
+    "mochi-1-preview": "genmo/mochi-1-preview",
+    "ltx-video": "Lightricks/LTX-Video",
+}
+
 
 @cache(ttl=7 * 24 * 3600)
 async def create_task(request: VideoRequest, token: Optional[str] = None):
     token = token or await get_next_token_for_polling(FEISHU_URL_FREE, check_token=check_token, from_redis=True)
 
     payload = request.model_dump(exclude_none=True)
-    payload["model"] = "tencent/HunyuanVideo"
+    payload["model"] = MODELS_MAP.get(request.model, "Lightricks/LTX-Video")
+
+    if payload["model"] in {"genmo/mochi-1-preview", "Lightricks/LTX-Video"}:  # 中文不友好
+        payload['prompt'] = (
+            await deeplx.translate(deeplx.DeeplxRequest(text=request.prompt, target_lang="EN"))).get("data")
 
-    client = OpenAI(
+    client = AsyncOpenAI(
         base_url=BASE_URL,
         api_key=token
     )
 
-    response = client.post("/video/submit", cast_to=object, body=payload)
+    response = await client.post("/video/submit", body=payload, cast_to=object)
     logger.debug(response)
-    task_id = response.get('requestId')
 
+    task_id = response.get('requestId')
     return TaskResponse(task_id=task_id, system_fingerprint=token)
 
 
 async def get_task(task_id, token: str):
-    client = OpenAI(
+    client = AsyncOpenAI(
         base_url=BASE_URL,
         api_key=token
     )
     payload = {"requestId": task_id}
-    response = client.post(f"/video/status", cast_to=object, body=payload)
+    response = await client.post(f"/video/status", cast_to=object, body=payload)
     logger.debug(response)
 
+    data = response.get("results") or {}
+
+    for video in data.get("videos", []):
+        video["url"] = await to_url_fal(video.get("url"))  # 异步执行
+
     return TaskResponse(
         task_id=task_id,
-        data=response.get("results"),
+        data=data,
         status=response.get("status"),
         message=response.get("reason"),
     )
@@ -60,12 +79,22 @@ async def get_task(task_id, token: str):
 
 if __name__ == '__main__':
     token = None
+    token = "sk-raapiguffsnsxgkfiwfusjmbpcyqoxhcohhxaybflrnvpqjw"
+
+    request = VideoRequest(
+        model="Lightricks/LTX-Video",
+        prompt="这个女人笑起来",
+        image='https://oss.ffire.cc/files/kling_watermark.png'  # 1148f2e4-0a62-4208-84de-0bf2c88f740d
+    )
     # tokens = config_manager.text.split()
 
     # tokens_ = arun(check_token_for_siliconflow(tokens, threshold=0.01))
 
-    token = "sk-raapiguffsnsxgkfiwfusjmbpcyqoxhcohhxaybflrnvpqjw"
-    arun(get_task("73a15e11-310f-4464-a9af-3ab84b201fff", token))
+    # arun(create_task(request, token=token))
+
+    # arun(get_task("fa248aac-00ec-4b00-a2f8-3d6bf1cea6d3", token))
+    # arun(get_task("c716a328-438e-4612-aff2-a669034499cb", token))
+    arun(get_task("1148f2e4-0a62-4208-84de-0bf2c88f740d", token))
 
     # token = "sk-oeptckzkhfzeidbtsqvbrvyrfdtyaaehubfwsxjytszbgohd"
     # arun(get_task("5ea22f57-45f0-425c-9d1e-bf3dae7e1e81", token))

meutils/data/VERSION

@@ -1 +1 @@
-2024.12.13.15.39.07
+2024.12.19.09.30.53