MeUtils 2024.3.4.13.4.45__py3-none-any.whl → 2025.1.16.17.15.52__py3-none-any.whl

meutils/apis/kuaishou/klingai_video.py

@@ -0,0 +1,197 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : klingai
+# @Time         : 2024/7/9 13:23
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+import jsonpath
+
+from meutils.pipe import *
+from meutils.schemas import kling_types
+from meutils.schemas.task_types import TaskType, Task
+from meutils.schemas.kuaishou_types import BASE_URL, KlingaiVideoRequest, FEISHU_URL, FEISHU_URL_VIP
+
+from meutils.decorators.retry import retrying
+from meutils.notice.feishu import send_message as _send_message
+from meutils.config_utils.lark_utils import get_next_token_for_polling
+
+from meutils.apis.kuaishou.klingai import get_reward, get_point, check_token, download, get_task_plus
+
+from meutils.apis.proxy.ips import get_one_proxy
+
+send_message = partial(
+    _send_message,
+    title=__name__,
+    url="https://open.feishu.cn/open-apis/bot/v2/hook/dc1eda96-348e-4cb5-9c7c-2d87d584ca18"
+)
+
+
+# BASE_URL = GUOJI_BASE_URL
+
+
+# 自动延长
+# {"type":"m2v_extend_video","inputs":[{"name":"input","inputType":"URL","url":"https://h1.inkwai.com/bs2/upload-ylab-stunt/special-effect/output/HB1_PROD_ai_web_29545092/8992112608804666920/output_ffmpeg.mp4","fromWorkId":29545092}],"arguments":[{"name":"prompt","value":""},{"name":"biz","value":"klingai"},{"name":"__initialType","value":"m2v_img2video"},{"name":"__initialPrompt","value":"母亲对着镜头挥手"}]}
+# 自定义创意延长
+# {"type":"m2v_extend_video","inputs":[{"name":"input","inputType":"URL","url":"https://h2.inkwai.com/bs2/upload-ylab-stunt/special-effect/output/HB1_PROD_ai_web_29542959/396308539942414182/output_ffmpeg.mp4","fromWorkId":29542959}],"arguments":[{"name":"prompt","value":"加点字"},{"name":"biz","value":"klingai"},{"name":"__initialType","value":"m2v_txt2video"},{"name":"__initialPrompt","value":"让佛祖说话，嘴巴要动，像真人一样"}]}
+@retrying(max_retries=8, max=8, predicate=lambda r: not r)
+async def create_task(request: KlingaiVideoRequest, cookie: Optional[str] = None, vip: bool = False):
+    cookie = cookie or await get_next_token_for_polling(FEISHU_URL_VIP if vip else FEISHU_URL, check_token=check_token,
+                                                        from_redis=True)
+
+    task_type = TaskType.kling_vip if vip else TaskType.kling
+
+    headers = {
+        'Cookie': cookie,
+        'Content-Type': 'application/json;charset=UTF-8'
+    }
+    request_kwargs = {}
+    for i in range(3):
+        async with httpx.AsyncClient(base_url=BASE_URL, headers=headers, **request_kwargs) as client:
+            response = await client.post("/api/task/submit", json=request.payload)
+            if response.is_success:
+                data = response.json()  # metadata
+                send_message(bjson(data))
+
+                # 触发重试 404 429 520
+                if any(i in str(data) for i in {"页面未找到", "请求超限", "配额耗尽", "积分消费失败"}):
+                    send_message(f"{data}\n\n{cookie}")
+
+                    # 走代理
+                    request_kwargs = {
+                        "proxy": await get_one_proxy(from_redis=True),
+                        # "proxies": proxies,
+                    }
+
+                    return
+
+                try:
+                    task_ids = jsonpath.jsonpath(data, "$..task.id")  # $..task..[id,arguments]
+                    if task_ids:
+                        task_id = f"{task_type}-{task_ids[0]}"
+                        return Task(id=task_id, data=data, system_fingerprint=cookie)
+                    else:
+                        return Task(status=0, data=data)
+
+                except Exception as e:
+                    logger.error(e)
+                    send_message(f"未知错误：{e}")
+
+
+@retrying(max_retries=3, predicate=lambda r: not r)
+async def create_task_plus(request: KlingaiVideoRequest, token: Optional[str] = None, vip: bool = False):
+    token = token or await get_next_token_for_polling(FEISHU_URL_VIP if vip else FEISHU_URL, check_token=check_token,
+                                                      from_redis=True)
+
+    if not vip: request.mode = "mini"  # 针对普号
+
+    headers = {
+        'Cookie': token,
+        'Content-Type': 'application/json;charset=UTF-8'
+    }
+
+    request_kwargs = {}
+    for i in range(3):
+        async with httpx.AsyncClient(base_url=BASE_URL, headers=headers, timeout=100, **request_kwargs) as client:
+            response = await client.post("/api/task/submit", json=request.payload)
+            if response.is_success:
+                data = response.json()
+                kling_types.send_message(data)
+
+                # 触发重试 404 429 520
+                if any(i in str(data) for i in {"页面未找到", "请求超限", "配额耗尽", "积分消费失败"}):
+                    kling_types.send_message(f"{data}\n\n{token}")
+
+                    # 走代理
+                    request_kwargs = {
+                        "proxy": await get_one_proxy(from_redis=True),
+                        # "proxies": proxies,
+                    }
+
+                    return
+
+                task_id = message = task_status_msg = ''
+                code = data.get("status")
+                if data.get("data"):
+                    data = data.get("data")
+                    if task := data.get("task"):
+                        code = 0
+                        task_id = f"""kling-video-{request.mode}-{task.get("id")}"""
+                        task_status = "processing"
+                        task_status_msg = data.get("message")
+                    else:
+                        task_status = "failed"
+                        task_status_msg = data.get("message")
+
+                else:
+                    message = data.get("message")
+                    task_status = "failed"
+
+                return kling_types.TaskResponse(
+                    code=code,
+                    message=message,
+                    data=kling_types.Task(task_id=task_id, task_status=task_status, task_status_msg=task_status_msg),
+                    system_fingerprint=token,
+                )
+
+
+if __name__ == '__main__':
+    # https://xchatllm.feishu.cn/sheets/Bmjtst2f6hfMqFttbhLcdfRJnNf?sheet=v8vcZY
+
+    # cookie = "_did=web_474598569404BC09;did=web_43cec8fac6fa32077f3b12bf131b77a7310f;kuaishou.ai.portal_ph=0e15bfbb51aa3d52d065602d7e566d622ba4;kuaishou.ai.portal_st=ChVrdWFpc2hvdS5haS5wb3J0YWwuc3QSsAHW9jii6pSdK5oTimSzQslDxQ5mLAW8m2j8dKFP4uptOr8aycVOX72ydltRPJhO6QbD4fGYD2pFXD_c4gqsAZPLZluo4DeIxFAWhEplxrmSA61cb-VBtCEB2bxyM1gC-sooTaSESNkekMI5WBEq1_NL5E2x-wigAGi6jm1Chpq-bJ3oTAXe9yMLV3oY4qilfoV9M77nn3cY-r76Z2Z9G-3JICGNjTmN2OAQSf_q3EXtMRoSITGi7VYyBd022GVFnVcqtiPoIiDjv6HmF2lhkn6FgvYhjgLMGinYaZxZj84kbjL8GfxuiygFMAE;userId=415950813"
+    # token = "weblogger_did=web_9665177505014D12; _gcl_au=1.1.961392860.1725442018; did=web_d77db05423a7f37445a28355351bf2736bf0; anonymous-message-release-notice-1=true; userId=5138238; ksi18n.ai.portal_st=ChNrc2kxOG4uYWkucG9ydGFsLnN0EqABA02dJVaF0Q5XvWFZVNWLa-7fNeGjTJZ_BP3zPN4JN1Ct8LRrLlT8ZAtqXZa0X_fOBXuQFiOqfoIuTziI8bNUaIKPNjhfcCT2CUcLv9bIVdYkwlJGlzMai8-2o2MoF9yopDzeYfwphWxBh575DrMjMjnPoNp0MqSq3iDIIA9glYGxEOYRcotovl63JPPI1Vy4mSX46S1Omh1X0suF2oWKCBoSPFqCkp3UokbKTh0UPCIW9AgsIiBZRgCwD_UpSF1elCUq5jwFhe-Wr2hErFmS73Zc4GyXOSgFMAE; ksi18n.ai.portal_ph=c52b0d2188bc0e0d209a401f9cb4c4889227"
+    request = KlingaiVideoRequest(prompt="一只可爱的黑白边境牧羊犬，头伸出车窗，毛发被风吹动，微笑着伸出舌头",
+                                  duration=5)  # 27638649
+    # e = KlingaiVideoRequest.Config.json_schema_extra.get('examples')[-1]  # 尾帧
+    # request = KlingaiVideoRequest(**e)
+    #
+    print(arun(create_task(request, vip=True)))
+    # arun(get_task("31298981", cookie))
+    # arun(get_task("vip-47426519", token))
+
+    # pprint(arun(create_image(rquest)))
+
+    # request
+    # request = KlingaiVideoRequest(
+    #     prompt="一条可爱的小狗",
+    #     url="https://p2.a.kwimgs.com/bs2/upload-ylab-stunt/special-effect/output/HB1_PROD_ai_web_30135907/1706269798026373672/output_ffmpeg.mp4"
+    # )
+    # pprint(arun(create_task(request, cookie)))
+    # pprint(arun(get_task(28106800, cookie)))  # 拓展的id 28106800  可能依赖账号 跨账号失败: 单账号测试成功
+
+    # url = "http://p2.a.kwimgs.com/bs2/upload-ylab-stunt/ai_portal/1720681052/LZcEugmjm4/whqrbrlhpjcfofjfywqqp9.png"
+    # request = KlingaiVideoRequest(prompt="狗狗跳起来", url=url)  # 28110824
+    # pprint(arun(create_task(request, cookie, feishu_url="https://xchatllm.feishu.cn/sheets/Bmjtst2f6hfMqFttbhLcdfRJnNf?sheet=EXxwtQ")))
+
+    # pprint(arun(get_task(28110824, cookie)))
+
+    # url = "http://p2.a.kwimgs.com/bs2/upload-ylab-stunt/ai_portal/1720681052/LZcEugmjm4/whqrbrlhpjcfofjfywqqp9.png"
+
+    # request = KlingaiVideoRequest(prompt="狗狗跳起来", url=url)  # 28110824
+    # pprint(arun(create_task(request, vip=True)))
+
+    # pprint(arun(get_task(30974235, cookie)))
+    # pprint(arun(get_task(28377631, cookie)))
+    # pprint(arun(get_task(28383134, cookie)))
+
+    # pprint(arun(beautify_prompt()))
+
+    # 国际
+    # cookie = "did=web_b11919c67a1966b83eaef4a19fb2de266cba;ksi18n.ai.portal_ph=644033a151612d07cbdedc21513f5d2191b6;ksi18n.ai.portal_st=ChNrc2kxOG4uYWkucG9ydGFsLnN0EqABChr9Is3s_8NQtSRM3A10k93e-Yg2PRAw5SR8BswQAdCW33dIk_7cWf5EQohyx45HoVG4nYAdpPZRg02_y4vsA5AA2TCzde2-cxAMecVk_Rg_oOnQBGWqMjVachSC82Qf4xA-vpOj1KYRGv4XwQ6ZvpqLRysjpt0543UjaasSUa8sHlD6XT_nwPnOc2LGaIDdhXExgvh85OqK7-FpcvniWRoSSiD-9Kd-wI-i_qkoWz9SxkEvIiDCLGRyg7lgXBHZcPxIy0hnLZfkuyO9AMaVYFzUapkQmCgFMAE;userId=3412057;weblogger_did=web_8897873009A74F8"
+    # arun(create_task(request, cookie))
+
+    # cookie = arun(get_next_token_for_polling(feishu_url=FEISHU_URL_VIP, check_token=check_token))
+    # arun(get_task("kling@vip-41009887", cookie))
+
+    # with timer():
+    #     arun(check_token(cookie))
+    #
+    #
+    #
+    #
+    # arun(get_reward(token))  # 签到
+    # arun(get_point(token))  # 签到
+
+    # arun(get_task_plus('60909700', token=token))

