This commit is contained in:
Windpicker-owo
2025-09-01 21:13:00 +08:00

View File

@@ -7,7 +7,7 @@ import io
import asyncio import asyncio
import numpy as np import numpy as np
from typing import Optional, Tuple from typing import Optional, Tuple, Dict, Any
from PIL import Image from PIL import Image
from rich.traceback import install from rich.traceback import install
@@ -24,6 +24,22 @@ install(extra_lines=3)
logger = get_logger("chat_image") logger = get_logger("chat_image")
def is_image_message(message: Dict[str, Any]) -> bool:
"""
判断消息是否为图片消息
Args:
message: 消息字典
Returns:
bool: 是否为图片消息
"""
return message.get("type") == "image" or (
isinstance(message.get("content"), dict) and
message["content"].get("type") == "image"
)
class ImageManager: class ImageManager:
_instance = None _instance = None
IMAGE_DIR = "data" # 图像存储根目录 IMAGE_DIR = "data" # 图像存储根目录