MeUtils 2025.8.29.20.5.48__py3-none-any.whl → 2025.9.5.20.57.37__py3-none-any.whl

meutils/llm/check_utils.py

@@ -7,6 +7,7 @@
 # @WeChat       : meutils
 # @Software     : PyCharm
 # @Description  :
+import os
 
 from meutils.pipe import *
 from meutils.caches import rcache
@@ -227,7 +228,7 @@ async def check_token_for_volc(api_key, threshold: float = 1, purpose: Optional[
 
             payload = {
                 # "model": "doubao-seedance-1-0-pro-250528",
-                "model":"doubao-seedance-1-0-lite-i2v-250428",
+                "model": "doubao-seedance-1-0-lite-i2v-250428",
                 "content": [
                     {
                         "type": "text",
@@ -358,6 +359,32 @@ async def check_token_for_gitee(api_key, threshold: float = 1):
         return False
 
 
+@retrying()
+# @rcache(ttl=1 * 24 * 3600, skip_cache_func=skip_cache_func)
+async def check_token_for_openrouter(api_key, threshold: float = 1):
+    if not isinstance(api_key, str):
+        return await check_tokens(api_key, check_token_for_openrouter)
+
+    try:
+        # {"data": {"total_credits": 215.00205323, "total_usage": 215.20147321775545}} %
+        client = AsyncOpenAI(base_url=os.getenv("OPENROUTER_BASE_URL"), api_key=api_key)
+        data = await client.get("/credits", cast_to=object)
+        # logger.debug(data)
+        # data = await client.get("/models/user", cast_to=object)
+        # logger.debug(bjson(data))
+
+        #
+        # logger.debug(bjson(data))
+
+        return True
+    except TimeoutException as e:
+        raise
+
+    except Exception as e:
+        logger.error(f"Error: {e}\n{api_key}")
+        return False
+
+
 async def get_valid_token_for_fal(feishu_url: Optional[str] = None):
     feishu_url = feishu_url or "https://xchatllm.feishu.cn/sheets/Z59Js10DbhT8wdt72LachSDlnlf?sheet=iFRwmM"
     _ = await get_next_token(feishu_url, check_token_for_fal, ttl=600)
@@ -400,9 +427,9 @@ if __name__ == '__main__':
 
     # arun(check_token_for_ppinfra("sk_F0kgPyCMTzmOH_-VCEJucOK8HIrbnLGYm_IWxBToHZQ"))
 
-    # arun(check_token_for_volc("c720b2fb-e2be-42a7-a9e5-0a42c29d5766"))
+    # arun(check_token_for_volc("827b41d2-7e8c-46e2-9854-0720ca1bd2e4"))
     # arun(check_token_for_volc("279749bd-ba5e-4962-9c65-eb6604b65594"))
-    arun(check_token_for_volc("8a5af7cb-42a3-4391-ac40-9d0f4502acde", purpose='seedance'))
+    # arun(check_token_for_volc("8a5af7cb-42a3-4391-ac40-9d0f4502acde", purpose='seedance'))
 
     # arun(check_token_for_ppinfra("sk_mCb5sRGTi6GXkSRp5F679Rbs0V_Hfee3p85lccGXCOo"))
 
@@ -410,8 +437,12 @@ if __name__ == '__main__':
 
     # arun(check_token_for_fal("56d8a95e-2fe6-44a6-8f7d-f7f9c83eec24:537f06b6044770071f5d86fc7fcd6d6f"))
 
-    # arun(check_token_for_ppinfra("sk_ib2EjSVnXfB5hSlVuckpejRNXLIU3MaD1wxvXnsvdxQ", threshold=18000))
+    # arun(check_token_for_ppinfra("sk_4Ja29OIUBVwKo5GWx-PRTsRcTyxxRjZDpYxSdPg75QU", threshold=18000))
 
     # arun(check_token_for_gitee("NWVXUPI38OQVXZGOEL3D23I9YUQWZPV23GVVBW1X"))
 
     # arun(get_valid_token_for_fal())
+
+    api_key = "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"
+    # api_key = None
+    arun(check_token_for_openrouter(api_key=api_key or os.getenv("OPENROUTER_API_KEY")))

meutils/llm/openai_polling/chat.py

@@ -34,18 +34,61 @@ class Completions(object):
         self.client = AsyncOpenAI(base_url=self.base_url, api_key=self.api_key, http_client=http_client)
 
     async def create(self, request: CompletionRequest):
+        # todo:
+        # 1. 可思考模型 思考优先
+        # request.thinking = {"type": "enabled"}
+
+        if request.thinking and request.model.startswith("doubao-seed") and request.thinking.get("type") == "disabled":
+            choices = {
+                          "doubao-seed-1.6-250615",
+                          "doubao-seed-1-6-250615",
+                          "doubao-seed-1-6-vision-250815"
+                      } - {request.model}
+            request.model = np.random.choice(list(choices))
+        elif request.model.startswith("doubao-seed"):
+            request.thinking = {"type": "enabled"}
+            choices = {
+                          "doubao-seed-1.6-250615",
+                          "doubao-seed-1-6-250615",
+                          "doubao-seed-1-6-vision-250815",
+                          "doubao-seed-1-6-thinking-250615",
+                          "doubao-seed-1-6-thinking-250715"
+                      } - {request.model}
+            request.model = np.random.choice(list(choices))
+
+
+        elif request.model.startswith(("doubao-1-5-thinking",)):  # thinking
+            request.thinking = None
+
+            choices = {
+                          "doubao-1-5-thinking-pro-250415",
+                          "doubao-1-5-thinking-pro-m-250428",
+                          "doubao-1-5-thinking-vision-pro-250428",
+                          "doubao-1-5-ui-tars-250428",
+                      } - {request.model}
+            request.model = np.random.choice(list(choices))
+
+        elif request.model.startswith(("doubao-1-5", "doubao-1.5")):  # nothinking
+            request.thinking = {"type": "disabled"}
+
+            choices = {
+                          "doubao-1-5-pro-32k-250115",
+                          "doubao-1-5-pro-256k-250115",
+                          "doubao-1-5-pro-32k-character-250715",
+                          "doubao-1-5-pro-32k-character-250228",
+                          "doubao-1.5-vision-pro-250328",
+                          "doubao-1-5-vision-pro-32k-250115",
+                          "doubao-1-5-thinking-pro-250415",
+                          "doubao-1-5-thinking-pro-m-250428",
+                          "doubao-1-5-thinking-vision-pro-250428",
+                          "doubao-1-5-ui-tars-250428",
+                      } - {request.model}
+            request.model = np.random.choice(list(choices))
+        # todo vision
 
-        if request.model.startswith("doubao-seed"):  # todo: 1-5
-            choices = """
-            doubao-seed-1-6-250615
-            doubao-seed-1-6-thinking-250615
-            doubao-seed-1-6-thinking-250715
-            doubao-seed-1-6-vision-250815
-            """.split()
-            request.model = np.random.choice(choices)
         ###########################################################################
         # 开启视觉模型
-        if not any(i in request.model.lower() for i in ["v-", "vl", 'vision', 'doubao-seed']) and (
+        if not any(i in request.model.lower() for i in ["v-", "vl", "ui", 'vision', 'doubao-seed']) and (
                 urls := request.last_urls.get("image_url")):
             # logger.debug(request)
             if request.model.startswith(("gemini",)):
@@ -125,8 +168,12 @@ if __name__ == '__main__':
         # model="gemini-2.0-flash",
         # model="glm-4-flash",
         # model="deepseek-ai/DeepSeek-V3",
-        model="Qwen/Qwen3-8B1",
-
+        # model="Qwen/Qwen3-8B1",
+        # model="Qwen/Qwen3-8B1",
+        # model="doubao-1-5-thinking",
+        # model="doubao-1-5",
+        model="doubao-seed",
+        thinking={"type": "disabled"},
         messages=[
             {
                 "role": "system",
@@ -141,7 +188,7 @@ if __name__ == '__main__':
             {"role": "user", "content": [
                 {
                     "type": "text",
-                    "text": "解释下" * 30000
+                    "text": "解释下" * 1
                 },
                 # {
                 #     "image_url": {
@@ -153,7 +200,7 @@ if __name__ == '__main__':
             ]}
         ],
         stream=False,
-        max_tokens=None,
+        max_tokens=10,
     )
     # arun(Completions().create(request))
     # d = {
@@ -222,6 +269,10 @@ if __name__ == '__main__':
     #     stream=False,
     #     max_tokens=None,
     # )
-    api_key = os.getenv("SILICONFLOW_API_KEY")
-    base_url = "https://api.siliconflow.cn"
+    # api_key = os.getenv("SILICONFLOW_API_KEY")
+    # base_url = "https://api.siliconflow.cn"
+    # arun(Completions(api_key=api_key, base_url=base_url).create(request))
+
+    api_key = os.getenv("VOLC_API_KEY")
+    base_url = os.getenv("VOLC_BASE_URL")
     arun(Completions(api_key=api_key, base_url=base_url).create(request))

meutils/llm/openai_utils/adapters.py

@@ -33,26 +33,6 @@ async def chat_for_image(
 ):
     generate = generate and partial(generate, api_key=api_key, base_url=base_url)
 
-    if not request.stream or request.last_user_content.startswith(  # 跳过nextchat
-            (
-                    "hi",
-                    "使用四到五个字直接返回这句话的简要主题",
-                    "简要总结一下对话内容，用作后续的上下文提示 prompt，控制在 200 字以内"
-            )):
-        chat_completion.choices[0].message.content = "请设置`stream=True`"
-        return chat_completion
-
-        # request.stream = True  # 流转非流
-        # response = await chat_for_image(generate, request, api_key)
-        # chunks = await stream.list(response)
-        #
-        # logger.debug(chunks)
-        #
-        # if chunks and isinstance(chunks[0], ChatCompletion):
-        #     response = chunks[0]
-        # else:
-        #     response = create_chat_completion(chunks)
-        # return response
     image = None
     prompt = request.last_user_content
     if image_urls := request.last_urls.get("image_url"):  # image_url
@@ -63,32 +43,55 @@ async def chat_for_image(
             urls = await to_url(image_urls, content_type="image/png")  # 数组
             image = urls
 
-    request = ImageRequest(
+    image_request = ImageRequest(
         model=request.model,
         prompt=prompt,
         image=image
     )
-    if not request.image:
-        request.image, request.prompt = request.image_and_prompt
+    if not image_request.image:
+        image_request.image, image_request.prompt = image_request.image_and_prompt
 
-    if '--' in request.prompt:
-        prompt_dict = parse_command_string(request.prompt)
+    if '--' in image_request.prompt:
+        prompt_dict = parse_command_string(image_request.prompt)
         # 缩写补充
         prompt_dict['aspect_ratio'] = prompt_dict.get('aspect_ratio') or prompt_dict.get('ar')
 
         data = {
-            **request.model_dump(exclude_none=True, exclude={"extra_fields", "aspect_ratio"}),
+            **image_request.model_dump(exclude_none=True, exclude={"extra_fields", "aspect_ratio"}),
             **prompt_dict
         }
-        request = ImageRequest(**data)
-        logger.debug(request)
+        image_request = ImageRequest(**data)
+        logger.debug(image_request)
 
+    # 非流式
+    if not request.stream or request.last_user_content.startswith(  # 跳过nextchat
+            (
+                    "hi",
+                    "使用四到五个字直接返回这句话的简要主题",
+                    "简要总结一下对话内容，用作后续的上下文提示 prompt，控制在 200 字以内"
+            )):
+        # chat_completion.choices[0].message.content = "请设置`stream=True`"
+        # return chat_completion
+
+        response = await generate(image_request)
+
+        if not isinstance(response, dict):
+            response = response.model_dump()
+
+        content = ""
+        for image in response['data']:
+            content += f"""![{image.get("revised_prompt")}]({image['url']})\n\n"""
+
+        chat_completion.choices[0].message.content = content
+        return chat_completion
+
+    # 流式
     if not generate: return
 
-    future_task = asyncio.create_task(generate(request))  # 异步执行
+    future_task = asyncio.create_task(generate(image_request))  # 异步执行
 
     async def gen():
-        text = request.model_dump_json(exclude_none=True).replace("free", "")
+        text = image_request.model_dump_json(exclude_none=True).replace("free", "")
         for i in f"""> 🖌️正在绘画\n\n```json\n{text}\n```\n\n""":
             await asyncio.sleep(0.05)
             yield i
@@ -151,6 +154,7 @@ async def chat_for_video(
 
 
 if __name__ == '__main__':
+    from meutils.apis.images.generations import generate
     request = CompletionRequest(
         model="deepseek-r1-Distill-Qwen-1.5B",
         messages=[
@@ -159,3 +163,14 @@ if __name__ == '__main__':
         stream=True,
     )
     arun(chat_for_image(None, request))
+
+    request = CompletionRequest(
+        model="gemini-2.5-flash-image-preview",
+        messages=[
+            {"role": "user", "content": "画条狗"}
+        ],
+        # stream=True,
+    )
+    api_key = "sk-MAZ6SELJVtGNX6jgIcZBKuttsRibaDlAskFAnR7WD6PBSN6M-openai"
+    base_url = "https://new.yunai.link/v1"
+    arun(chat_for_image(generate, request, api_key, base_url))

meutils/schemas/oneapi/common.py

@@ -101,6 +101,11 @@ FAL_MODELS = {
     "fal-ai/wan/v2.2-5b/text-to-video": 0.15 * FAL_,
     "fal-ai/wan/v2.2-5b/image-to-video": 0.15 * FAL_,
 
+    # pika
+    "fal-ai/pika/v2.2/text-to-video": 0.45 * FAL,
+    "fal-ai/pika/v2.2/image-to-video": 0.45 * FAL,
+    "fal-ai/pika/v2.2/pikascenes": 0.45 * FAL,
+
 }
 
 FAL_MODELS = {
@@ -661,6 +666,7 @@ MODEL_RATIO = {
     "moonshotai/kimi-k2-instruct": 2,
     "kimi-k2-0711-preview": 2,
     "kimi-k2-turbo-preview": 2,
+    "kimi-k2-250711": 2,
 
     # 智谱 https://www.bigmodel.cn/pricing
     'glm-4-9b-chat': 0.1,
@@ -1251,6 +1257,7 @@ COMPLETION_RATIO = {
     "moonshotai/kimi-k2-instruct": 4,
     "kimi-k2-0711-preview": 4,
     "kimi-k2-turbo-preview": 4,
+    "kimi-k2-250711": 4,
 
     "moonshot-v1-8k": 5,
     "moonshot-v1-32k": 4,

meutils/serving/fastapi/gunicorn.conf.py

@@ -78,10 +78,10 @@ backlog = 1024
 
 workers = int(os.getenv("FASTAPI_WORKERS", 2))
 worker_class = "uvicorn.workers.UvicornWorker"
-worker_connections = 1000
-timeout = 300
-graceful_timeout = 100  # 在接收到重新启动信号后，worker有这么多时间来完成服务请求。超时后（从接收到重启信号开始）仍然活着的worder被强制杀死。
-keepalive = 5
+worker_connections = 1024
+timeout = 360
+graceful_timeout = 128  # 在接收到重新启动信号后，worker有这么多时间来完成服务请求。超时后（从接收到重启信号开始）仍然活着的worder被强制杀死。
+keepalive = 16
 reload = False
 
 #

meutils/str_utils/__init__.py

@@ -10,7 +10,7 @@
 
 
 from meutils.pipe import *
-from meutils.str_utils.regular_expression import parse_url, parse_command_string
+from meutils.str_utils.regular_expression import parse_url, parse_command_string, parse_base64
 from meutils.request_utils.crawler import Crawler
 from urllib.parse import urlencode, parse_qs, parse_qsl, quote_plus, unquote_plus, urljoin
 

meutils/str_utils/regular_expression.py

@@ -114,6 +114,7 @@ def parse_url(text: str, for_image=False, fn: Optional[Callable] = None):
 def parse_url_from_json():
     pass
 
+
 def parse_command_string(command_str: str) -> dict:
     """
     解析一个类似 "prompt --key1 value1 --key2 value2" 格式的字符串。
@@ -188,6 +189,16 @@ def parse_command_string(command_str: str) -> dict:
     return result
 
 
+def parse_base64(text, pattern=r'!\[.*?\]\((.*?)\)'):
+    """
+    :param text:
+    :param pattern:
+        pattern=r'!\[.*?\]\((data:image/.*?)\)'
+    :return:
+    """
+    base64_strings = re.findall(pattern, text)
+    return base64_strings
+
 
 if __name__ == '__main__':
     # from urllib.parse import urlparse
@@ -258,7 +269,6 @@ https://i.miji.bid/2025/06/10/d018000aed9b872c7b248dccf14c4450.pngA
     """
     print(parse_url(text, for_image=True))
 
-
     # print(parse_url(text, for_image=False))
 
     # text = """https://photog.art/api/oss/R2yh8N Convert this portrait into a straight-on,front-facing ID-style headshot."""
@@ -266,4 +276,9 @@ https://i.miji.bid/2025/06/10/d018000aed9b872c7b248dccf14c4450.pngA
     #
     # valid_urls = parse_url(text, for_image=True)
 
-    print(mimetypes.guess_type("xx.ico"))
+    print(mimetypes.guess_type("xx.ico"))
+
+    text = "这是一个示例文本，包含一个图片：![image](data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAABAAAAAQACAIAAADwf7zUAAAgAElEQ) 这张图片很棒。"
+    # text = "这是一个示例文本，。"
+
+    print(parse_base64(text * 2))