MeUtils 2025.9.7.1.36.27__py3-none-any.whl → 2025.9.9.15.11.11__py3-none-any.whl

examples/_openaisdk/openai_ffire.py

@@ -14,8 +14,10 @@ from openai import OpenAI
 from openai import OpenAI, APIStatusError
 
 client = OpenAI(
-    # base_url=os.getenv("FFIRE_BASE_URL"),
-    # api_key=os.getenv("FFIRE_API_KEY") #+"-29463"
+    base_url=os.getenv("FFIRE_BASE_URL"),
+    # api_key=os.getenv("FFIRE_API_KEY") +"-29487"
+    api_key=os.getenv("FFIRE_API_KEY") + "-29492",
+
     # base_url=os.getenv("ONEAPIS_BASE_URL"),
     # api_key=os.getenv("ONEAPIS_API_KEY") + "-3"
 
@@ -30,18 +32,23 @@ for i in range(1):
             # model="qwen3-235b-a22b-instruct-2507",
             # model="qwen-image",
             # model="glm-4.5",
-            model="deepseek-v3-1-think",
+            # model="deepseek-v3-1-think",
+            # model="kimi-k2-0905-preview",
+            # model="kimi-k2-0711-preview",
+
+            model="glm-4.5-air",
+
             #
             messages=[
                 {"role": "user", "content": 'are you ok?'}
             ],
             stream=True,
-            # max_tokens=1000,
+            max_tokens=10,
             # extra_body={"xx": "xxxxxxxx"}
             extra_body={
                 "thinking": {"type": "enabled"},
 
-                "enable_thinking": True # parameter.enable_thinking only support stream
+                "enable_thinking": True  # parameter.enable_thinking only support stream
             }
         )
         print(completion)

examples/_openaisdk/openai_moon.py

@@ -18,7 +18,7 @@ client = OpenAI(
     # base_url="http://ppu.chatfire.cc/v1"
 )
 
-print(client.models.list())
+# print(client.models.list())
 
 # completion = client.chat.completions.create(
 #     model="moonshot-v1-8k",
@@ -49,11 +49,11 @@ if __name__ == '__main__':
 
     from meutils.pipe import timer
 
-    # with timer():
-    #     completion = client.chat.completions.create(
-    #         model="moonshot-v1-128k",
-    #         messages=[
-    #             {"role": "user", "content": "字"*50000 + "一共有多少个字"},
-    #         ],
-    #         # temperature=0.3, extra_body={"refs": ["cn0bmk198onv4v01aafg"]}
-    #     )
+    with timer():
+        completion = client.chat.completions.create(
+            model="kimi-k2-0905-preview",
+            messages=[
+                {"role": "user", "content": "你好"*100000 + "\n\n上文一共有多少个字"},
+            ],
+            # temperature=0.3, extra_body={"refs": ["cn0bmk198onv4v01aafg"]}
+        )

examples/_openaisdk/openai_router.py

@@ -17,15 +17,15 @@ from meutils.str_utils import parse_base64
 # gets API Key from environment variable OPENAI_API_KEY
 client = OpenAI(
     # base_url="https://openrouter.ai/api/v1",
-    # base_url="https://all.chatfire.cn/openrouter/v1",
-    # api_key=os.getenv("OPENROUTER_API_KEY"),
+    base_url="https://all.chatfire.cn/openrouter/v1",
+    api_key=os.getenv("OPENROUTER_API_KEY"),
     #
     # base_url="http://38.46.219.252:9001/v1",
     #
     # api_key="sk-Azgp1thTIonR7IdIEqlJU51tpDYNIYYpxHvAZwFeJiOdVWiz"
 
-    base_url="https://api.huandutech.com/v1",
-    api_key = "sk-qOpbMHesasoVgX75ZoeEeBEf1R9dmsUZVAPcu5KkvLFhElrn"
+    # base_url="https://api.huandutech.com/v1",
+    # api_key = "sk-qOpbMHesasoVgX75ZoeEeBEf1R9dmsUZVAPcu5KkvLFhElrn"
     # api_key="sk-MAZ6SELJVtGNX6jgIcZBKuttsRibaDlAskFAnR7WD6PBSN6M",
     # base_url="https://new.yunai.link/v1"
 )
