MeUtils 2024.3.4.13.4.45__py3-none-any.whl → 2025.1.16.17.15.52__py3-none-any.whl

meutils/schemas/kimi_types.py

@@ -0,0 +1,86 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : kimi_types
+# @Time         : 2024/2/29 16:13
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+
+from meutils.pipe import *
+
+
+class KimiData(BaseModel):
+    """
+    {"error":{"error_type":"openai.completion.stream","message":"Kimi 现在有点累了，晚一点再来问问我吧！","detail":"runtime error: invalid memory address or nil pointer dereference"},"event":"error"}
+
+    data: {"content":"1+1","event":"req","group_id":"cljdett0tc12f8jcpc7g","id":"cljdett0tc12f8jcpc8g","refs":[]}
+    data: {"event":"resp","group_id":"cljdett0tc12f8jcpc7g","id":"cljdett0tc12f8jcpc90"}
+    data: {"event":"debug","message":{"need2search":false,"search_target":"","search_language":""}}
+    data: {"event":"debug","message":{"need2search":true,"search_target":"南京今天天气","search_language":"zh-cn"}}
+    data: {"event":"search_plus","msg":{"type":"start"}}
+    data: {"event":"search_plus","msg":{"successNum":1,"title":"南京天气","type":"get_res","url":"http://m.nmc.cn/publish/forecast/AJS/nanjing.html"}}
+        {'successNum': 1, 'title': '南京天气预报25天|南京未来25天天气|南京天气预报未来25天...', 'type': 'get_res', 'url': 'http://www.tqw1.com/jsnanjing_25.shtml'}
+    data: {"event":"cmpl","text":" "}
+    data: {"event":"cmpl","text":"1"}
+    data: {"event":"cmpl","text":"加"}
+    data: {"event":"done"}
+    data: {"event":"cmpl","text":"1"}
+    data: {"event":"cmpl","text":"等"}
+    data: {"event":"cmpl","text":"于"}
+    data: {"event":"cmpl","text":"2"}
+    data: {"event":"cmpl","text":"。"}
+    data: {"event":"all_done"}
+
+    {"error":{"error_type":"openai.completion.token_length_too_long","message":"转眼间，你和 Kimi 的这个对话已经超过了 20 万字。Kimi 在不断提升自己对话最大长度，但现在只能麻烦你开启一个新会话。期待与你再相遇！"},"event":"error"}
+    """
+    event: str = 'cmpl'  # 事件类型 ping
+
+    prompt: str = Field(default='', alias='content')
+    content: str = Field(default='', alias='text')  # 生成的文本
+
+    # id: Optional[str] = None
+    # group_id: Optional[str] = None
+
+    msg: dict = {}
+    refs: Optional[List[str]] = None  # 文件问答
+
+    # {"error_type":"auth.token.invalid","message":"您的授权已过期，请重新登录"}
+    # {"error_type":"chat.forbidden","message":"此会话已被禁用"}
+    error_type: Optional[str] = None
+    message: Optional[Any] = None
+
+    # todo
+    # markdown
+
+
+class Data(BaseModel):
+    event: str = 'cmpl'  # 事件类型 ping
+
+    prompt: str = Field(default='', alias='content')
+    content: str = Field(default='', alias='text')  # 生成的文本
+
+    # id: Optional[str] = None
+    # group_id: Optional[str] = None
+
+    msg: dict = {}
+    refs: Optional[List[str]] = None  # 文件问答
+
+    # {"error_type":"auth.token.invalid","message":"您的授权已过期，请重新登录"}
+    # {"error_type":"chat.forbidden","message":"此会话已被禁用"}
+    error_type: Optional[str] = None
+    message: Optional[Any] = None
+
+    # chunk
+    markdown_data: str = ''
+
+    def __init__(self, **data):
+        super().__init__(**data)
+
+        if self.event == 'search_plus':
+            msg_type = self.msg.get("type")
+            if msg_type:  # start_res
+                title = self.msg.get("title")
+                url = self.msg.get("url")
+                self.markdown_data = f"""\n```{msg_type}\n{self.msg}\n```\n"""  # f"""- [🔗{title}]({url})\n"""

