SupplyAgent/supply_infra/db/models/__init__.py

"""ORM entity models — one file per table."""

from supply_infra.db.models.agent_document_injection import AgentDocumentInjection
from supply_infra.db.models.category_tree_weight import CategoryTreeWeight
from supply_infra.db.models.demand_belong_category import DemandBelongCategory
from supply_infra.db.models.demand_belong_pool_rel import DemandBelongPoolRel
from supply_infra.db.models.demand_grade import DemandGrade
from supply_infra.db.models.demand_grade_category_rel import DemandGradeCategoryRel
from supply_infra.db.models.demand_grade_plan import (
    DemandGradePlan,
    DemandGradePlanGroup,
    DemandGradePlanGroupItem,
)
from supply_infra.db.models.demand_popularity_stats import DemandPopularityStats
from supply_infra.db.models.demand_video_expansion import (
    DemandVideoExpansion,
    DemandVideoExpansionRun,
)
from supply_infra.db.models.generated_demand import GeneratedDemand
from supply_infra.db.models.global_tree_category import GlobalTreeCategory
from supply_infra.db.models.global_tree_element import GlobalTreeElement
from supply_infra.db.models.multi_demand_pool_di import MultiDemandPoolDi
from supply_infra.db.models.multi_demand_video_detail import MultiDemandVideoDetail
from supply_infra.db.models.multi_demand_video_point import MultiDemandVideoPoint
from supply_infra.db.models.oss_log import OssLog
from supply_infra.db.models.pipeline_lock import PipelineLock
from supply_infra.db.models.pipeline_outbox import PipelineOutbox
from supply_infra.db.models.pipeline_run import PipelineRun
from supply_infra.db.models.pipeline_step_run import PipelineStepRun
from supply_infra.db.models.video_discovery import (
    VideoDiscoveryCandidate,
    VideoDiscoveryRun,
    VideoDiscoverySearch,
)

__all__ = [
    "AgentDocumentInjection",
    "CategoryTreeWeight",
    "DemandBelongCategory",
    "DemandBelongPoolRel",
    "DemandGrade",
    "DemandGradeCategoryRel",
    "DemandGradePlan",
    "DemandGradePlanGroup",
    "DemandGradePlanGroupItem",
    "DemandPopularityStats",
    "DemandVideoExpansion",
    "DemandVideoExpansionRun",
    "GeneratedDemand",
    "GlobalTreeCategory",
    "GlobalTreeElement",
    "MultiDemandPoolDi",
    "MultiDemandVideoDetail",
    "MultiDemandVideoPoint",
    "OssLog",
    "PipelineLock",
    "PipelineOutbox",
    "PipelineRun",
    "PipelineStepRun",
    "VideoDiscoveryCandidate",
    "VideoDiscoveryRun",
    "VideoDiscoverySearch",
]