feat:现在日程表会根据节日生成东西了(gemini你再有问题我就砍了你)

This commit is contained in:
minecraft1024a
2025-08-22 16:22:42 +08:00
parent 943b21f533
commit e72b5064f5

View File

@@ -3,6 +3,7 @@ import asyncio
import random import random
from datetime import datetime, time, timedelta from datetime import datetime, time, timedelta
from typing import Optional, List, Dict, Any from typing import Optional, List, Dict, Any
from lunar_python import Lunar
from pydantic import BaseModel, ValidationError, validator from pydantic import BaseModel, ValidationError, validator
from src.common.database.sqlalchemy_models import Schedule, get_db_session from src.common.database.sqlalchemy_models import Schedule, get_db_session
@@ -175,6 +176,18 @@ class ScheduleManager:
current_month_str = now.strftime("%Y-%m") current_month_str = now.strftime("%Y-%m")
weekday = now.strftime("%A") weekday = now.strftime("%A")
# 新增:获取节日信息
lunar = Lunar.fromDate(now)
festivals = lunar.getFestivals()
other_festivals = lunar.getOtherFestivals()
all_festivals = festivals + other_festivals
festival_block = ""
if all_festivals:
festival_text = "".join(all_festivals)
festival_block = f"**今天也是一个特殊的日子: {festival_text}!请在日程中考虑和庆祝这个节日。**"
# 获取月度计划作为额外参考 # 获取月度计划作为额外参考
monthly_plans_block = "" monthly_plans_block = ""
used_plan_ids = [] used_plan_ids = []
@@ -198,7 +211,7 @@ class ScheduleManager:
prompt = f""" prompt = f"""
我,{global_config.bot.nickname},需要为自己规划一份今天({today_str},星期{weekday})的详细日程安排。 我,{global_config.bot.nickname},需要为自己规划一份今天({today_str},星期{weekday})的详细日程安排。
{festival_block}
**关于我**: **关于我**:
- **核心人设**: {personality} - **核心人设**: {personality}
- **具体习惯与兴趣**: - **具体习惯与兴趣**: