49 lines
2.6 KiB
Python
49 lines
2.6 KiB
Python
from sqlalchemy import Boolean, CheckConstraint, Integer, String, Text
|
||
from sqlalchemy.orm import Mapped, mapped_column
|
||
|
||
from app.models.base import Base, TimestampMixin, SoftDeleteMixin
|
||
|
||
|
||
class ImageEngine(Base, TimestampMixin, SoftDeleteMixin):
|
||
__tablename__ = "image_engines"
|
||
__table_args__ = (
|
||
CheckConstraint("max_generation_count BETWEEN 1 AND 5", name="ck_image_engines_max_generation_count"),
|
||
CheckConstraint("multi_image_max_images BETWEEN 1 AND 15", name="ck_image_engines_multi_image_max_images"),
|
||
CheckConstraint("max_reference_image_count BETWEEN 0 AND 14", name="ck_image_engines_max_reference_image_count"),
|
||
)
|
||
|
||
id: Mapped[str] = mapped_column(String(32), primary_key=True)
|
||
name: Mapped[str] = mapped_column(String(64), nullable=False)
|
||
provider: Mapped[str] = mapped_column(String(32), nullable=False)
|
||
api_base: Mapped[str] = mapped_column(String(512), nullable=False)
|
||
api_key: Mapped[str] = mapped_column(String(256), default="")
|
||
model_name: Mapped[str] = mapped_column(String(128), default="")
|
||
supported_models: Mapped[str] = mapped_column(Text, default='["doubao-seedream-5-0-260128"]')
|
||
# {"2K":{"1:1":"2048×2048",...}, "4K":{"1:1":"4096×4096",...}}
|
||
supported_sizes: Mapped[str] = mapped_column(Text, default='{}')
|
||
default_size: Mapped[str] = mapped_column(String(32), default="2K")
|
||
max_image_count: Mapped[int] = mapped_column(Integer, default=0)
|
||
|
||
# 管理后台只配置能力开关与数量上限；本次实际生成数量保存在 ChatGenerationTask.generation_count。
|
||
multi_generation_enabled: Mapped[bool] = mapped_column(
|
||
Boolean,
|
||
default=False,
|
||
server_default="false",
|
||
nullable=False,
|
||
)
|
||
max_generation_count: Mapped[int] = mapped_column(
|
||
Integer,
|
||
default=1,
|
||
server_default="1",
|
||
nullable=False,
|
||
)
|
||
|
||
# 火山组图接口能力约束。多份图片始终只调用一次 sequential_image_generation=auto 接口。
|
||
multi_image_max_images: Mapped[int] = mapped_column(Integer, default=15, server_default="15", nullable=False)
|
||
max_reference_image_count: Mapped[int] = mapped_column(Integer, default=14, server_default="14", nullable=False)
|
||
# 留空表示不向供应商传 output_format；用于兼容不支持该参数的模型。
|
||
output_format: Mapped[str] = mapped_column(String(16), default="", server_default="", nullable=False)
|
||
generate_url: Mapped[str | None] = mapped_column(String(512), nullable=True, default="")
|
||
is_active: Mapped[bool] = mapped_column(Boolean, default=True)
|
||
priority: Mapped[int] = mapped_column(Integer, default=0)
|