bvandeusen
0a4eb0bdc0
Base.metadata's convention applies ck_%(table_name)s_%(constraint_name)s.
ml_settings and tag_allowlist passed already-prefixed names
(ck_ml_settings_singleton / ck_tag_allowlist_confidence_range), so the
ORM-side names came out doubled (ck_ml_settings_ck_ml_settings_singleton
etc.) and the migration-0003 smoke tests failed.
Same class of bug fixed in FC-2a for ImportSettings — should have applied
that lesson here. Bare names ('singleton', 'confidence_range') let the
convention produce the final names that match migration 0003's literal
DDL. Migration unchanged; only the model __table_args__.
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
"""MLSettings — single-row table holding ML pipeline tunables."""
|
|
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import CheckConstraint, DateTime, Float, Integer, String, func
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from .base import Base
|
|
|
|
|
|
class MLSettings(Base):
|
|
__tablename__ = "ml_settings"
|
|
# Bare name — Base.metadata's naming convention prepends ck_<table>_,
|
|
# producing the final ck_ml_settings_singleton (matches migration 0003).
|
|
__table_args__ = (CheckConstraint("id = 1", name="singleton"),)
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
|
suggestion_threshold_artist: Mapped[float] = mapped_column(
|
|
Float, nullable=False, default=0.30
|
|
)
|
|
suggestion_threshold_character: Mapped[float] = mapped_column(
|
|
Float, nullable=False, default=0.50
|
|
)
|
|
suggestion_threshold_copyright: Mapped[float] = mapped_column(
|
|
Float, nullable=False, default=0.50
|
|
)
|
|
suggestion_threshold_general: Mapped[float] = mapped_column(
|
|
Float, nullable=False, default=0.95
|
|
)
|
|
centroid_similarity_threshold: Mapped[float] = mapped_column(
|
|
Float, nullable=False, default=0.55
|
|
)
|
|
min_reference_images: Mapped[int] = mapped_column(
|
|
Integer, nullable=False, default=5
|
|
)
|
|
tagger_model_version: Mapped[str] = mapped_column(
|
|
String(128), nullable=False, default="camie-tagger-v2"
|
|
)
|
|
embedder_model_version: Mapped[str] = mapped_column(
|
|
String(128), nullable=False, default="siglip-so400m-patch14-384"
|
|
)
|
|
updated_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True), nullable=False, server_default=func.now()
|
|
)
|