dekun
cee641ba5d
Aggregate four-account trades via hub_ai module and /api/hub/trades/today; store sessions in JSON; default OpenAI config matches instances. Co-authored-by: Cursor <cursoragent@cursor.com>
109 lines
3.6 KiB
Python
109 lines
3.6 KiB
Python
"""中控 AI FastAPI 路由。"""
|
|
from __future__ import annotations
|
|
|
|
from typing import Callable
|
|
|
|
from fastapi import APIRouter, HTTPException
|
|
from pydantic import BaseModel, Field
|
|
|
|
from hub_ai.chat import get_chat_state, send_chat_message, start_new_chat
|
|
from hub_ai.client import model_label
|
|
from hub_ai.config import trading_day_reset_hour
|
|
from hub_ai.context import build_daily_context
|
|
from hub_ai.store import get_latest_summary, list_summaries
|
|
from hub_ai.summary import generate_daily_summary
|
|
from hub_trades_lib import current_trading_day
|
|
|
|
|
|
class ChatSendBody(BaseModel):
|
|
message: str = ""
|
|
trading_day: str = ""
|
|
|
|
|
|
class SummaryGenerateBody(BaseModel):
|
|
trading_day: str = ""
|
|
force: bool = False
|
|
|
|
|
|
class ChatNewBody(BaseModel):
|
|
trading_day: str = ""
|
|
|
|
|
|
def create_hub_ai_router(*, load_all_exchanges: Callable[[], list]) -> APIRouter:
|
|
router = APIRouter(prefix="/api/ai", tags=["hub-ai"])
|
|
|
|
def _day(raw: str = "") -> str:
|
|
d = (raw or "").strip()[:10]
|
|
return d or current_trading_day(reset_hour=trading_day_reset_hour())
|
|
|
|
@router.get("/meta")
|
|
def api_ai_meta():
|
|
return {
|
|
"ok": True,
|
|
"model": model_label(),
|
|
"trading_day_reset_hour": trading_day_reset_hour(),
|
|
"trading_day": current_trading_day(reset_hour=trading_day_reset_hour()),
|
|
"storage": {
|
|
"summaries": "hub_ai_summaries.json",
|
|
"chat": "hub_ai_chat.json",
|
|
},
|
|
}
|
|
|
|
@router.get("/context")
|
|
def api_ai_context(trading_day: str = ""):
|
|
exchanges = load_all_exchanges()
|
|
ctx = build_daily_context(exchanges, trading_day=_day(trading_day))
|
|
return {"ok": True, **ctx}
|
|
|
|
@router.get("/summary")
|
|
def api_ai_summary_list(trading_day: str = ""):
|
|
day = _day(trading_day) if trading_day.strip() else ""
|
|
items = list_summaries(trading_day=day or None, limit=20)
|
|
latest = get_latest_summary(_day(trading_day)) if trading_day.strip() else (
|
|
items[0] if items else None
|
|
)
|
|
return {
|
|
"ok": True,
|
|
"trading_day": _day(trading_day) if trading_day.strip() else None,
|
|
"summaries": items,
|
|
"latest": latest,
|
|
"model": model_label(),
|
|
}
|
|
|
|
@router.post("/summary/generate")
|
|
def api_ai_summary_generate(body: SummaryGenerateBody = SummaryGenerateBody()):
|
|
exchanges = load_all_exchanges()
|
|
result = generate_daily_summary(
|
|
exchanges,
|
|
trading_day=_day(body.trading_day) if body.trading_day.strip() else None,
|
|
force=bool(body.force),
|
|
)
|
|
if not result.get("ok"):
|
|
raise HTTPException(status_code=502, detail=result.get("msg") or "生成失败")
|
|
result.pop("context", None)
|
|
return result
|
|
|
|
@router.get("/chat/session")
|
|
def api_ai_chat_session():
|
|
state = get_chat_state()
|
|
return {"ok": True, **state, "model": model_label()}
|
|
|
|
@router.post("/chat/new")
|
|
def api_ai_chat_new(body: ChatNewBody = ChatNewBody()):
|
|
day = _day(body.trading_day)
|
|
return start_new_chat(trading_day=day)
|
|
|
|
@router.post("/chat/send")
|
|
def api_ai_chat_send(body: ChatSendBody):
|
|
exchanges = load_all_exchanges()
|
|
result = send_chat_message(
|
|
exchanges,
|
|
body.message,
|
|
trading_day=_day(body.trading_day) if body.trading_day.strip() else None,
|
|
)
|
|
if not result.get("ok"):
|
|
raise HTTPException(status_code=502, detail=result.get("msg") or "发送失败")
|
|
return result
|
|
|
|
return router
|