meutils/apis/kuaishou/kolors.py

@@ -0,0 +1,189 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : kolors
+# @Time         : 2024/7/10 13:13
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : https://kolors.kuaishou.com/
+
+
+import jsonpath
+
+from meutils.pipe import *
+from meutils.decorators.retry import retrying
+from meutils.schemas.kuaishou_types import KolorsRequest
+from meutils.notice.feishu import send_message as _send_message
+from meutils.config_utils.lark_utils import get_next_token_for_polling
+
+BASE_URL = "https://kolors.kuaishou.com"
+OSS_URL = "https://s2-111386.kwimgs.com/bs2/mmu-kolors-public"
+# https://s2-111386.kwimgs.com/bs2/mmu-kolors-public/5f9c7b42688fe95c4b8d9ebd1dba3431.png?x-oss-process=image/resize,m_mfit,w_305
+# https://s2-111386.kwimgs.com/bs2/mmu-kolors-public/mmu-kolors-4243501072-1720596454510-1a862f8dbde042349b97445fc45ec231 # upload
+
+FEISHU_URL = "https://xchatllm.feishu.cn/sheets/Bmjtst2f6hfMqFttbhLcdfRJnNf?sheet=I9NCss"
+
+send_message = partial(
+    _send_message,
+    title=__name__,
+    url="https://open.feishu.cn/open-apis/bot/v2/hook/dc1eda96-348e-4cb5-9c7c-2d87d584ca18"
+)
+
+
+@retrying(predicate=lambda r: not isinstance(r, str))  # 触发重试
+async def upload(file_or_url: Union[bytes, str], cookie: Optional[str] = None):  # todo: 垫图
+    cookie = cookie or await get_next_token_for_polling(FEISHU_URL)
+
+    headers = {
+        'Cookie': cookie,
+    }
+
+    files = params = None
+    if isinstance(file_or_url, bytes):
+        files = [('file', file_or_url)]
+
+    else:
+        params = {
+            'url': file_or_url
+        }
+
+    # files = [
+    #     ('file', ('v的副本.jpg', open('/Users/betterme/Downloads/v的副本.jpg', 'rb'), 'image/jpeg'))
+    # ]
+
+    async with httpx.AsyncClient(base_url=BASE_URL, headers=headers, timeout=100) as client:
+        response = await client.post("/api/kolors/uploadToBlobStore", params=params, files=files)
+
+        logger.debug(response.text)
+
+        if response.is_success:
+            data = response.json()
+            send_message(data)
+
+            try:
+                rawKeys = jsonpath.jsonpath(data, "$..rawKey")
+                if rawKeys:
+                    url = f"{OSS_URL}/{rawKeys[0]}?x-oss-process=image/resize,m_mfit,w_1024"
+                    return url
+
+            except Exception as e:  # 429
+                logger.error(e)
+
+            else:
+                return data
+
+
+@retrying(max_retries=5, predicate=lambda r: not r)
+async def create_task(request: KolorsRequest, cookie: Optional[str] = None):
+    cookie = cookie or await get_next_token_for_polling(FEISHU_URL)
+
+    headers = {
+        'Cookie': cookie,
+        'Content-Type': 'application/json;charset=UTF-8'
+    }
+
+    async with httpx.AsyncClient(base_url=BASE_URL, headers=headers, timeout=60) as client:
+        response = await client.post("/api/kolors/textToImage", json=request.model_dump())
+
+        logger.debug(response.text)
+        logger.info(response.status_code)
+
+        if response.is_success:
+            data = response.json()
+            send_message(data)
+
+            if data.get("code") == 500:
+                return data  # 不重试
+                # raise Exception(data)
+
+            if any(i in str(data) for i in {"页面未找到", "请求超限"}):
+                send_message(f"{data}\n\n{cookie}")
+                return  # 404 429 重试
+
+            # {"code":500,"result":{"status":"ERROR","code":33001,"reason":""}} 可能是内容审核的问题
+            try:
+                requestIds = jsonpath.jsonpath(data, "$..requestId")
+                if requestIds:
+                    logger.info(f"requestId or taskId: {requestIds[0]}")
+                    response = await client.get("/api/kolors/queryAsyncTaskResult", params={"requestId": requestIds[0]})
+                    logger.debug(response.text)
+
+                    if response.is_success:
+                        return requestIds[0]
+                    else:
+                        response.raise_for_status()
+
+            except Exception as e:  # 429
+                logger.error(e)
+
+            else:
+                return data
+
+
+@retrying(max_retries=16, exp_base=1.1, min=2, predicate=lambda r: r == "RETRYING")  # 触发重试
+async def get_task(task_id, cookie: str):
+    headers = {
+        'Cookie': cookie,
+        'Content-Type': 'application/json;charset=UTF-8'
+    }
+
+    async with httpx.AsyncClient(base_url=BASE_URL, headers=headers, timeout=60) as client:
+        response = await client.get(f"/api/kolors/requests/{task_id}", params={"withWatermark": False})
+        if response.is_success:
+            data = response.json()
+
+            logger.debug(data)
+
+            if not task_id or "ERROR" in str(data): return "TASK_FAILED"  # 跳出条件
+
+            urls = jsonpath.jsonpath(data, '$..imageBlobKey')
+            if urls and all(urls):
+                images = [{"url": f"{OSS_URL}/{url.replace('mmu:kolors-public:', '')}"} for url in urls]
+                return images
+            else:
+                return "RETRYING"  # 重试
+
+
+@retrying(max_retries=3, predicate=lambda r: r == "TASK_FAILED")
+async def create_image(request: KolorsRequest, file: Optional[bytes] = None):
+    cookie = await get_next_token_for_polling(FEISHU_URL)
+
+    if file:
+        request.referImage = await upload(file, cookie)
+
+    logger.info(request)
+
+    task_id = await create_task(request, cookie)
+    logger.info(f"taskId: {task_id}")
+    if isinstance(task_id, dict):
+        return task_id
+
+    data = await get_task(task_id, cookie)
+
+    return data
+
+
+if __name__ == '__main__':
+    cookie = "kuaishou.dmo.ketu_ph=f711fb055f50eeb21fdb39bd6e05cc611744;kuaishou.dmo.ketu_st=ChRrdWFpc2hvdS5kbW8ua2V0dS5zdBKgAULU9RPZzOXOMUukfVPm51Ghgxee3d2ZFqxfGxQ4WRIWWbZ8omUKzDni1d_3R0s3hv9RfvRU7g0dW9v1lKXIMX23-ctLJ10YovTsS6HDb_A6zzruMKa6qLFOiRWf4RvJ0DB07y7UbxuNlof78rWMMKftZEKSZzDVSSQnGhEf0-V-4WWRcEzOaAImqyJZccV3lqcMr6UHfqnJmaJG3ufNw2gaEtBzrwEitKSmOEU59oMJzRgeLSIgvpsGbded8NVshgwPYn9xkkH1dsTLZct7SIl-I82yHDQoBTAB;did=web_e022fde52721456f43cb66d90a7d6f14e462;userId=742626779;weblogger_did=web_47164250171DB527"
+    request = KolorsRequest(prompt="阳光下", imageCount=4)
+    request = KolorsRequest(prompt="a cat", imageCount=4)
+
+    # print(arun(create_task(request, cookie)))
+
+    # print(arun(get_task(851793, cookie)))
+    url = "https://s2-111386.kwimgs.com/bs2/mmu-kolors-public/5f9c7b42688fe95c4b8d9ebd1dba3431.png?x-oss-process=image/resize,m_mfit,w_305"
+    # print(arun(upload(url)))
+    #
+    # request = KolorsRequest(referImage="mmu-kolors-742626779-1720600931170-5ebc098c01b14baf965f3ea9e2a60228")
+    # request = KolorsRequest(referImage="https://s2-111386.kwimgs.com/bs2/mmu-kolors-public/mmu-kolors-742626779-1720760149252-f276504646014593a73f5b36c88ccafa?x-oss-process=image/resize,m_mfit,w_1000")
+    # print(request)
+
+    # file = open("/Users/betterme/PycharmProjects/AI/x.jpg", "rb").read()
+    #
+    # pprint(arun(upload(file_or_url=file, cookie=cookie)))
+
+    # url = "https://s2-111386.kwimgs.com/bs2/mmu-kolors-public/a1bf746085fa4bf39b0b299e91d267fd.png"
+
+    # pprint(arun(create_image(request, file)))
+    arun(create_image(request))

