diff --git a/app/summary/task.py b/app/summary/task.py index 7a5349a..3197cc7 100644 --- a/app/summary/task.py +++ b/app/summary/task.py @@ -9,36 +9,25 @@ from app.client.milvus_client_manager import milvus_client from app.client.mysql_client_manager import db_assistant_mysql_client_manager from app.repository.archive_messages_repository import ArchiveMessagesRepository from app.repository.milvus.summary_repository import SummaryRepository -from app.summary.service import SummaryService +from app.summary.service import SummaryService, build from app.core.log import logger app = FastAPI(title="消息摘要服务", version="1.0") scheduler = AsyncIOScheduler(timezone='Asia/Shanghai') async def generate_daily_summary(date: Optional[str] = None): + logger.info("16:30分定时任务测试") if not date: yesterday = datetime.now() - timedelta(days=1) - date = yesterday.strftime("%Y-%-m-%-d") # 格式: 2026-6-15 + #date = yesterday.strftime("%Y-%-m-%-d") # 格式: 2026-6-15 + date = datetime(yesterday.year, yesterday.month, yesterday.day) - logger.info(f"开始生成 {date} 的消息摘要...") + # 按 年-月-日 无补零格式输出 + date_format = f"{date.year}-{date.month}-{date.day}" - try: - db_assistant_mysql_client_manager.init() - embedding_client.init() - milvus_client.init() - - async with db_assistant_mysql_client_manager.session_factory() as db_assistant: - archive_messages_repository = ArchiveMessagesRepository(db_assistant) - summary_repository = SummaryRepository() - service = SummaryService(archive_messages_repository, embedding_client, summary_repository) - await service.get_message_summary(date) + await build(date_format) logger.info(f"{date} 消息摘要生成完成") - except Exception as e: - logger.error(f"生成摘要失败: {str(e)}", exc_info=True) - raise - finally: - await embedding_client.close() @app.on_event("startup") async def startup_event():