增加大模型

This commit is contained in:
dekun
2026-05-26 09:49:43 +08:00
parent 27031ab676
commit 86aa804c21
8 changed files with 547 additions and 90 deletions
+16 -4
View File
@@ -18,7 +18,7 @@ from .llm_service import get_interpret_state, run_interpretation_batch
from .scheduler import job_finalize_yesterday, job_push_wecom, job_refresh_today, start_scheduler, startup_tasks, stop_scheduler
from .stats import compute_three_day_stats
from .aggregator import aggregate_period
from .wecom import build_markdown, send_wecom_markdown
from .wecom import build_markdown, build_push_payload, send_wecom_markdown
from .state import get_today_cache
logging.basicConfig(
@@ -93,8 +93,17 @@ async def api_stats_three_day():
return compute_three_day_stats()
@app.get("/api/push/preview")
async def api_push_preview():
"""预览企微推送内容(三日交集,列表排版)。"""
return build_push_payload()
@app.post("/api/push/test")
async def api_push_test():
payload = build_push_payload()
if not payload.get("ok"):
raise HTTPException(400, payload.get("message") or "三日交集数据未就绪")
snap = get_latest_snapshot("yesterday")
if not snap:
start, end = get_yesterday_period()
@@ -104,12 +113,15 @@ async def api_push_test():
snap = get_latest_snapshot("yesterday")
if not snap:
raise HTTPException(500, "无法生成昨日数据")
content = build_markdown(snap)
ok, msg = await send_wecom_markdown(content)
ok, msg = await send_wecom_markdown(payload["markdown"])
log_push(snap["period_start"], snap["period_end"], ok, msg)
if not ok:
raise HTTPException(500, f"推送失败: {msg}")
return {"success": True, "message": "推送成功"}
return {
"success": True,
"message": f"已推送 {payload.get('count', 0)} 个三日交集币种",
"count": payload.get("count", 0),
}
@app.post("/api/refresh/yesterday")