meutils/schemas/kling_types.py

@@ -0,0 +1,235 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : kling_types
+# @Time         : 2024/9/24 10:15
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+from meutils.pipe import *
+from meutils.notice.feishu import send_message as _send_message
+
+send_message = partial(
+    _send_message,
+    title=__name__,
+    url="https://open.feishu.cn/open-apis/bot/v2/hook/ad8c3010-7706-4ba0-b945-645c492e535b"
+)
+
+STATUSES = {
+    5: "submitted",
+    10: "processing",
+    99: "succeed",
+    50: "failed",
+
+    6: "The uploaded image size should be greater than 300PX",
+    7: 'NSFW：输入的提示词包含敏感词',
+    8: "NSFW：上传图片包含敏感信息",  # 上传图片包含敏感信息
+
+    400: "非法请求",
+    500: "内部系统繁忙",
+
+    "submitted": "SUBMITTED",
+    "processing": "IN_PROGRESS",
+    "succeed": "SUCCESS",
+    "failed": "FAILURE",
+
+}
+
+API_BASE_URL = "https://api.klingai.com"
+API_FEISHU_URL = "https://xchatllm.feishu.cn/sheets/GYCHsvI4qhnDPNtI4VPcdw2knEd?sheet=LGVKwN"
+
+
+class TryOnRequest(BaseModel):
+    model_name: str = "kolors-virtual-try-on-v1"
+    """
+    ● 支持传入图片Base64编码或图片URL（确保可访问）
+    ● 图片格式支持.jpg / .jpeg / .png
+    ● 图片文件大小不能超过10MB，图片分辨率不小于300*300px
+    """
+    human_image: str = "https://oss.ffire.cc/files/try-on.png"
+    cloth_image: str = "https://oss.ffire.cc/files/x.png"
+
+    callback_url: Optional[str] = None
+
+    class Config:
+        frozen = True
+
+
+class Config(BaseModel):
+    """包含六个字段，用于指定摄像机的运动或变化"""
+    """水平运镜，可选，取值范围：[-10, 10]"""
+    horizontal: Optional[int] = 0
+    """水平摇镜，可选，取值范围：[-10, 10]"""
+    pan: Optional[int] = 0
+    """旋转运镜，可选，取值范围：[-10, 10]"""
+    roll: Optional[int] = 0
+    """垂直摇镜，可选，取值范围：[-10, 10]"""
+    tilt: Optional[int] = 0
+    """垂直运镜，可选，取值范围：[-10, 10]"""
+    vertical: Optional[int] = 0
+    """变焦，可选，取值范围：[-10, 10]"""
+    zoom: Optional[int] = 0
+
+
+class CameraControl(BaseModel):
+    """控制摄像机运动的协议，可选，未指定则智能匹配"""
+    """包含六个字段，用于指定摄像机的运动或变化"""
+    config: Config = Config()
+    type: str = "simple"  # empty
+
+
+class VideoRequest(BaseModel):
+    model: Optional[str] = "kling-v1"
+
+    """生成视频的模式，可选，枚举值：std（高性能）或 pro（高表现）"""
+    mode: Optional[str] = "std"
+
+    """正向文本提示，必须，不能超过500个字符"""
+    prompt: str
+
+    """生成视频的画面纵横比，可选，枚举值：16:9, 9:16, 1:1"""
+    aspect_ratio: Optional[str] = "1:1"
+
+    """控制摄像机运动的协议，可选，未指定则智能匹配"""
+    camera_control: CameraControl = CameraControl()
+
+    """生成视频的自由度，可选，值越大，相关性越强，取值范围：[0,1]"""
+    cfg_scale: Optional[float] = None
+
+    """生成视频时长，单位秒，可选，枚举值：5，10"""
+    duration: Optional[int] = 5
+
+    """负向文本提示，可选，不能超过200个字符"""
+    negative_prompt: Optional[str] = None
+
+    # 图生视频多出来的参数
+    """参考图像，必须，支持Base64编码或图片URL，支持.jpg / .jpeg / .png格式，大小不能超过10MB，分辨率不小于300*300px"""
+    image: Optional[str] = None
+
+    """参考图像 - 尾帧控制，可选，支持Base64编码或图片URL，支持.jpg / .jpeg / .png格式，大小不能超过10MB，分辨率不小于300*300px"""
+    image_tail: Optional[str] = None
+
+    """生成视频数量，可选，取值范围：[1,4]"""
+    n: int = 1
+
+
+class ImageRequest(BaseModel):
+    model: Optional[str] = None
+
+    """正向文本提示，必须，不能超过500个字符"""
+    prompt: str
+    """生成图片的纵横比，可选，枚举值：16:9, 9:16, 1:1, 4:3, 3:4, 3:2, 2:3"""
+    aspect_ratio: Optional[str] = None
+
+    """生成图片数量，可选，取值范围：[1,9]"""
+    n: int = 1
+
+    """参考图片，可选，支持Base64编码或图片URL，支持.jpg / .jpeg / .png格式，大小不能超过10MB，图片分辨率不小于300*300px。
+    Base64仅提供编码部分，data:image/png;base64,后面的部分。
+    """
+    image: Optional[str] = None
+
+    """生成过程中对用户上传图片的参考强度，可选，取值范围：[0,1]"""
+    image_fidelity: Optional[float] = None
+
+    """负向文本提示，可选，不能超过200个字符"""
+    negative_prompt: Optional[str] = None
+
+    """本次任务结果回调通知地址，可选"""
+    callback_url: Optional[str] = None
+
+
+# {
+#     "code": 0,
+#     "message": "string",
+#     "request_id": "string",
+#     "data": {
+#         "task_id": "string",
+#         "task_status": "string",
+#         "created_at": 0,
+#         "updated_at": 0
+#     }
+# }
+
+class TaskResult(BaseModel):
+    """视频ID，系统生成"""
+    id: str
+
+    """视频URL"""
+    url: str
+
+    """视频时长，单位ms"""
+    duration: int
+
+
+class Task(BaseModel):
+    """任务ID，系统生成"""
+    task_id: str
+
+    """任务状态，枚举值：submitted（已提交）、processing（处理中）、succeed（成功）、failed（失败）"""
+    task_status: str = "processing"
+
+    """任务状态信息，具体定义状态信息"""
+    task_status_msg: Optional[str] = None
+
+    """任务创建时间，Unix时间戳、单位ms"""
+    created_at: int = Field(default_factory=lambda: int(time.time() * 1000))
+
+    """任务更新时间，Unix时间戳、单位ms"""
+    updated_at: int = Field(default_factory=lambda: int(time.time() * 1000))
+
+    """任务最终结果"""
+    task_result: Optional[dict] = None
+
+
+class TaskResponse(BaseModel):
+    code: int = 0
+    message: str = ""
+    request_id: str = ""
+
+    data: Optional[Task] = None
+
+    # 系统水印
+    system_fingerprint: Optional[str] = None
+
+
+# {
+#     "code": 0,
+#     "message": "string",
+#     "request_id": "string",
+#     "data": {
+#         "task_id": "string",
+#         "task_status": "string",
+#         "created_at": 0,
+#         "updated_at": 0
+#     }
+# }
+
+
+# {
+#     "code": 0,
+#     "message": "string",
+#     "request_id": "string",
+#     "data": {
+#         "task_id": "string",
+#         "task_status": "string",
+#         "task_status_msg": "string",
+#         "created_at": 1722769557708,
+#         "updated_at": 1722769557708,
+#         "task_result": {
+#             "videos": [
+#                 {
+#                     "id": "string",
+#                     "url": "string",
+#                     "duration": "string"
+#                 }
+#             ]
+#         }
+#     }
+# }
+
+
+if __name__ == '__main__':
+    print(TaskResponse(data=Task(task_id='xx')).model_dump(exclude_none=True))

