Agent/examples/content_finder/db/__init__.py

"""
数据库封装模块

- connection: 共享数据库连接
- schedule: 定时任务相关（demand_content、demand_find_task）
- store_results: 推荐结果写入（demand_find_author、demand_find_content_result）
"""

from .connection import get_connection
from .open_aigc_pattern_connection import get_open_aigc_pattern_connection
from .schedule import (
    get_next_unprocessed_demand,
    create_task_record,
    update_task_status,
    update_task_on_complete,
)
from .store_results import upsert_good_authors, insert_contents, update_content_plan_ids

__all__ = [
    "get_connection",
    "get_open_aigc_pattern_connection",
    "get_next_unprocessed_demand",
    "create_task_record",
    "update_task_status",
    "update_task_on_complete",
    "upsert_good_authors",
    "insert_contents",
    "update_content_plan_ids",
]