meutils/apis/llm_qa.py

@@ -0,0 +1,55 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : llm_qa
+# @Time         : 2024/4/24 18:11
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+from meutils.pipe import *
+
+from langchain.chat_models import ChatOpenAI
+from langchain.output_parsers import OutputFixingParser
+from langchain.output_parsers import PydanticOutputParser
+
+
+class ExpressRequest(BaseModel):
+    express_no: str = Field(description="快递单号")
+    phone_number: str = Field(description="手机号")
+
+    def __init__(self, /, **kwargs):
+        super().__init__(**kwargs)
+        self.express_no = f"{self.express_no}:{self.phone_number[-4:]}"
+
+
+parser = PydanticOutputParser(pydantic_object=ExpressRequest)
+
+print(parser.get_format_instructions())
+
+
+# parser = OutputFixingParser.from_llm(
+#     parser=parser,
+#     # llm=ChatOpenAI(temperature=0),
+#     llm=ChatOpenAI(model="kimi"),
+# )
+#
+text = """
+快递号
+SF1442153365663
+手机号是
+121215651
+"""
+#
+# result = parser.parse(text)  # 错误被自动修正
+# print(text)
+# print(result)
+
+
+print(parser.parse("""{'快递号': SF1442153365663}"""))
+
+
+import openai
+
+openai.OpenAI