meutils/schemas/kuaishou_types.py

@@ -0,0 +1,328 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : kuaishou_types
+# @Time         : 2024/7/9 13:26
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+from meutils.pipe import *
+
+# oss
+"https://s2-111386.kwimgs.com/bs2/mmu-kolors-public/5f9c7b42688fe95c4b8d9ebd1dba3431.png?x-oss-process=image/resize,m_mfit,w_305"
+
+# BASE_URL = "https://klingai.kuaishou.com"
+# UPLOAD_BASE_URL = "https://upload.kuaishouzt.com"
+# DOWNLOAD_URL= "https://klingai.kuaishou.com/api/works/batch_download_v2"
+
+BASE_URL = "https://klingai.com"
+UPLOAD_BASE_URL = "https://upload.uvfuns.com"
+DOWNLOAD_URL = "https://klingai.com/api/works/batch_download_v2"  # /api/works/batch_download_v2?workIds=100331495&fwm=false
+
+FEISHU_URL = "https://xchatllm.feishu.cn/sheets/Bmjtst2f6hfMqFttbhLcdfRJnNf?sheet=XuxPYK"
+FEISHU_URL_VIP = "https://xchatllm.feishu.cn/sheets/GYCHsvI4qhnDPNtI4VPcdw2knEd?sheet=hw8MXS"
+
+ASPECT_RATIOS = {
+    "1:1": "1024x1024",
+
+    "16:9": "1344x768",
+    "9:16": "768x1344",
+
+    "4:3": "1152x896",
+    "3:4": "896x1152",
+
+    "3:2": "1216x832",
+    "2:3": "832x1216",
+
+}
+
+STYLES = {
+    "默认", "皮克斯", "卡通盲盒", "新海诚", "动漫3D", "怀旧动漫", "电子游戏", "水彩插画", "莫奈油画", "高清写实"
+}
+
+STATES = {
+    5: "queue",
+    10: "processing",
+    99: "completed",
+    50: "failed",
+}
+
+
+class KolorsRequest(BaseModel):
+    prompt: str = '多肉植物，带着水珠，潮玩盲盒风格，皮克斯，3D质感，温馨的环境，丰富的场景，最佳画质，超精细，Octane渲染'
+    negativePrompt: str = ''  # 不希望出现的内容
+
+    style: str = "默认"
+    imageCount: int = 1
+    resolution: str = "1024x1024"
+
+    #
+    aspect_ratio: Literal["1:1", "2:3", "3:2", "3:4", "4:3", "9:16", "16:9"] = "1:1"
+
+    # 垫图
+    referImage: str = ''
+    fidelity: float = 0.5  # 参考度
+
+    cfgScale: float = 7
+    strength: float = 0.8
+
+    def __init__(self, /, **data: Any):
+        super().__init__(**data)
+        if self.style not in STYLES:
+            self.style = "默认"
+
+        if self.resolution not in set(ASPECT_RATIOS.values()):
+            self.resolution = ASPECT_RATIOS.get(self.aspect_ratio, "1024x1024")
+
+        if self.referImage and self.referImage.startswith("http"):
+            self.imageCount = 4
+            self.referImage = self.referImage.split("/")[-2].split('?')[0]
+
+
+class KlingaiImageRequest(BaseModel):
+    mode: str = "pro"
+
+    prompt: str = '清凉夏季美少女，微卷短发，运动服，林间石板路，斑驳光影，超级真实，16K'
+    style: str = "默认"
+    aspect_ratio: Optional[Literal["1:1", "2:3", "3:2", "3:4", "4:3", "9:16", "16:9"]] = "1:1"
+
+    n: int = 1
+
+    url: Optional[str] = None
+
+    image_fidelity: Optional[float] = None
+
+    #
+    payload: dict = {}
+
+    def __init__(self, /, **data: Any):
+        super().__init__(**data)
+        inputs = []
+        task_type = "mmu_txt2img_aiweb"
+        if self.url:
+            task_type = "mmu_img2img_aiweb"
+            inputs.append({'name': 'input', 'inputType': 'URL', 'url': self.url})
+            # inputs[-1]["fromWorkId"] = self.url.split('/')[-3].split('_')[-1]
+
+        arguments = [
+            {'name': 'prompt', 'value': self.prompt},
+            {'name': 'style', 'value': self.style},
+            {'name': 'aspect_ratio', 'value': self.aspect_ratio or "1:1"},
+            {'name': 'imageCount', 'value': self.n},
+            {"name": "fidelity", "value": self.image_fidelity or 0.5},
+            {'name': 'biz', 'value': 'klingai'},
+            {"name": "kolors_version", "value": "1.0"} ####### 1.5
+        ]
+        self.payload = {
+            'arguments': arguments,
+            'type': task_type,
+            'inputs': inputs
+        }
+
+
+class Camera(BaseModel):
+    type: Literal[
+        "empty", "horizontal", "vertical", "zoom", "tilt", "pan", "roll", "down_back",
+        "simple", "down_back", "forward_up", "left_turn_forward", "right_turn_forward",
+    ] = "empty"
+    horizontal: Optional[int] = 0
+    vertical: Optional[int] = 0
+    zoom: Optional[int] = 0
+    tilt: Optional[int] = 0
+    pan: Optional[int] = 0
+    roll: Optional[int] = 0
+    # """水平运镜，可选，取值范围：[-10, 10]"""
+    # horizontal: Optional[int] = None
+    # """水平摇镜，可选，取值范围：[-10, 10]"""
+    # pan: Optional[int] = None
+    # """旋转运镜，可选，取值范围：[-10, 10]"""
+    # roll: Optional[int] = None
+    # """垂直摇镜，可选，取值范围：[-10, 10]"""
+    # tilt: Optional[int] = None
+    # """垂直运镜，可选，取值范围：[-10, 10]"""
+    # vertical: Optional[int] = None
+    # """变焦，可选，取值范围：[-10, 10]"""
+    # zoom: Optional[int] = None
+
+    camera_json: str = ""  # 运镜
+
+    def __init__(self, /, **data: Any):
+        super().__init__(**data)
+        if self.type == "down_back":
+            self.vertical = self.vertical or -10
+            self.zoom = self.zoom or -10
+
+        if self.type == "SIMPLE":
+            self.type = "empty"
+        # elif hasattr(self, self.type):
+        #     self.__setattr__(self.type, self.__getattr__(self.type))
+
+        self.camera_json = json.dumps(self.dict(exclude={'camera_json'}))
+
+
+class KlingaiVideoRequest(BaseModel):
+    """增加计费模式，支持高质量模式
+    kling-v1.0-std-5s
+
+    kling-v1.5-std-5s 20积分
+    kling-v1.5-std-10s
+
+    kling-v1.5-pro-5s 35积分
+    kling-v1.5-pro-10s
+
+    kling-v1.6-std-5s 20积分
+    kling-v1.6-std-10s
+
+    kling-v1.6-pro-5s 35积分
+    kling-v1.6-pro-10s
+
+    """
+    model: str = 'kling-v1.6'
+    mode: Literal["mini", "std", "pro"] = "std"
+
+    prompt: str = ''
+    negative_prompt: Optional[str] = ''  # 不希望出现的内容
+
+    n: Literal[1, 2, 3, 4] = 1
+
+    duration: Optional[int] = 5
+    aspect_ratio: Literal["1:1", "2:3", "3:2", "3:4", "4:3", "9:16", "16:9"] = "16:9"
+    cfg_scale: Optional[float] = 0.5  # 创意程度
+    camera: Camera = Camera()  # 运镜
+
+    url: Optional[str] = ''
+    # 图生视频
+    tail_image_url: Optional[str] = ''
+
+    #
+    payload: dict = {}
+
+    def __init__(self, /, **data: Any):
+        super().__init__(**data)
+
+        if self.duration > 5:
+            self.duration = 10
+        else:
+            self.duration = 5
+
+        if self.tail_image_url or self.mode == 'mini':
+            self.duration = 5  # 尾帧仅支持 5s
+
+        arguments = [
+            {'name': 'prompt', 'value': self.prompt},
+            {'name': 'negative_prompt', 'value': self.negative_prompt},
+            {'name': 'duration', 'value': self.duration},
+            {'name': 'aspect_ratio', 'value': self.aspect_ratio},
+            {"name": "imageCount", "value": self.n},
+            {'name': 'cfg', 'value': self.cfg_scale},
+            {'name': 'camera_json', 'value': self.camera.camera_json},
+
+            {'name': 'biz', 'value': 'klingai'},
+        ]
+
+        kling_version = '1.5'
+        if self.model == 'kling-v1.6':
+            kling_version = "1.6"
+            arguments.append({"name": "camera_control_enabled", "value": "false"})  # 暂不支持
+        else:
+            arguments.append({"name": "camera_control_enabled", "value": "true"})
+
+        arguments.append({"name": "kling_version", "value": kling_version})
+
+        inputs = []
+        task_type = "m2v_txt2video"
+        if self.url:
+            task_type = "m2v_img2video"
+            inputs.append({'name': 'input', 'inputType': 'URL', 'url': self.url})
+            if self.url.endswith(".mp4"):  # 待测试
+                inputs[-1]["fromWorkId"] = self.url.split('/')[-3].split('_')[-1]
+                task_type = "m2v_extend_video"
+
+        if self.tail_image_url:
+            task_type = "m2v_img2video"  # 尾帧 高质量模式
+
+            arguments.append({"name": "tail_image_enabled", "value": "true"})
+            inputs.append({'name': 'tail_image', 'inputType': 'URL', 'url': self.tail_image_url})
+
+        self.payload = {
+            'arguments': arguments,
+            'type': f"""{task_type}{'_hq' if self.mode == 'pro' else ''}""",
+            'inputs': inputs
+        }
+
+    class Config:
+        json_schema_extra = {
+            "examples": [
+                {
+                    "prompt": "一只可爱的黑白边境牧羊犬，头伸出车窗，毛发被风吹动，微笑着伸出舌头",
+
+                    "url": "",
+                    "tail_image_url": "",
+
+                    "negative_prompt": "",
+                    "duration": 5,
+                    "aspect_ratio": "16:9",
+                    "cfg": 0.5,
+                    "camera": {
+                        "type": "empty",
+                        "horizontal": 0,
+                        "vertical": 0,
+                        "zoom": 0,
+                        "tilt": 0,
+                        "pan": 0,
+                        "roll": 0
+                    }
+                },
+
+                {
+                    "prompt": "一只可爱的黑白边境牧羊犬，头伸出车窗，毛发被风吹动，微笑着伸出舌头",
+
+                    "url": "https://p2.a.kwimgs.com/bs2/upload-ylab-stunt/ai_portal/1720681052/LZcEugmjm4/whqrbrlhpjcfofjfywqqp9.png",
+                    "tail_image_url": "",
+
+                    "negative_prompt": "",
+                    "duration": 5,
+                    "aspect_ratio": "16:9",
+                    "cfg": 0.5,
+                    "camera": {
+                        "type": "empty",
+                        "horizontal": 0,
+                        "vertical": 0,
+                        "zoom": 0,
+                        "tilt": 0,
+                        "pan": 0,
+                        "roll": 0
+                    }
+                },
+
+                {
+                    "prompt": "开花过程",
+
+                    "url": "https://h2.inkwai.com/bs2/upload-ylab-stunt/ai_portal/1721561429/3z7bHmSC1Y/3yrmpqh7typspfcchmepar.png",
+                    "tail_image_url": "https://p2.a.kwimgs.com/bs2/upload-ylab-stunt/ai_portal/1721561524/GPoJWxBS8s/y9wlbuku3exeo7ra85s4su.png",
+
+                    "negative_prompt": "",
+                    "duration": 5,
+                    "aspect_ratio": "16:9",
+                    "cfg": 0.5,
+                    "camera": {
+                        "type": "empty",
+                        "horizontal": 0,
+                        "vertical": 0,
+                        "zoom": 0,
+                        "tilt": 0,
+                        "pan": 0,
+                        "roll": 0
+                    }
+                }
+
+            ]
+        }
+
+
+"https://p2.a.kwimgs.com/bs2/upload-ylab-stunt/special-effect/output/HB1_PROD_ai_web_30121850/8784600431869866912/out.mp4"
+if __name__ == '__main__':
+    # print(Camera().camera_json)
+    print(KlingaiVideoRequest.Config.json_schema_extra.get('examples')[-1])