MeUtils 2025.8.7.20.31.52__py3-none-any.whl → 2025.8.11.16.30.21__py3-none-any.whl

meutils/llm/openai_utils/billing_utils.py

@@ -21,6 +21,8 @@ from contextlib import asynccontextmanager
 from meutils.pipe import *
 from meutils.llm.clients import AsyncOpenAI
 from meutils.apis.utils import make_request
+from meutils.apis.oneapi.user import get_user_money
+from fastapi import status, HTTPException
 
 base_url = "https://api.chatfire.cn/flux/v1"
 
@@ -200,15 +202,27 @@ async def billing_flow_for_async_task(
         n: float = 1,
         api_key: Optional[str] = None
 ):
-    if n:  # 计费
-        await billing_for_async_task(model, task_id=task_id, n=n, api_key=api_key)
+    if n and (user_money := await get_user_money(api_key)):  # 计费
+
+        # 判断
+        if user_money < 1:
+            raise HTTPException(status_code=status.HTTP_402_PAYMENT_REQUIRED, detail="余额不足或API-KEY限额")
+
+        # 执行
         yield
 
+        # 计费
+        await billing_for_async_task(model, task_id=task_id, n=n, api_key=api_key)
+
     else:  # 不计费
         a = yield
         logger.debug(a)
 
 
+# # 检查余额
+# if user_money := await get_user_money(api_key):
+#     if user_money < 1:
+#         raise HTTPException(status_code=status.HTTP_402_PAYMENT_REQUIRED, detail="余额不足或API-KEY限额")
 @asynccontextmanager
 async def billing_flow_for_tokens(
         model: str = "usage-chat",

meutils/schemas/oneapi/common.py

@@ -81,6 +81,7 @@ FAL_MODELS = {
 
     # veo
     "fal-ai/veo3/fast": 3.2 * FAL_,  # 2 3.2
+    "fal-ai/veo3/fast/image-to-video": 3.2 * FAL_,
 
     # seedance
     "fal-ai/bytedance/seedance/v1/lite/text-to-video": 1,
@@ -141,7 +142,7 @@ MODEL_PRICE = {
     "gemini-2.5-pro-video": 0.1,
 
     # rix
-    "kling_image": 0.05,
+    "kling_image": 0.025,
     "kling_image_expand": 0.3,
     "kling_virtual_try_on": 1,
     "kling_effects": 1,
@@ -167,6 +168,7 @@ MODEL_PRICE = {
     # 火山
     "jimeng-video-3.0": 0.5,
     "doubao-seedream-3-0-t2i-250415": 0.1,
+    "doubao-seededit-3-0-i2i-250628": 0.1,
 
     "doubao-seedance-1-0-pro-250528": 0.5,  # 480 1080
     "api-doubao-seedance-1-0-pro-250528-1080p": 2,  # 480 1080
@@ -410,6 +412,7 @@ MODEL_PRICE = {
     "gpt-4o-image": 0.05,
     "sora-image": 0.1,
     "gpt-image-1": 0.1,
+    "gpt-5-thinking": 0.05,
 
     "gpt-4-gizmo-*": 0.1,
     "advanced-voice": 1,
@@ -965,6 +968,7 @@ MODEL_RATIO = {
     "claude-sonnet-4-20250514-thinking": 1.5,
     "claude-opus-4-20250514": 7.5,
     "claude-opus-4-20250514-thinking": 7.5,
+    "claude-opus-4.1": 7.5,
 
     "deepclaude": 1.5,
     "deep-claude": 1.5,
@@ -1059,8 +1063,6 @@ MODEL_RATIO = {
     "gpt-4-1106-vision-preview": 5,
     "gpt-4-32k": 30,
     "gpt-4-32k-0613": 30,
-    "gpt-4-all": 15,
-    "gpt-4-gizmo-*": 15,
     "gpt-4-turbo": 5,
     "gpt-4-turbo-2024-04-09": 5,
     "gpt-4-turbo-preview": 5,
@@ -1094,6 +1096,14 @@ MODEL_RATIO = {
     "gpt-oss-20b": 0.025,
     "gpt-oss-120b": 0.05,
 
+    "gpt-5": 0.625,
+    "gpt-5-2025-08-07": 0.625,
+    "gpt-5-chat-latest": 0.625,
+    "gpt-5-mini": 0.125,
+    "gpt-5-mini-2025-08-07": 0.125,
+    "gpt-5-nano": 0.025,
+    "gpt-5-nano-2025-08-07": 0.025,
+
     "o1": 7.5,
     "o1-2024-12-17": 7.5,
 
@@ -1217,9 +1227,6 @@ COMPLETION_RATIO = {
     "grok-3-mini-fast-beta": 4 / 0.6,
     "grok-4": 5,
 
-    "gpt-4-all": 4,
-    "gpt-4-gizmo-*": 4,
-    "gpt-4o-all": 4,
     "gpt-4.5-preview-2025-02-27": 2,
 
     "o1-mini": 4,
@@ -1254,6 +1261,14 @@ COMPLETION_RATIO = {
     "gpt-oss-20b": 4,
     "gpt-oss-120b": 4,
 
+    "gpt-5": 8,
+    "gpt-5-2025-08-07": 8,
+    "gpt-5-chat-latest": 8,
+    "gpt-5-mini": 8,
+    "gpt-5-mini-2025-08-07": 8,
+    "gpt-5-nano": 4,
+    "gpt-5-nano-2025-08-07": 4,
+
     # claude
     "claude-3-5-haiku-20241022": 5,
     "anthropic/claude-3-5-haiku-20241022:beta": 5,
@@ -1273,6 +1288,7 @@ COMPLETION_RATIO = {
     "claude-sonnet-4-20250514-thinking": 5,
     "claude-opus-4-20250514": 5,
     "claude-opus-4-20250514-thinking": 5,
+    "claude-opus-4.1": 5,
 
     "llama-3.1-70b-instruct": 2,
     "meta-llama/Meta-Llama-3.1-70B-Instruct": 2,
@@ -1778,3 +1794,6 @@ if __name__ == '__main__':
 
     print("FAL按次")
     print(','.join(FAL_MODELS))  # fal 按次
+
+    print('\n\n')
+    print([k for k in MODEL_RATIO if k.startswith(('gpt-5'))] | xjoin(","))