meutils/apis/luma/__init__.py

@@ -0,0 +1,11 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : __init__.py
+# @Time         : 2024/7/22 10:44
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+from meutils.pipe import *

meutils/apis/luma/luma.py

@@ -0,0 +1,123 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : luma
+# @Time         : 2024/7/22 11:12
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+
+from meutils.pipe import *
+from meutils.decorators.retry import retrying
+from meutils.schemas.luma_types import BASE_URL, LumaRequest, EXAMPLES
+from meutils.schemas.task_types import Task
+
+from meutils.notice.feishu import send_message as _send_message
+from meutils.config_utils.lark_utils import get_next_token_for_polling
+
+FEISHU_URL = "https://xchatllm.feishu.cn/sheets/Bmjtst2f6hfMqFttbhLcdfRJnNf?sheet=RsoADK"
+
+send_message = partial(
+    _send_message,
+    title=__name__,
+    url="https://open.feishu.cn/open-apis/bot/v2/hook/dc1eda96-348e-4cb5-9c7c-2d87d584ca18"
+)
+
+
+def refresh_token():
+    import requests
+
+    url = "https://lumalabs.ai/?_rsc=1ovtp"
+
+    payload = {}
+    headers = {
+        'pragma': 'no-cache',
+        'priority': 'u=1, i',
+        'Cookie': '_clck=1pg7ir3%7C2%7Cfno%7C0%7C1664; _ga=GA1.1.708578083.1721630345; luma_session=SkMABQUEQBc6TA4PUVoaD0ERUEsvDFFxUFl6Cyl4Zk14UC8PLxJ6CjcNAid9AnFeEWtjcS9dHVNKek8GNHhaeF8SCAI-L18HCQFQAGNYU3wLXFxrMQdKe2R2WDsZaEduZQUKKA9RAykycgwoYFxQbyVQQXwyMll7d2JYLydaSm4DNwwvDxZaPVdAET5jAQhtU19eew8URlJkSEYEMEZabWkJFi8MDksqH3EZKl55QXgLWE9UNEpJBVVUUS5XdV5RegsSAlNWWA5UUVQJRkMVAjd3dnhUVQFsA15xDSUBVkJaQ01GRBNWAhddEgxrQFdeBF0XCEVGVk95WFclAFh8XnsoNBwvUH0NLEsoCmYBW3YoBXxZFTxldXkJGwcae0lTZigIKQgSWgA9dg0HWA0JUTZfXnsPC1pvZ1NMLzR3Xm5LOBU_MgVYKgwIUSljflV5NVtdaCEHR3hkZl8vJ2NeenUKGD9UN14tDE8IPQZMSG82BgVqVwhYf1lAQAY0SUBRYhYIPD4JRC0PVxkqTn1BeAt-BH8PD0lQYh5wDjNXR050BVFfAgJ3MA5sJAZCZ0tWFmV9XQkqC0JRRGM4FWFabn47CFYzQx9ER1sNDVFaTGoIVxcIRQpGWl9N; _clsk=k5j7yi%7C1721631686971%7C7%7C0%7Cw.clarity.ms%2Fcollect; _ga_67JX7C10DX=GS1.1.1721630344.1.1.1721631758.0.0.0',
+        'User-Agent': 'Apifox/1.0.0 (https://apifox.com)'
+    }
+
+    response = requests.request("GET", url, headers=headers, data=payload)
+    #
+    # print(response.cookies)
+    # print(response.cookies.values())
+    # print(response.headers)
+    #
+    # print(response.text)
+
+
+@retrying(max_retries=6, predicate=lambda r: not r)
+async def create_task(request: Union[dict, LumaRequest], token: Optional[str] = None):
+    token = token or await get_next_token_for_polling(FEISHU_URL)
+    logger.debug(token)
+
+    headers = {
+        'Cookie': token,
+    }
+
+    payload = request if isinstance(request, dict) else request.model_dump(exclude_none=True)
+
+    async with httpx.AsyncClient(base_url=BASE_URL, headers=headers, timeout=30) as client:
+        response = await client.post("/generations/", json=payload)
+        # [
+        #     {
+        #         "id": "82507194-91c0-4fe3-ad12-f459acbcf3a5",
+        #         "prompt": "清凉夏季美少女，微卷短发，运动服，林间石板路，斑驳光影，超级真实，16K",
+        #         "state": "pending",
+        #         "created_at": "2024-07-22T05:26:20.871974Z",
+        #         "video": null,
+        #         "liked": null,
+        #         "estimate_wait_seconds": null,
+        #         "thumbnail": null,
+        #         "last_frame": null
+        #     }
+        # ]
+
+        logger.debug(response.text)
+        logger.debug(response.status_code)
+
+        if response.is_success:
+            data = response.json()
+            send_message(bjson(data))
+            data = data[0]
+
+            return Task(id=data.get("id"), data=data, system_fingerprint=token)
+
+        response.raise_for_status()
+
+
+# @retrying(predicate=lambda r: not r)  # 触发重试
+# @retrying()  # 触发重试
+async def get_task(task_id: str, token: str):  # 1ee02f7b-42df-4926-b034-f56e4c4e2d31
+
+    headers = {
+        'Cookie': token,
+    }
+    params = {
+        'offset': 0,
+        'limit': 32
+    }
+    async with httpx.AsyncClient(base_url=BASE_URL, headers=headers, timeout=30) as client:
+        # response = await client.get(f"/user/generations/", params=params)
+        # if response.is_success:
+        #     data = response.json()
+        #     return list(filter(lambda x: x.get("id") == task_id, data))
+        response = await client.get(f"/generations/{task_id}", params=params)
+
+        logger.debug(response.text)
+        logger.debug(response.status_code)
+        if response.is_success:
+            return response.json()
+
+        response.raise_for_status()
+
+
+if __name__ == '__main__':
+    pass
+    # cookie = "_clck=1tbe3d1%7C2%7Cfno%7C0%7C1664; _ga=GA1.1.1897085930.1721613998; luma_session=SkMABQUEQBc6TA4PUVoaD0ERUEsvDFFxUFl6Cyl4Zk14UC8PLxJ6CjcNAid9AnFeEWtjcS9dHVNKek8GNHhaeF8SCAI-L18HCQFQAGNYU3wLXFxrMQdKe2R2WDsZaEduZQUKKA9RAykycgwoYFxQbyVQQXwyMll7d2JYLydaSm4DNwwvDxZaPVdAET5jAQhtU19eew8URlJkSEYEMEZabWkJFi8MDksqH3EZKl59C3glWExUNEpccnsHWS8wB3lffhIoLQRTWDYDdwpVGXJ8TS93b3QREFlGRARmASxCAmEFQ01GRBNWAhddEgxrQFdeBF0XCEVGVk95WFclAFh8XnsoNBwvUH0NLEsoCmYBW3YoBXxZFTxldXkJGwcae0lTZigIKQgSWgA9dg0HWA0JUTZfXnsPC1pvZ1NMLzR3Xm5LOBU_MgVYKgwIUSljflV5NVtdaCEHR3hkZl8vJ2NeenUKGD9UN14tDE8IPQZMSG82BgVqVwhYf1lAQAY0SUBRYhYIPD4JRC0PVxkqTn1BeAt-Qn8hD0pQYh5tVTtZYlIDDjlTKAVhNgxJJzMBXwENUB5AcC8NYH1-YQdbAAN9TWMbGAUNQx9ER1sNDVFaTGoIVxcIRQpGWl9N"
+    # cookie = "_clck=1tbe3d1%7C2%7Cfno%7C0%7C1664; _ga=GA1.1.1897085930.1721613998; luma_session=SkMABQUEQBc6TA4PUVoaD0ERUEsvDFFxUFl6Cyl4Zk14UC8PLxJ6CjcNAid9AnFeEWtjcS9dHVNKek8GNHhaeF8SCAI-L18HCQFQAGNYU3wLXFxrMQdKe2R2WDsZaEduZQUKKA9RAykycgwoYFxQbyVQQXwyMll7d2JYLydaSm4DNwwvDxZaPVdAET5jAQhtU19eew8URlJkSEYEMEZabWkJFi8MDksqH3EZKl59DHo1dkxUNEpqZVJldgA1WmZ8QxUAIycRchwHcCUmfwEJAwBnAgM1MFp_QXFjKFpHSQV8Q01GRBNWAhddEgxrQFdeBF0XCEVGVk95WFclAFh8XnsoNBwvUH0NLEsoCmYBW3YoBXxZFTxldXkJGwcae0lTZigIKQgSWgA9dg0HWA0JUTZfXnsPC1pvZ1NMLzR3Xm5LOBU_MgVYKgwIUSljflV5NVtdaCEHR3hkZl8vJ2NeenUKGD9UN14tDE8IPQZMSG82BgVqVwhYf1lAQAY0SUBRYhYIPD4JRC0PVxkqTn1BeAt-TX0xIUpQYh5TByJhAn5wODYsCTIHLhB6DFZNUWsYE3sAZB0gBA9ydVInDFZyRGYAUgIrQx9ER1sNDVFaTGoIVxcIRQpGWl9N"
+    arun(create_task(EXAMPLES[0]))
+    # arun(get_task("6de49fdc-f3ed-4590-8a1a-0c53cfca055b", cookie))
+    # print(refresh_token())

meutils/apis/minicpm/__init__.py

@@ -0,0 +1,9 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : __init__.py
+# @Time         : 2024/8/20 10:38
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :