MeUtils 2024.12.23.16.36.51__tar.gz → 2024.12.31.17.51.53__tar.gz

{meutils-2024.12.23.16.36.51 → meutils-2024.12.31.17.51.53/MeUtils.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: MeUtils
-Version: 2024.12.23.16.36.51
+Version: 2024.12.31.17.51.53
 Summary: description
 Home-page: https://github.com/yuanjie-ai/MeUtils
 Author: yuanjie
@@ -108,42 +108,42 @@ Requires-Dist: pandas-profiling[notebook]; extra == "pd"
 Requires-Dist: pandas_summary; extra == "pd"
 Requires-Dist: polars; extra == "pd"
 Provides-Extra: all
-Requires-Dist: missingno; extra == "all"
-Requires-Dist: pymysql; extra == "all"
-Requires-Dist: uvicorn; extra == "all"
-Requires-Dist: reportlab; extra == "all"
-Requires-Dist: langchain; extra == "all"
-Requires-Dist: iteration_utilities; extra == "all"
-Requires-Dist: faiss-cpu; extra == "all"
+Requires-Dist: pymongo; extra == "all"
+Requires-Dist: jmespath; extra == "all"
 Requires-Dist: thriftpy2; extra == "all"
-Requires-Dist: fastapi[all]; extra == "all"
-Requires-Dist: streamlit; extra == "all"
-Requires-Dist: fastapi; extra == "all"
-Requires-Dist: pandas_summary; extra == "all"
-Requires-Dist: redis-py-cluster; extra == "all"
+Requires-Dist: simplejson; extra == "all"
 Requires-Dist: schedule; extra == "all"
 Requires-Dist: thefuck; extra == "all"
-Requires-Dist: pandas-profiling[notebook]; extra == "all"
-Requires-Dist: jmespath; extra == "all"
-Requires-Dist: pretty_errors; extra == "all"
-Requires-Dist: geopy; extra == "all"
-Requires-Dist: dataframe_image; extra == "all"
-Requires-Dist: jieba; extra == "all"
+Requires-Dist: pymilvus; extra == "all"
 Requires-Dist: filetype; extra == "all"
+Requires-Dist: pandas_summary; extra == "all"
+Requires-Dist: geopy; extra == "all"
 Requires-Dist: polars; extra == "all"
 Requires-Dist: seaborn; extra == "all"
-Requires-Dist: simplejson; extra == "all"
+Requires-Dist: cachetools; extra == "all"
+Requires-Dist: faiss-cpu; extra == "all"
+Requires-Dist: pandas-profiling[notebook]; extra == "all"
+Requires-Dist: pyarrow; extra == "all"
+Requires-Dist: uvicorn; extra == "all"
+Requires-Dist: pymysql; extra == "all"
+Requires-Dist: langchain; extra == "all"
+Requires-Dist: pretty_errors; extra == "all"
 Requires-Dist: sqlalchemy; extra == "all"
-Requires-Dist: asyncmy; extra == "all"
+Requires-Dist: missingno; extra == "all"
 Requires-Dist: gunicorn; extra == "all"
-Requires-Dist: pymongo; extra == "all"
+Requires-Dist: sse_starlette; extra == "all"
+Requires-Dist: jieba; extra == "all"
 Requires-Dist: jinja2; extra == "all"
-Requires-Dist: pyarrow; extra == "all"
+Requires-Dist: streamlit; extra == "all"
 Requires-Dist: openai; extra == "all"
-Requires-Dist: pymilvus; extra == "all"
+Requires-Dist: fastapi[all]; extra == "all"
+Requires-Dist: reportlab; extra == "all"
+Requires-Dist: redis-py-cluster; extra == "all"
+Requires-Dist: dataframe_image; extra == "all"
+Requires-Dist: asyncmy; extra == "all"
+Requires-Dist: iteration_utilities; extra == "all"
 Requires-Dist: pymupd; extra == "all"
-Requires-Dist: cachetools; extra == "all"
-Requires-Dist: sse_starlette; extra == "all"
+Requires-Dist: fastapi; extra == "all"
 
 [![Downloads](http://pepy.tech/badge/meutils)](http://pepy.tech/project/meutils)
 

{meutils-2024.12.23.16.36.51 → meutils-2024.12.31.17.51.53}/MeUtils.egg-info/SOURCES.txt

@@ -180,6 +180,7 @@ meutils/apis/oneapi/log.py
 meutils/apis/oneapi/token.py
 meutils/apis/oneapi/token_.py
 meutils/apis/oneapi/user.py
+meutils/apis/oneapi/utils.py
 meutils/apis/pixverse/__init__.py
 meutils/apis/pixverse/pixverse.py
 meutils/apis/proxy/__init__.py
@@ -302,6 +303,7 @@ meutils/decorators/decorator_demo.py
 meutils/decorators/demo.py
 meutils/decorators/fastapi_decorator.py
 meutils/decorators/feishu.py
+meutils/decorators/polling.py
 meutils/decorators/retry.py
 meutils/decorators/scheduler.py
 meutils/decorators/schedulers.py
@@ -347,8 +349,6 @@ meutils/io/image.py
 meutils/io/minio_utils.py
 meutils/io/tf_io.py
 meutils/io/x.py
-meutils/jwt_utils/__init__.py
-meutils/jwt_utils/common.py
 meutils/llm/__init__.py
 meutils/llm/check_api.py
 meutils/llm/check_utils.py
@@ -361,6 +361,7 @@ meutils/llm/completions/cp.py
 meutils/llm/completions/delilegal.py
 meutils/llm/completions/dify.py
 meutils/llm/completions/kimi.py
+meutils/llm/completions/modelscope.py
 meutils/llm/completions/oi.py
 meutils/llm/completions/rag.py
 meutils/llm/completions/tryblend.py
@@ -433,14 +434,18 @@ meutils/queues/demo.py
 meutils/queues/smooth_queue.py
 meutils/queues/uniform_queue.py
 meutils/request_utils/__init__.py
+meutils/request_utils/ark.py
 meutils/request_utils/a爬虫.py
 meutils/request_utils/crawler.py
 meutils/request_utils/download.py
 meutils/request_utils/results.py
 meutils/request_utils/tiangong.py
+meutils/request_utils/volc.py
 meutils/request_utils/wechat.py
 meutils/request_utils/公网ip.py
 meutils/request_utils/爬虫.py
+meutils/request_utils/jwt_utils/__init__.py
+meutils/request_utils/jwt_utils/common.py
 meutils/schemas/__init__.py
 meutils/schemas/baidu_api.py
 meutils/schemas/baidu_types.py

{meutils-2024.12.23.16.36.51/MeUtils.egg-info → meutils-2024.12.31.17.51.53}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: MeUtils
-Version: 2024.12.23.16.36.51
+Version: 2024.12.31.17.51.53
 Summary: description
 Home-page: https://github.com/yuanjie-ai/MeUtils
 Author: yuanjie
@@ -108,42 +108,42 @@ Requires-Dist: pandas-profiling[notebook]; extra == "pd"
 Requires-Dist: pandas_summary; extra == "pd"
 Requires-Dist: polars; extra == "pd"
 Provides-Extra: all
-Requires-Dist: missingno; extra == "all"
-Requires-Dist: pymysql; extra == "all"
-Requires-Dist: uvicorn; extra == "all"
-Requires-Dist: reportlab; extra == "all"
-Requires-Dist: langchain; extra == "all"
-Requires-Dist: iteration_utilities; extra == "all"
-Requires-Dist: faiss-cpu; extra == "all"
+Requires-Dist: pymongo; extra == "all"
+Requires-Dist: jmespath; extra == "all"
 Requires-Dist: thriftpy2; extra == "all"
-Requires-Dist: fastapi[all]; extra == "all"
-Requires-Dist: streamlit; extra == "all"
-Requires-Dist: fastapi; extra == "all"
-Requires-Dist: pandas_summary; extra == "all"
-Requires-Dist: redis-py-cluster; extra == "all"
+Requires-Dist: simplejson; extra == "all"
 Requires-Dist: schedule; extra == "all"
 Requires-Dist: thefuck; extra == "all"
-Requires-Dist: pandas-profiling[notebook]; extra == "all"
-Requires-Dist: jmespath; extra == "all"
-Requires-Dist: pretty_errors; extra == "all"
-Requires-Dist: geopy; extra == "all"
-Requires-Dist: dataframe_image; extra == "all"
-Requires-Dist: jieba; extra == "all"
+Requires-Dist: pymilvus; extra == "all"
 Requires-Dist: filetype; extra == "all"
+Requires-Dist: pandas_summary; extra == "all"
+Requires-Dist: geopy; extra == "all"
 Requires-Dist: polars; extra == "all"
 Requires-Dist: seaborn; extra == "all"
-Requires-Dist: simplejson; extra == "all"
+Requires-Dist: cachetools; extra == "all"
+Requires-Dist: faiss-cpu; extra == "all"
+Requires-Dist: pandas-profiling[notebook]; extra == "all"
+Requires-Dist: pyarrow; extra == "all"
+Requires-Dist: uvicorn; extra == "all"
+Requires-Dist: pymysql; extra == "all"
+Requires-Dist: langchain; extra == "all"
+Requires-Dist: pretty_errors; extra == "all"
 Requires-Dist: sqlalchemy; extra == "all"
-Requires-Dist: asyncmy; extra == "all"
+Requires-Dist: missingno; extra == "all"
 Requires-Dist: gunicorn; extra == "all"
-Requires-Dist: pymongo; extra == "all"
+Requires-Dist: sse_starlette; extra == "all"
+Requires-Dist: jieba; extra == "all"
 Requires-Dist: jinja2; extra == "all"
-Requires-Dist: pyarrow; extra == "all"
+Requires-Dist: streamlit; extra == "all"
 Requires-Dist: openai; extra == "all"
-Requires-Dist: pymilvus; extra == "all"
+Requires-Dist: fastapi[all]; extra == "all"
+Requires-Dist: reportlab; extra == "all"
+Requires-Dist: redis-py-cluster; extra == "all"
+Requires-Dist: dataframe_image; extra == "all"
+Requires-Dist: asyncmy; extra == "all"
+Requires-Dist: iteration_utilities; extra == "all"
 Requires-Dist: pymupd; extra == "all"
-Requires-Dist: cachetools; extra == "all"
-Requires-Dist: sse_starlette; extra == "all"
+Requires-Dist: fastapi; extra == "all"
 
 [![Downloads](http://pepy.tech/badge/meutils)](http://pepy.tech/project/meutils)
 

{meutils-2024.12.23.16.36.51 → meutils-2024.12.31.17.51.53}/meutils/_utils.py

@@ -10,6 +10,7 @@
 
 import json
 import base64
+from pydantic import BaseModel
 
 
 # bytes/string
@@ -39,8 +40,21 @@ def list_intersection(l1, l2):
 
 list4log = lambda ls: "\n\t" + "\n\t".join(ls)
 
+
 # 字典
-dict2json = bjson = lambda dic: json.dumps(dic, indent=4, ensure_ascii=False)
+
+
+def bjson(json_data):
+    if isinstance(json_data, dict):
+        return json.dumps(json_data, indent=4, ensure_ascii=False)
+    elif isinstance(json_data, BaseModel):
+        return json_data.model_dump_json(indent=4)
+
+    else:
+        return json_data
+
+
+dict2json = bjson
 json2dict = lambda s: json.loads(s.replace("'", '"'))
 
 

{meutils-2024.12.23.16.36.51 → meutils-2024.12.31.17.51.53}/meutils/apis/chatglm/glm_video_api.py

@@ -14,6 +14,8 @@ from meutils.notice.feishu import send_message as _send_message
 from meutils.config_utils.lark_utils import get_next_token_for_polling
 
 from meutils.schemas.video_types import VideoRequest
+from meutils.schemas.image_types import ImageRequest, ImagesResponse
+from meutils.str_utils.regular_expression import parse_url
 
 send_message = partial(
     _send_message,
@@ -60,6 +62,28 @@ async def get_task(task_id, api_key):
     )
     return response
 
+
+async def generate(request: ImageRequest, n: int = 30):  # 兼容dalle3
+    request.model = "cogvideox-flash"
+    # request.duration = 10 # 模型后缀
+
+    url = None
+    if urls := parse_url(request.prompt):
+        url = urls[-1]
+        request.prompt = request.prompt.replace(url, "")
+
+    request = VideoRequest(image_url=url, **request.model_dump(exclude_none=True))
+    task_response = await create_task(request)
+
+    for i in range(n):
+        await asyncio.sleep(max(1., n / (i + 1)))
+        response = await get_task(task_response.id, task_response.system_fingerprint)
+
+        logger.debug(response)
+        if response.task_status == "SUCCESS" or response.video_result:
+            return ImagesResponse(data=response.model_dump()["video_result"])
+
+
 # {
 #     "model": "cogvideox",
 #     "request_id": "8868902201637896192",
@@ -78,7 +102,15 @@ async def get_task(task_id, api_key):
 
 # VideoResult
 if __name__ == '__main__':
-
     api_key = "e21bd630f681c4d90b390cd609720483.WSFVgA3Kk1wNCX0mN"
-    r = arun(create_task(VideoRequest(), api_key=api_key))
+    api_key = "7d10426c06afa81e8d7401d97781249c.DbqlSsicRtaUdKXI"  # 新号
+
+    request = VideoRequest(
+        model='cogvideox-flash'
+    )
+    # r = arun(create_task(request, api_key=api_key))
     pass
+
+    request = ImageRequest(prompt="https://oss.ffire.cc/files/kling_watermark.png 让这个女人笑起来")
+
+    arun(generate(request, n=30))

{meutils-2024.12.23.16.36.51 → meutils-2024.12.31.17.51.53}/meutils/apis/hailuoai/videos.py

@@ -335,7 +335,7 @@ async def get_task(task_id: str, token: str):
         response = VideoResponse(
             task_id=task_id,
             base_resp=data.get('statusInfo', {}),
-            videos=data.get('data', {}).get("videos")
+            videos=(data.get('data') or {}).get("videos")
         )
         return response
 

meutils-2024.12.31.17.51.53/meutils/apis/images/recraft.py

@@ -0,0 +1,126 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : ecraft
+# @Time         : 2024/10/31 16:30
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+
+from meutils.pipe import *
+from meutils.io.files_utils import to_url
+from meutils.schemas.image_types import ImageRequest, ImagesResponse, RecraftImageRequest
+from meutils.notice.feishu import IMAGES, send_message as _send_message
+from meutils.config_utils.lark_utils import get_next_token_for_polling, aget_spreadsheet_values
+from meutils.decorators.retry import retrying
+
+BASE_URL = "https://api.recraft.ai"
+
+FEISHU_URL = "https://xchatllm.feishu.cn/sheets/GYCHsvI4qhnDPNtI4VPcdw2knEd?sheet=Lrhtf2"
+
+DEFAULT_MODEL = "recraftv3"
+MODELS = {}
+
+send_message = partial(
+    _send_message,
+    title=__name__,
+    url=IMAGES
+)
+
+
+@alru_cache(ttl=10 * 60)
+@retrying()
+async def get_access_token(token: Optional[str] = None):
+    token = token or await get_next_token_for_polling(feishu_url=FEISHU_URL, check_token=check_token)
+    headers = {"cookie": token}
+
+    async with httpx.AsyncClient(base_url="https://www.recraft.ai", headers=headers, timeout=60) as client:
+        response = await client.get("/api/auth/session")
+        response.raise_for_status()
+        logger.debug(response.json())
+        return response.json()["accessToken"]
+
+
+# @retrying()
+async def generate(request: RecraftImageRequest, token: Optional[str] = None):
+    token = await get_access_token(token)
+    headers = {"Authorization": f"Bearer {token}"}
+    # params = {"project_id": "26016b99-3ad0-413b-821b-5f884bd9454e"}  # project_id 是否是必要的
+    params = {}  # project_id 是否是必要的
+    # params = {"project_id": "47ba6825-0fde-4cea-a17e-ed7398c0a298"}
+    payload = request.model_dump(exclude_none=True)
+    logger.debug(payload)
+
+    async with httpx.AsyncClient(base_url=BASE_URL, headers=headers, timeout=60) as client:
+        response = await client.post(f"/queue_recraft/prompt_to_image", params=params, json=payload)
+        response.raise_for_status()
+        params = {
+            "operation_id": response.json()["operationId"]
+        }
+        logger.debug(params)
+
+        response = await client.get("/poll_recraft", params=params)
+        response.raise_for_status()
+        metadata = response.json()
+        logger.debug(metadata)
+
+        # {'credits': 1,
+        #  'height': 1024,
+        #  'images': [{'image_id': 'f9d8e7dd-c31f-4208-abe4-f44cdff050c2',
+        #              'image_invariants': {'preset': 'any'},
+        #              'transparent': False,
+        #              'vector_image': False}],
+        #  'random_seed': 1423697946,
+        #  'request_id': '77bd917d-0960-4921-916f-038c773a41fd',
+        #  'transform_model': 'recraftv3',
+        #  'width': 1024}
+
+        params = {"raster_image_content_type": "image/webp"}  #####
+        params = {"raster_image_content_type": "image/png"}
+
+        images = []
+        for image in response.json()["images"]:
+            response = await client.get(f"""/image/{image["image_id"]}""", params=params)
+            url = await to_url(response.content)
+            images.append(url)
+
+        return ImagesResponse(image=images, metadata=metadata)
+
+
+async def check_token(token, threshold: float = 1):
+    if not isinstance(token, str):
+        tokens = token
+        r = []
+        for batch in tqdm(tokens | xgroup(32)):
+            bools = await asyncio.gather(*map(check_token, batch))
+            r += list(itertools.compress(batch, bools))
+        return r
+    try:
+        access_token = await get_access_token(token)
+        headers = {"Authorization": f"Bearer {access_token}"}
+
+        async with httpx.AsyncClient(base_url=BASE_URL, headers=headers, timeout=60) as client:
+            response = await client.get("/users/me")
+            response.raise_for_status()
+            data = response.json()
+            logger.debug(data["credits"])
+            return data["credits"] >= threshold
+    except Exception as e:
+        logger.error(e)
+        logger.debug(token)
+        return False
+
+
+if __name__ == '__main__':
+    token = None
+    # arun(get_access_token())
+    request = RecraftImageRequest(
+        prompt='一条猫'
+    )
+    # arun(generate(request, token=token))
+
+    tokens = list(arun(aget_spreadsheet_values(feishu_url=FEISHU_URL, to_dataframe=True))[0]) | xfilter_
+
+
+    r = arun(check_token(tokens))

{meutils-2024.12.23.16.36.51 → meutils-2024.12.31.17.51.53}/meutils/apis/jimeng/common.py

@@ -89,8 +89,8 @@ async def get_credit(token):
 async def check_token(token, threshold: int = 1):
     try:
         response = await get_credit(token)
+        logger.debug(bjson(response))
         credits = sum(response['data']['credit'].values())
-        # logger.debug(credits)
         return credits >= threshold
     except Exception as e:
         return False
@@ -244,8 +244,11 @@ async def create_draft_content(request: ImageRequest, token: str):
 if __name__ == '__main__':
     token = "693701c43e477b7c405cc7e2fef0ddbd"
     token = "eb4d120829cfd3ee957943f63d6152ed"
-    arun(get_credit(token))
-    # arun(check_token(token))
+    token = "dcf7bbc31faed9740b0bf748cd4d2c74"
+    token = "38d7d300b5e0a803431ef88d8d2acfef"
+    token = "916fed81175f5186a2c05375699ea40d"
+    # arun(get_credit(token))
+    arun(check_token(token))
 
     # arun(get_upload_token(token))
     #

{meutils-2024.12.23.16.36.51 → meutils-2024.12.31.17.51.53}/meutils/apis/jimeng/images.py

@@ -70,8 +70,19 @@ async def get_task(task_id, token):
         response = await client.post(url, json=payload)
         response.raise_for_status()
         data = response.json()
+        logger.debug(bjson(data))
         # {'ret': '1015', 'errmsg': 'login error', 'systime': '1734524280', 'logid': '20241218201800AC3267447B287E9E6C46', 'data': None}
-        item_list = (data.get("data") or {}).get(task_id, {}).get("item_list")
+        task_info = (data.get("data") or {}).get(task_id, {})
+        item_list = task_info.get("item_list")  # "status": 30,
+
+        status_code = task_info.get("status")
+        logger.debug(f"status: {status_code}")
+
+        """
+        "status": 30, # 内容审核
+        "status": 50,
+        """
+
         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")})
@@ -79,18 +90,29 @@ async def get_task(task_id, token):
         return TaskResponse(
             task_id=task_id,
             data=task_data,
-            message=data.get("errmsg"),
-            status="success" if task_data else None,
+            message=data.get("errmsg") if status_code!=30 else "触发内容审核451",
+            status="success" if item_list else 'processing',
+            code=status_code,
         )
 
 
+async def generate(request: ImageRequest):
+    task_response = await create_task(request)
+    for i in range(1, 10):
+        await asyncio.sleep(max(10 / i, 1))
+        response = await get_task(task_response.task_id, task_response.system_fingerprint)
+        logger.debug(response)
+        if data := response.data:
+            return {"data": data}
+
+
 if __name__ == '__main__':
-    token = "693701c43e477b7c405cc7e2fef0ddbd"
+    token = "eb4d120829cfd3ee957943f63d6152ed"
 
-    request = ImageRequest(prompt="一个漂亮的姑娘", size="1024x1024")
+    # request = ImageRequest(prompt="做一个圣诞节的海报", size="1024x1024")
     # request = ImageRequest(prompt="https://oss.ffire.cc/files/kling_watermark.png 让她带上墨镜", size="1024x1024")
 
-    arun(create_task(request))
+    # task = arun(create_task(request))
 
     # task_id = "10040025470722"
 
@@ -103,3 +125,16 @@ if __name__ == '__main__':
     # task_id = "10082971040514"
     #
     # arun(get_task(task_id, token))
+
+    # arun(get_task(task.task_id, task.system_fingerprint))
+
+    task_id = "10184295086338"
+    system_fingerprint = "eb4d120829cfd3ee957943f63d6152ed"
+
+    t1 = ("10184295086338", "eb4d120829cfd3ee957943f63d6152ed")
+    t2 = ("10184877310722", "dcf7bbc31faed9740b0bf748cd4d2c74")
+    t3 = ("10186352959490", "eb4d120829cfd3ee957943f63d6152ed")
+
+    arun(get_task(*t3))
+
+    # arun(generate(ImageRequest(prompt="做一个圣诞节的海报")))

{meutils-2024.12.23.16.36.51 → meutils-2024.12.31.17.51.53}/meutils/apis/kling/kolors_virtual_try_on.py

@@ -10,10 +10,13 @@
 import jwt
 
 from meutils.pipe import *
-from meutils.schemas.task_types import TaskResponse  ############# 根据这个重构
+from meutils.schemas.task_types import TaskResponse
+from meutils.schemas.image_types import ImageRequest, ImagesResponse
+
 from meutils.schemas.kling_types import API_BASE_URL, TryOnRequest, TaskResponse
 from meutils.config_utils.lark_utils import get_next_token_for_polling
 from meutils.decorators.retry import retrying
+from meutils.str_utils.regular_expression import parse_url
 
 FEISHU_URL = "https://xchatllm.feishu.cn/sheets/GYCHsvI4qhnDPNtI4VPcdw2knEd?sheet=cjTepf"
 
@@ -39,7 +42,7 @@ def encode_jwt_token(token):
 ######################################################################
 @alru_cache(ttl=3600)
 @retrying(title=__name__)
-async def create_task(request: TryOnRequest, token: Optional[str] = None):  ###### 所有的 TaskResponse标准化
+async def create_task(request: TryOnRequest, token: Optional[str] = None):
     token = token or await get_next_token_for_polling(FEISHU_URL)
     token = encode_jwt_token(token)
 
@@ -68,8 +71,23 @@ async def get_task(task_id, token):
         return TaskResponse(**data)
 
 
+async def generate(request: ImageRequest):
+    human_image, cloth_image = parse_url(request.prompt)  # 两张图片
+
+    request = TryOnRequest(human_image=human_image, cloth_image=cloth_image)
+    task_response = await create_task(request)
+
+    for i in range(1, 10):
+        await asyncio.sleep(max(10 / i, 1))
+
+        response = await get_task(task_response.data.task_id, task_response.system_fingerprint)
+        if task_result := response.data.task_result:
+            images = task_result.get('images', [])
+            return ImagesResponse(data=images)
+
+
 if __name__ == '__main__':
-    request = TryOnRequest()
+    # request = TryOnRequest()
     # arun(create_task(request))
     # {
     #     "code": 0,
@@ -83,5 +101,11 @@ if __name__ == '__main__':
     #     }
     # }
 
-    token = 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJkMmU1NmZkY2ZiODY0MmZmYWEyMjZlZWIxNzY4MzMyMiIsImV4cCI6MTczMzE4OTI4OSwibmJmIjoxNzMzMTAyODg0fQ.YjB792REbuFaUoyUXvd5bpI76_0PBWawDHDjPHOBA5Q'
-    arun(get_task("CjiL9mdJhswAAAAAAG2C6g", token))
+    # token = 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJkMmU1NmZkY2ZiODY0MmZmYWEyMjZlZWIxNzY4MzMyMiIsImV4cCI6MTczMzE4OTI4OSwibmJmIjoxNzMzMTAyODg0fQ.YjB792REbuFaUoyUXvd5bpI76_0PBWawDHDjPHOBA5Q'
+    # arun(get_task("CjiL9mdJhswAAAAAAG2C6g", token))
+
+    human_image: str = "https://oss.ffire.cc/files/try-on.png"
+    cloth_image: str = "https://oss.ffire.cc/files/x.png"
+
+    request = ImageRequest(prompt=f"{human_image}\n{cloth_image}")
+    arun(generate(request))

{meutils-2024.12.23.16.36.51 → meutils-2024.12.31.17.51.53}/meutils/apis/oneapi/user.py

@@ -91,10 +91,10 @@ async def put_user(payload, add_money: float = 0):
 
 if __name__ == '__main__':
     # api-key => get_one_log => get_user => put_user
-    # arun(get_user(814))
+    arun(get_user(814))
     # payload = arun(get_user(924))
     # print(payload)
     # arun(put_user(payload, -1))
 
-    arun(get_api_key_log('sk-gpoH1z3G6nHovD8MY40i6xx5tsC1vbh7B3Aao2jmejYNoKhv'))
-    # arun(get_user_money("sk-1UNhOrvuJ4PeBBpsB202577c14114f7fA234F32d3e8dF767"))
+    # arun(get_api_key_log('sk-gpoH1z3G6nHovD8MY40i6xx5tsC1vbh7B3Aao2jmejYNoKhv'))
+    # arun(get_user_money("sk-gpoH1z3G6nHovD8MY40i6xx5tsC1vbh7B3Aao2jmejYNoKhv"))

meutils-2024.12.31.17.51.53/meutils/apis/oneapi/utils.py

@@ -0,0 +1,32 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : utils
+# @Time         : 2024/12/25 18:13
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+from meutils.pipe import *
+
+from meutils.caches.redis_cache import cache
+from meutils.db.orm import select_first
+from meutils.schemas.db.oneapi_types import OneapiTask, OneapiUser, OneapiToken
+
+
+# engine
+
+
+@cache()
+async def token2user(api_key: str):
+    logger.debug(api_key)
+    filter_kwargs = {
+        "key": api_key[3:],
+    }
+
+    return await select_first(OneapiToken, filter_kwargs)
+
+
+if __name__ == '__main__':
+    arun(token2user('sk-q3uMFm0TFWi5lBQVhGwAWKup28j4omr9vThL4V5ynvdWdeZ8'))

{meutils-2024.12.23.16.36.51 → meutils-2024.12.31.17.51.53}/meutils/async_utils/common.py

@@ -10,7 +10,7 @@
 
 import inspect
 import asyncio
-from typing import Union, Any, Coroutine, AsyncIterator, Iterator, Iterable, AsyncIterable
+from typing import Union, Any, Coroutine, AsyncIterator, Iterator, Iterable, AsyncIterable, Optional, Callable
 
 import numpy as np
 import pandas as pd
@@ -146,6 +146,16 @@ async def arequest(url, method='get', payload=None, **client_params):
         return resp
 
 
+async def poll(fn, n: int = 30, sleep_fn: Optional[Callable] = None):
+    if sleep_fn is None:
+        sleep_fn = lambda i: max(n / (i + 1), 1)  # 预估大概 2~3 *n
+
+    for i in range(n):
+        await asyncio.sleep(sleep_fn(i))
+        if _ := await fn():  # 跳出
+            return _
+
+
 if __name__ == '__main__':
     from meutils.pipe import *
 

{meutils-2024.12.23.16.36.51 → meutils-2024.12.31.17.51.53}/meutils/common.py

@@ -31,6 +31,8 @@ import warnings
 warnings.filterwarnings("ignore")
 
 import functools
+from functools import wraps
+
 import argparse
 import traceback
 import threading