159 lines
6.4 KiB
Python
159 lines
6.4 KiB
Python
import datetime
|
||
import json
|
||
import re
|
||
|
||
from loguru import logger
|
||
|
||
from konabot.common.apis.ali_content_safety import AlibabaGreen
|
||
from konabot.common.llm import get_llm
|
||
|
||
|
||
SYSTEM_PROMPT = """你是一个专门解析提醒请求的助手。请分析用户输入，识别其中是否包含提醒信息，并输出标准化的JSON格式结果。
|
||
|
||
输入格式通常是："xxxx提醒我yyyy"，其中：
|
||
- xxxx 是用户提供的时间信息
|
||
- yyyy 是提醒内容。有些时候用户会有一些需求。你可以在合理的范围进行衍生。
|
||
|
||
输出要求：
|
||
- 必须是有效的JSON对象
|
||
- 包含以下字段：
|
||
* datetime: 如果是绝对时间，填入ISO 8601格式的日期时间字符串；否则为null
|
||
* datetime_delta: 如果是相对时间，填入ISO 8601持续时间格式；否则为null
|
||
* datetime_delta_minus: 如果时间偏移量是负数，则此项为 true，否则为 false
|
||
* content: 提醒内容的字符串
|
||
* is_notice: 布尔值，表示这是否是真正的提醒请求
|
||
|
||
时间处理规则：
|
||
- 绝对时间示例：如果 xxxx 输入了非常明确的时间点，如"2024年12月25日" → 转换为具体datetime
|
||
- 相对时间示例：如果 xxxx 没有输入非常明确的时间点，如"10分钟后"、"2小时后"、"3天后" → 转换为datetime_delta
|
||
- 如果用户输入了需要计算的时间，你需要计算出正确的结果，如"10分钟后的8分钟前" → 转换为 “PT2M”
|
||
- zzzz 是系统提供的时间，每句话肯定都有，这不是你判断相对或绝对时间的依据，需严格按照 xxx 来判断
|
||
- datetime和datetime_delta有且仅有一个不为null
|
||
|
||
时间格式要求：
|
||
- datetime: "YYYY-MM-DDTHH:MM:SS" (ISO 8601)
|
||
- datetime_delta: "PxYxMxDTxHxMxS" 格式 (如"PT1H30M"表示1小时30分钟，"P3DT4H"表示三天四小时，"P5MT2M"表示五个月两分钟)
|
||
|
||
判断标准：
|
||
- is_notice=true: 明确包含时间+提醒内容的请求
|
||
- is_notice=false: 闲聊、疑问句、或不符合提醒格式的内容
|
||
|
||
示例：
|
||
用户："明天下午2点提醒我开会"
|
||
输出：{"datetime": "2024-01-16T14:00:00", "datetime_delta": null,
|
||
"datetime_delta_minus": false, "content": "开会", "is_notice": true}
|
||
|
||
用户："5分钟后提醒我关火"
|
||
输出：{"datetime": null, "datetime_delta": "PT5M", "datetime_delta_minus": false, "content": "关火", "is_notice": true}
|
||
|
||
用户："5分钟前提醒我关火"
|
||
输出：{"datetime": null, "datetime_delta": "PT5M", "datetime_delta_minus": true, "content": "关火", "is_notice": true}
|
||
|
||
用户："五百年后提醒我关火"
|
||
输出：{"datetime": null, "datetime_delta": "P500Y", "datetime_delta_minus": false, "content": "关火", "is_notice": true}
|
||
|
||
用户："昨天提醒我关火"
|
||
输出：{"datetime": null, "datetime_delta": "P1D", "datetime_delta_minus": true, "content": "关火", "is_notice": true}
|
||
|
||
用户："什么是提醒功能？"
|
||
输出：{"datetime": null, "datetime_delta": null, "datetime_delta_minus": false, "content": "", "is_notice": false}
|
||
|
||
用户："过一会会，用可爱的语气提醒我该睡觉了"
|
||
输出：{"datetime": null, "datetime_delta": "PT10M", "datetime_delta_minus": false, "content": "呼呼！该睡觉了哦！ヾ(•ω•`)o", "is_notice": true}
|
||
|
||
请严格按照上述格式输出JSON，不要添加任何其他文字说明。现在是 DATETIME"""
|
||
|
||
pt_pattern = re.compile(
|
||
r"^P"
|
||
r"((?P<year>\d+)Y)?"
|
||
r"((?P<month>\d+)M)?"
|
||
r"((?P<day>\d+)D)?"
|
||
r"(T((?P<hour>\d+)H)?"
|
||
r"((?P<minute>\d+)M)?"
|
||
r"((?P<second>\d+)S)?)?$"
|
||
)
|
||
|
||
|
||
def tryint(s: str | None):
|
||
if s:
|
||
if re.match(r"^\d+$", s):
|
||
return int(s)
|
||
return 0
|
||
|
||
|
||
async def ask_ai(expression: str, now: datetime.datetime | None = None) -> tuple[datetime.datetime | None, str]:
|
||
if now is None:
|
||
now = datetime.datetime.now()
|
||
prompt = SYSTEM_PROMPT.replace("DATETIME", f"{now}, 星期 {now.weekday() + 1}")
|
||
|
||
is_safe = await AlibabaGreen.detect(expression)
|
||
if not is_safe:
|
||
logger.info(f"提醒功能：消息被阿里绿网拦截 message={expression}")
|
||
return None, ""
|
||
|
||
llm = get_llm()
|
||
message = await llm.chat([
|
||
{ "role": "system", "content": prompt },
|
||
{ "role": "user", "content": expression },
|
||
])
|
||
result = message.content
|
||
if result is None:
|
||
return (None, "")
|
||
|
||
try:
|
||
data = json.loads(result)
|
||
except json.JSONDecodeError:
|
||
logger.info(f"提醒功能：解析 AI 返回值时出现问题 raw={result}")
|
||
return (None, "")
|
||
|
||
datetime_absolute = data.get("datetime", None)
|
||
datetime_delta = data.get("datetime_delta", None)
|
||
is_minus = data.get("datetime_delta_minus", False)
|
||
content = data.get("content", "")
|
||
is_notice = data.get("is_notice", False)
|
||
|
||
if not is_notice:
|
||
return (None, "")
|
||
if datetime_absolute:
|
||
try:
|
||
res = datetime.datetime.strptime(datetime_absolute, "%Y-%m-%dT%H:%M:%S"), content
|
||
logger.info(f"提醒功能：使用绝对时间解析 AI 返回值 raw={result} target={res[0]}")
|
||
return res
|
||
except ValueError:
|
||
pass
|
||
|
||
if datetime_delta and (match := pt_pattern.match(datetime_delta)):
|
||
years = tryint(match.group("year"))
|
||
months = tryint(match.group("month"))
|
||
days = tryint(match.group("day"))
|
||
hours = tryint(match.group("hour"))
|
||
minutes = tryint(match.group("minute"))
|
||
seconds = tryint(match.group("second"))
|
||
|
||
dt = datetime.timedelta(days=days, hours=hours, minutes=minutes, seconds=seconds)
|
||
if is_minus:
|
||
dt = -dt
|
||
if is_minus:
|
||
now2 = now.replace(year=now.year - years)
|
||
m = now2.month
|
||
if (months - m) >= 0:
|
||
neg_months = -(m - months - 1)
|
||
neg_years = (neg_months + 11) // 12
|
||
target_month = 12 - ((neg_months - 1) % 12)
|
||
now2 = now2.replace(year=now2.year - neg_years, month=target_month)
|
||
else:
|
||
now2 = now2.replace(month=m - months)
|
||
else:
|
||
now2 = now.replace(year=now.year + years)
|
||
m = now2.month
|
||
now2 = now2.replace(
|
||
year=now2.year + (m + months - 1) // 12,
|
||
month=(m + months - 1) % 12 + 1
|
||
)
|
||
logger.info(f"提醒功能：使用相对时间解析 AI 返回值 raw={result} target={now2+dt}")
|
||
return (now2 + dt, content)
|
||
|
||
logger.warning(f"提醒功能：解析 AI 返回值时没有找到解析方法 raw={result}")
|
||
return (None, "")
|
||
|