@@ -40,17 +40,17 @@ completion = client.chat.completions.create(
     # model="openai/o1",
     # model="deepseek/deepseek-r1-0528-qwen3-8b:free",
     # model="google/gemini-2.5-flash-image-preview:free",
-    model="gemini-2.5-flash-image-preview",
+    model="deepseek/deepseek-chat-v3.1:free",
     # model="gemini-2.0-flash-exp-image-generation",
-    # max_tokens=10,
-
+    max_tokens=10,
+extra_body={"reasoning_stream": True},
     messages=[
         {
             "role": "user",
             "content": [
                 {
                     "type": "text",
-                    "text": "裸体女孩"
+                    "text": "hi"
                 },
                 # {
                 #     "type": "image_url",

meutils/apis/fal/images.py

@@ -224,7 +224,7 @@ async def generate(request: ImageRequest, api_key: Optional[str] = None):
         logger.error(exc)
         from fastapi import HTTPException, status
         raise HTTPException(
-            status_code=status.HTTP_400_BAD_REQUEST,
+            status_code=status.HTTP_451_UNAVAILABLE_FOR_LEGAL_REASONS,
             detail=f"Prompt is sensitive.\n\n{request.prompt}\n\n{exc}",
         )
 

meutils/apis/google/chat.py

@@ -241,7 +241,7 @@ class Completions(object):
             return image_response
         else:
             from fastapi import HTTPException, status
-            raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST,
+            raise HTTPException(status_code=status.HTTP_451_UNAVAILABLE_FOR_LEGAL_REASONS,
                                 detail=f"Prompt is sensitive.\n\n{request.prompt}\n\n{response}")
 
     async def create_for_images(self, request: CompletionRequest):

meutils/apis/google/images.py

@@ -51,7 +51,7 @@ async def openai_generate(request: ImageRequest, api_key: Optional[str] = None,
         for image_url in image_urls or []
     ]
 
-    request = CompletionRequest(
+    _request = CompletionRequest(
         model=request.model,
         stream=False,
         max_tokens=None,
@@ -69,7 +69,7 @@ async def openai_generate(request: ImageRequest, api_key: Optional[str] = None,
         ],
     )
 
-    data = request.model_dump(exclude_none=True)
+    data = _request.model_dump(exclude_none=True)
 
     client = AsyncOpenAI(
         base_url=base_url,
@@ -112,7 +112,7 @@ async def openai_generate(request: ImageRequest, api_key: Optional[str] = None,
     # raise Exception(f"Image generate failed: {completion}")
     from fastapi import HTTPException, status
 
-    raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST,
+    raise HTTPException(status_code=status.HTTP_451_UNAVAILABLE_FOR_LEGAL_REASONS,
                         detail=f"Prompt is sensitive.\n\n{request.prompt}\n\n{completion}")
 
 

meutils/apis/meituan/__init__.py

@@ -8,4 +8,3 @@
 # @Software     : PyCharm
 # @Description  : 
 
-from meutils.pipe import *

meutils/apis/meituan/chat.py

@@ -9,27 +9,30 @@
 # @Description  :  todo
 
 
-from openai import AsyncOpenAI
+from openai import AsyncOpenAI, OpenAI
 
 from meutils.pipe import *
 from meutils.decorators.retry import retrying
 from meutils.io.files_utils import to_bytes, guess_mime_type
 from meutils.caches import rcache
 
-from meutils.llm.openai_utils import to_openai_params
+from meutils.llm.utils import oneturn2multiturn
 
 from meutils.config_utils.lark_utils import get_next_token_for_polling
 from meutils.schemas.openai_types import chat_completion, chat_completion_chunk, CompletionRequest, CompletionUsage, \
-    ChatCompletion
+    ChatCompletion, ChatCompletionChunk
+from openai._streaming import AsyncStream, Stream
 
 base_url = "https://longcat.chat/api/v1/chat-completion"
 base_url = "https://longcat.chat/api/v1/chat-completion-oversea"
 base_url = "https://longcat.chat/api/v1"
 
+cookie = "_lxsdk_cuid=1990e1e8790c8-0b2a66e23040a48-16525636-1fa400-1990e1e8790c8; passport_token_key=AgEGIygg22VuoMYTPonur9FA_-EVg9UXLu3LYOzJ4kIHSjQZeSNhwpytTU_cZFP6V1Juhk0CHMrAgwAAAABYLAAA9vXtnciaZBu2V99EMRJYRHTDSraV_OPLemUuVpi2WLsaa6RqC0PAKAOm6W_hIpbV"
+
 
 class Completions(object):
     def __init__(self, api_key: Optional[str] = None):
-        self.api_key = api_key
+        self.api_key = api_key or cookie
 
     async def create(self, request: CompletionRequest, **kwargs):
         payload = self.requset2payload(request)
@@ -47,28 +50,38 @@ class Completions(object):
                 response.raise_for_status()
 
                 async for chunk in response.aiter_lines():
-                    logger.debug(chunk)
+                    if chunk := chunk.removeprefix("data:"):
+                        _chunk = json.loads(chunk)
+
+                        chat_completion_chunk = ChatCompletionChunk.model_validate(_chunk)
+                        # logger.debug(chunk)
+
+                        yield chat_completion_chunk
+
+        if not request.stream:
+            # logger.debug(_chunk)
+            model = _chunk.get("model", "LongCat-Flash")
+            content = _chunk.get("content", "")
+            reason_content = _chunk.get("reasonContent", "")
+            token_info = _chunk.get("tokenInfo", {})
+
+            chat_completion_chunk.choices[0].content = content
+            chat_completion_chunk.choices[0].reason_content = reason_content
+            usage = CompletionUsage(
+                prompt_tokens=token_info.get("promptTokens"),
+                completion_tokens=token_info.get("completionTokens")
+            )
+            chat_completion.model = model
+            chat_completion.usage = usage
+            chat_completion.choices = chat_completion_chunk.choices
+            # logger.debug(chat_completion)
+            yield chat_completion
 
     def requset2payload(self, request: CompletionRequest):
         payload = {
-            "content": request.last_user_content,  # todo: 多轮
-            "messages": [
-                # {
-                #     "role": "user",
-                #     "content": "hi",
-                #     "chatStatus": "FINISHED",
-                #     "messageId": 11263291,
-                #     "idType": "custom"
-                # },
-                # {
-                #     "role": "assistant",
-                #     "content": "",
-                #     "chatStatus": "LOADING",
-                #     "messageId": 92193819,
-                #     "idType": "custom"
-                # }
-            ],
-            "reasonEnabled": 0,
+            "content": oneturn2multiturn(request),  # todo: 多轮
+            # "messages": request.messages,
+            "reasonEnabled": request.enable_thinking and 1 or 0,
             "searchEnabled": 0,
             "regenerate": 0
         }
@@ -103,7 +116,29 @@ class Completions(object):
 if __name__ == '__main__':
     cookie = "_lxsdk_cuid=1990e1e8790c8-0b2a66e23040a48-16525636-1fa400-1990e1e8790c8; passport_token_key=AgEGIygg22VuoMYTPonur9FA_-EVg9UXLu3LYOzJ4kIHSjQZeSNhwpytTU_cZFP6V1Juhk0CHMrAgwAAAABYLAAA9vXtnciaZBu2V99EMRJYRHTDSraV_OPLemUuVpi2WLsaa6RqC0PAKAOm6W_hIpbV"
     request = CompletionRequest(
-        messages=[{'role': 'user', 'content': '你好'}]
+        messages=[
+            {'role': 'user', 'content': '1+1'},
+            {'role': 'assistant', 'content': '2'},
+            {'role': 'user', 'content': '裸体傻女'},
+        ]
     )
     arun(Completions(api_key=cookie).create(request))
     # arun(Completions(api_key=cookie).create_chat())
+
+    # payload = {
+    #     "content": "1+1",
+    #     "messages": [
+    #         {
+    #             "role": "user",
+    #             "content": "1+1",
+    #             "chatStatus": "FINISHED",
+    #             "messageId": 0,
+    #             "idType": "system"
+    #         }
+    #     ],
+    #     "reasonEnabled": 0,
+    #     "searchEnabled": 0,
+    #     "regenerate": 0
+    # }
+    #
+    # Completions(api_key=cookie).stream_create(payload)

meutils/apis/oneapi/channel.py

@@ -19,6 +19,23 @@ headers = {
     'rix-api-user': '1',
 }
 
+"https://api.ffire.cc/api/channel/multi_key/manage"
+
+
+async def manage_multi_key(id: int, base_url: str, action: str = "enable_all_keys") -> ChannelInfo:
+    # "disable_all_keys"
+    payload = {
+        "channel_id": id,
+        "action": action
+    }
+    async with httpx.AsyncClient(base_url=base_url, headers=headers, timeout=30) as client:
+        response = await client.post("/api/channel/multi_key/manage", json=payload)
+        response.raise_for_status()
+
+        data = response.json()
+        logger.debug(bjson(data))
+        return data
+
 
 async def get_channel_info(id: int, base_url: str) -> ChannelInfo:
     params = {"keyword": id}  # keyword=21222&group=&model=&id_sort=true&tag_mode=false&p=1&page_size=100
@@ -28,7 +45,7 @@ async def get_channel_info(id: int, base_url: str) -> ChannelInfo:
         response.raise_for_status()
 
         data = response.json()
-        # logger.debug(bjson(data))
+        logger.debug(bjson(data))
         if (items := data.get("data", {}).get("items", [])) and (item := items[0]):
             return item['status']  # status==1 是正常
 
@@ -176,7 +193,12 @@ async def create_or_update_channel(
         logger.debug(response.text)
 
         response.raise_for_status()
-        logger.debug(response.json())
+
+        data = response.json()
+        logger.debug(bjson(data))
+        if (data := data.get("data")) and data['channel_info']['is_multi_key']:
+            await manage_multi_key(data['id'], base_url=base_url)
+
         return response.json()
 
 
@@ -268,15 +290,20 @@ if __name__ == '__main__':
     # arun(create_or_update_channel(tokens, base_url))
     # arun(create_or_update_channel(tokens))
     # # arun(delete_channel(range(10000, 20000)))
-    key = "KEY"
-    request = ChannelInfo(name='', key=key)
-    request = ChannelInfo(id=21223, key=key, used_quota=0.001)
-
-    # arun(create_or_update_channel(request, base_url=base_url))
+    key = "KEY\nKEY2"
+    # request = ChannelInfo(id = 1, name='xx', key=key)
+    # request = ChannelInfo(id=21223, key=key, used_quota=0.001)
+    request = ChannelInfo(id=29483, key=key, used_quota=0.001,
+                          # key_mode="append"
+                          )
+
+    arun(create_or_update_channel(request, base_url=base_url))
     #
     # arun(exist_channel(request, base_url=base_url))
 
-    arun(get_channel_info(21222, base_url=base_url))
-    # arun(get_channel_info(21223, base_url=base_url))
+    # arun(get_channel_info(21222, base_url=base_url))
+    # arun(get_channel_info(29483, base_url=base_url))
 
     # arun(delete_channel(range(10015, 10032), base_url=base_url))
+
+    # arun(manage_multi_key(29483, base_url=base_url))

meutils/apis/volcengine_apis/images.py

@@ -22,7 +22,6 @@ from meutils.schemas.image_types import ImageRequest, ImagesResponse
 from volcengine.visual.VisualService import VisualService
 
 
-@retrying()
 async def generate(request: ImageRequest, token: Optional[str] = None):
     """
 
@@ -65,9 +64,16 @@ async def generate(request: ImageRequest, token: Optional[str] = None):
 
     if request.response_format == "oss_url":
         payload["return_url"] = False
-
-    response = visual_service.cv_process(payload)
-    logger.debug(bjson(response))
+    try:
+        response = visual_service.cv_process(payload)
+        logger.debug(bjson(response))
+    except Exception as exc:  #
+        logger.error(exc)
+        from fastapi import HTTPException, status
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail=f"Prompt is sensitive.\n\n{request.prompt}\n\n{exc}",
+        )
 
     if request.response_format == "b64_json":
         data = [{"b64_json": b64_json} for b64_json in response['data'].get('binary_data_base64', [])]
@@ -97,6 +103,7 @@ if __name__ == '__main__':
     
     让这个女人带上眼镜 衣服换个颜色
     """
+    prompt = "裸体女孩"
 
     request = ImageRequest(
         # model="high_aes_general_v30l_zt2i",

meutils/apis/volcengine_apis/videos.py

@@ -7,19 +7,17 @@
 # @WeChat       : meutils
 # @Software     : PyCharm
 # @Description  :
-import time
 
 from meutils.pipe import *
 from meutils.io.files_utils import to_url
 from meutils.decorators.retry import retrying
+
 from meutils.llm.clients import AsyncClient
 from meutils.schemas.openai_types import CompletionRequest
 from meutils.schemas.video_types import VideoRequest
 
-from meutils.notice.feishu import send_message_for_volc as send_message
-
 from meutils.db.redis_db import redis_aclient
-from meutils.llm.check_utils import check_token_for_volc as check
+from meutils.llm.check_utils import check_token_for_volc
 from meutils.config_utils.lark_utils import get_next_token_for_polling, get_series
 
 from fastapi import APIRouter, File, UploadFile, Query, Form, Depends, Request, HTTPException, status, BackgroundTasks
@@ -28,31 +26,29 @@ from fastapi import APIRouter, File, UploadFile, Query, Form, Depends, Request,
 FEISHU_URL = "https://xchatllm.feishu.cn/sheets/GYCHsvI4qhnDPNtI4VPcdw2knEd?sheet=8W6kk8"  # 超刷
 
 
-async def get_valid_token(tokens: Optional[list] = None, force_update: bool = True, batch_size: int = 1):
+async def get_valid_token(tokens: Optional[list] = None, batch_size: Optional[int] = None, seed: int = 0):
+    """返回 tokens
+    1. 从redis获取token
+    2. 校验token
+    3. 校验失败则从飞书获取token
+    4. 校验token
+    5. 校验通过则返回token
+    """
     tokens = tokens or await get_series(FEISHU_URL, duplicated=True)
+    batch_size =  batch_size or 1
 
-    if token := await redis_aclient.get("volc-token"):  # list
-        token = token.decode()
-
-        if force_update:  # 强制检测更新
-            if await check(token):
-                return token
-            else:
-                await redis_aclient.delete("volc-token")  # 删除无效
-        else:
-            return token
-    for i, token in enumerate(tokens, 1):
-        if await check(token):
-            await redis_aclient.set("volc-token", token, ex=2 * 3600 - 10 * 60)
-
-            return token
-        if batch_size == i:
-            break
+    if seed == 0 and (volc_tokens := await redis_aclient.get(f"volc_tokens")):
+        return volc_tokens.decode()
 
-    logger.debug(f"无效 {token}")
-    _ = f"{time.ctime()}\n\n{FEISHU_URL}\n\n所有token无效\n\n{token}"
-    logger.error(_)
-    send_message(_, n=3)
+    valid_tokens = []
+    for token in tokens:
+        if token := await check_token_for_volc([token]):
+            valid_tokens += token
+            logger.debug(valid_tokens)
+            if len(valid_tokens) == batch_size:
+                _ = '\n'.join(valid_tokens)
+                await redis_aclient.set(f"volc_tokens", _, ex=2 * 3600)
+                return _
 
 
 # check_image_and_video = partial(check, purpose='video and image')
@@ -61,7 +57,9 @@ async def get_valid_token(tokens: Optional[list] = None, force_update: bool = Tr
 @retrying()
 async def create_task(request: Union[CompletionRequest, VideoRequest], api_key: Optional[str] = None):
     # api_key = api_key or await get_next_token_for_polling(feishu_url=FEISHU_URL, check_token=check)
-    api_key = api_key or await get_valid_token(force_update=False)
+    api_key = api_key or await get_valid_token()
+
+    api_key = np.random.choice(api_key.split('\n'))
 
     # feishu_url = "https://xchatllm.feishu.cn/sheets/Z59Js10DbhT8wdt72LachSDlnlf?sheet=rcoDg7"
     # api_key = api_key or await get_next_token_for_polling(
@@ -267,7 +265,8 @@ c4356b58-4aa3-4a52-b907-b40c4dd2e502
     # arun(get_task_from_feishu(ids, tokens))
     # arun(get_task_from_feishu(ids, ))
 
-    arun(get_valid_token())
+    # arun(get_valid_token(['a93ea9a5-3831-47b8-863a-57e10233f922']))
+    arun(get_valid_token(batch_size=2, seed=3))
 
 """
 {'id': 'cgt-20250613160030-2dvd7',

meutils/data/VERSION

@@ -1 +1 @@
-2025.09.07.01.36.27
+2025.09.09.15.11.11

meutils/llm/check_utils.py

@@ -176,7 +176,7 @@ async def check_token_for_ppinfra(api_key, threshold: float = 1):  # 1块钱 100
     if not isinstance(api_key, str):
         return await check_tokens(api_key, partial(check_token_for_ppinfra, threshold=threshold))
     try:
-        client = AsyncOpenAI(base_url="https://api.ppinfra.com/v3/user", api_key=api_key)
+        client = AsyncOpenAI(base_url="https://api.ppinfra.com/v3/user", api_key=api_key.strip())
         data = await client.get("", cast_to=object)
         logger.debug(data)  # credit_balance
         return data["credit_balance"] >= threshold
@@ -214,7 +214,7 @@ async def check_token_for_sophnet(api_key, threshold: float = 1):
 
 #
 @retrying()
-# @rcache(ttl=7 * 24 * 3600, skip_cache_func=skip_cache_func)
+@rcache(ttl=7 * 24 * 3600, skip_cache_func=skip_cache_func)
 async def check_token_for_volc(api_key, threshold: float = 1, purpose: Optional[str] = None):
     if not isinstance(api_key, str):
         return await check_tokens(api_key, check_token_for_volc)

meutils/llm/utils.py

@@ -8,16 +8,13 @@
 # @Software     : PyCharm
 # @Description  :
 
-from contextlib import asynccontextmanager
 
 from meutils.pipe import *
-from meutils.llm.clients import AsyncOpenAI, OpenAI, AsyncStream
-from meutils.schemas.oneapi import MODEL_PRICE
-from meutils.notice.feishu import send_message
-from meutils.apis.oneapi.utils import get_user_quota
+from meutils.schemas.openai_types import CompletionRequest
 
 
-def oneturn2multiturn(messages, template: Optional[str] = None, ignore_system: bool = True):
+
+def oneturn2multiturn(messages: Union[dict, CompletionRequest], template: Optional[str] = None, ignore_system: bool = True):
     """todo: https://github.com/hiyouga/LLaMA-Factory/blob/e898fabbe3efcd8b44d0e119e7afaed4542a9f39/src/llmtuner/data/template.py#L423-L427
 
     _register_template(
@@ -45,6 +42,10 @@ def oneturn2multiturn(messages, template: Optional[str] = None, ignore_system: b
     #     role, content = message.get("role"), message.get("content")
     #     context += f"<|im_start|>{role}\n{content}<|im_end|>\n"
     # context += "<|im_start|>assistant\n"
+    if isinstance(messages, CompletionRequest):
+        messages = messages.messages
+
+
     if len(messages) == 1:
         content = messages[0].get("content")
         if isinstance(content, list):
@@ -53,6 +54,7 @@ def oneturn2multiturn(messages, template: Optional[str] = None, ignore_system: b
 
     context = "\n"
     for message in messages:
+        logger.info(message)
         role = message.get("role")
         content = message.get("content")
 

meutils/schemas/oneapi/_types.py

@@ -91,9 +91,9 @@ class ChannelInfo(BaseModel):
     # new
     max_input_tokens: int = 0
 
-    mode:Optional[Literal["multi_to_single"]] = None
+    mode: Optional[Literal["multi_to_single"]] = None
     multi_key_mode: Literal['random', 'polling'] = "polling"
-    key_mode: Optional[Literal['append', 'replace']] = None
+    key_mode: Optional[Literal['append', 'replace']] = None  # none是覆盖
 
     def __init__(self, /, **data: Any):
         super().__init__(**data)

meutils/schemas/oneapi/common.py

@@ -29,6 +29,7 @@ FAL_MODELS = {
     "fal-ai/clarity-upscaler": 1,  # Your request will cost $0.03 per upscaled megapixel.
 
     'fal-kling-video-lipsync-audio-to-video': 0.5,
+    'fal-ai/kling-video/lipsync/text-to-video': 0.14 * FAL,
 
     'fal-pixverse-v4.5-effects': 1,
     'fal-pixverse-v4.5-text-to-video': 0.9,