feat(hub): enrich AI coach with fund history, closed trades, and chat uploads
- Add 15-day fund snapshot store and /api/hub/account on all instances - Summary includes yesterday/today trades, fund columns, and section 5 操作建议 - Chat context distinguishes empty positions from local monitors - Support image/document attachments in AI chat Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
@@ -4,11 +4,46 @@ from __future__ import annotations
|
||||
from typing import Any
|
||||
|
||||
from hub_ai.client import generate_text, model_label
|
||||
from hub_ai.context import build_daily_context, format_account_remark
|
||||
from hub_ai.context import (
|
||||
build_daily_context,
|
||||
collect_closed_trades_snapshot,
|
||||
format_account_remark,
|
||||
)
|
||||
from hub_ai.prompts import SUMMARY_SYSTEM, build_summary_user_prompt
|
||||
from hub_ai.store import append_summary, get_latest_summary, list_summaries
|
||||
|
||||
|
||||
def _stats_snapshot_from_ctx(ctx: dict) -> dict:
|
||||
day = ctx.get("trading_day")
|
||||
prev = ctx.get("prev_trading_day")
|
||||
accounts = ctx.get("accounts") or []
|
||||
return {
|
||||
"totals": ctx.get("totals"),
|
||||
"prev_trading_day": prev,
|
||||
"fund_history": ctx.get("fund_history"),
|
||||
"closed_trades": collect_closed_trades_snapshot(accounts, today=day, yesterday=prev),
|
||||
"by_account": {
|
||||
str(ac.get("key") or ac.get("id")): {
|
||||
"key": ac.get("key"),
|
||||
"name": ac.get("name"),
|
||||
"status": ac.get("status"),
|
||||
"funding_usdt": ac.get("funding_usdt"),
|
||||
"trading_usdt": ac.get("trading_usdt"),
|
||||
"available_trading_usdt": ac.get("available_trading_usdt"),
|
||||
"pnl_u": (ac.get("trade_stats") or {}).get("total_pnl_u"),
|
||||
"pnl_u_yesterday": (ac.get("trade_stats_yesterday") or {}).get("total_pnl_u"),
|
||||
"closed_count": (ac.get("trade_stats") or {}).get("closed_count"),
|
||||
"closed_count_yesterday": (ac.get("trade_stats_yesterday") or {}).get("closed_count"),
|
||||
"float_pnl_u": ac.get("float_pnl_u"),
|
||||
"remark": format_account_remark(ac),
|
||||
"monitor_lines": ac.get("monitor_lines") or {},
|
||||
"issues": ac.get("issues") or [],
|
||||
}
|
||||
for ac in accounts
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
def generate_daily_summary(
|
||||
exchanges: list[dict],
|
||||
*,
|
||||
@@ -34,22 +69,7 @@ def generate_daily_summary(
|
||||
if content.startswith("AI 调用失败"):
|
||||
return {"ok": False, "msg": content, "trading_day": day}
|
||||
|
||||
stats_snapshot = {
|
||||
"totals": ctx.get("totals"),
|
||||
"by_account": {
|
||||
str(ac.get("key") or ac.get("id")): {
|
||||
"key": ac.get("key"),
|
||||
"name": ac.get("name"),
|
||||
"status": ac.get("status"),
|
||||
"pnl_u": (ac.get("trade_stats") or {}).get("total_pnl_u"),
|
||||
"closed_count": (ac.get("trade_stats") or {}).get("closed_count"),
|
||||
"float_pnl_u": ac.get("float_pnl_u"),
|
||||
"remark": format_account_remark(ac),
|
||||
"issues": ac.get("issues") or [],
|
||||
}
|
||||
for ac in ctx.get("accounts") or []
|
||||
},
|
||||
}
|
||||
stats_snapshot = _stats_snapshot_from_ctx(ctx)
|
||||
row = append_summary(
|
||||
trading_day=day,
|
||||
content_md=content,
|
||||
|
||||
Reference in New Issue
Block a user