MeUtils 2025.2.20.17.30.7__py3-none-any.whl → 2025.2.25.18.30.35__py3-none-any.whl

meutils/schemas/suno_types.py

@@ -77,7 +77,7 @@ class SunoAIRequest(BaseModel):  # 原始请求体
 
     make_instrumental: bool = False
 
-    mv: str = "chirp-v3-5"  # chirp-v3-5-tau
+    mv: str = "chirp-v4"  # chirp-v3-5-tau
     generation_type: str = "TEXT"
 
     task: Optional[str] = None  # "cover"

meutils/schemas/task_types.py

@@ -105,6 +105,7 @@ class TaskType(str, Enum):
     file_extract = "file-extract"
     moonshot_fileparser = "moonshot-fileparser"
     textin_fileparser = "textin-fileparser"
+    qwen = "qwen"
 
     watermark_remove = "watermark-remove"
 

meutils/schemas/vidu_types.py

@@ -92,7 +92,8 @@ class VideoRequest(BaseModel):
 
 class ViduRequest(BaseModel):
     """quality 倍率2"""
-    model: Union[str, Literal['vidu-2.0', 'vidu-1.5', 'vidu-high-performance', 'vidu-high-quality']] = "vidu-high-performance"
+    model: Union[
+        str, Literal['vidu-2.0', 'vidu-1.5', 'vidu-high-performance', 'vidu-high-quality']] = "vidu-high-performance"
 
     prompt: Optional[str] = None
 
@@ -156,13 +157,12 @@ class ViduRequest(BaseModel):
                 }
             )
 
-
-        if self.model in ("vidu-1.5", "vidu-2.0"):
+        if self.model in ("vidu-2.0",):
             self.payload = {
                 "input": input,
                 "type": self.type or type,
                 "settings": {
-                    "model_version": "2.0", #######
+                    "model_version": "2.0",  #######
                     "style": "general",
                     "duration": self.duration,
 

meutils/serving/fastapi/dependencies/auth.py

@@ -14,6 +14,8 @@ from typing import Optional, Union
 from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
 from fastapi import Depends, HTTPException, status
 
+from meutils.config_utils.lark_utils import get_series, get_next_token
+
 http_bearer = HTTPBearer()
 
 
@@ -32,9 +34,13 @@ async def get_bearer_token(
 
     token = auth.credentials
     if token.startswith('redis:'):  # 初始化吧，太长？
-        pass
-        if ',' in token:  # 初始化redis
+        if "feishu.cn" in token:
+            feishu_url = token.removeprefix("redis:")
+            token = await get_next_token(feishu_url)  # 初始化redis
+
+        elif ',' in token:  # todo: 初始化redis
             pass
+
     elif ',' in token:  # 分隔符
         token = np.random.choice(token.split(','))
 

meutils/serving/fastapi/dependencies/headers.py

@@ -0,0 +1,31 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : headers
+# @Time         : 2025/2/23 00:20
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+
+from fastapi import FastAPI, Request, Depends, HTTPException
+from typing import Dict, Optional
+
+
+# 定义一个依赖函数来获取所有请求头
+# def get_headers(request: Request) -> Dict[str, str]:
+#     return dict(request.headers)
+
+def get_headers(request: Request):
+    return request.headers
+
+# lambda request: dict(request.headers)
+# @app.get("/headers/")
+# async def read_headers(headers: Dict[str, str] = Depends(get_headers)):
+#     # 在这里你可以使用 headers 字典
+#     if "upstream_api_key" not in headers:
+#         raise HTTPException(status_code=400, detail="API key is required")
+#
+#     # 返回所有请求头
+#     return {"headers": headers}