修复web前端，增加openai

2026-05-23 11:53:36 +08:00
parent 4439bedcb7
commit ada9478713
20 changed files with 335 additions and 355 deletions
@@ -141,9 +141,14 @@ FORCE_CLOSE_ENABLED=false
 WECHAT_TIMEOUT_SECONDS=10
 AI_TIMEOUT_SECONDS=120

-# AI 复盘服务地址（本机 Ollama 默认地址）
+# AI 提供方：openai（默认，OpenAI 兼容网关）| ollama（本机 Ollama）
+AI_PROVIDER=openai
+# OpenAI 兼容接口（示例：https://op.bz121.com/v1 ，账号见 gateway.json）
+OPENAI_API_BASE=https://op.bz121.com/v1
+OPENAI_API_KEY=你的密钥
+OPENAI_MODEL=gemma4:e4b
+# 本机 Ollama（AI_PROVIDER=ollama 时使用）
 OLLAMA_API=http://127.0.0.1:11434/api/generate
-# AI 模型名称
 AI_MODEL=huihui_ai/deepseek-r1-abliterated:latest

 # Binance 代理（可选）：本机网络不稳定时通过 SSH 动态转发 SOCKS5 出口
@@ -34,6 +34,7 @@ import sys

 if _REPO_ROOT not in sys.path:
    sys.path.insert(0, _REPO_ROOT)
+from ai_client import ai_generate, ai_review, ai_short_advice
 from fib_key_monitor_lib import (
    FIB_KEY_MONITOR_TYPES,
    calc_fib_plan,
@@ -212,8 +213,6 @@ BREAKEVEN_RR_TRIGGER = float(os.getenv("BREAKEVEN_RR_TRIGGER", "1.0"))
 BREAKEVEN_OFFSET_PCT = float(os.getenv("BREAKEVEN_OFFSET_PCT", "0.02"))
 BREAKEVEN_STEP_R = float(os.getenv("BREAKEVEN_STEP_R", "1.0"))
 DEFAULT_TRADE_STYLE = (os.getenv("DEFAULT_TRADE_STYLE", "trend") or "trend").strip().lower()
-OLLAMA_API = os.getenv("OLLAMA_API", "http://127.0.0.1:11434/api/generate")
-AI_MODEL = os.getenv("AI_MODEL", "huihui_ai/deepseek-r1-abliterated:latest")

 BINANCE_SOCKS_PROXY = (os.getenv("BINANCE_SOCKS_PROXY") or "").strip()
 BINANCE_HTTP_PROXY = (os.getenv("BINANCE_HTTP_PROXY") or "").strip()
@@ -533,61 +532,6 @@ def _journal_row_lines_for_ai(idx, row, *, include_hold_duration=True):
    return "\n".join(lines) + "\n"


-def ai_review(trades_text, period_title, image_paths=None):
-    prompt = f"""
-你是一位专业交易教练。下面是用户的{period_title}交易记录，请做简洁、可执行的复盘（中文）。
-
-【硬性规则 — 必须遵守】
- 你只能根据「交易记录」里**明确出现的字段**陈述事实；禁止编造：是否触发止损、是否扛单、亏损是否扩大、图上具体结构/进出场点位等记录里**没有**的信息。
- 「平仓/离场」只是交易员自述摘要，不是客观成交明细；若记录未写明代币是否打到止损价、是否软件平仓等，不要断言执行路径，可用「在记录有限前提下，一种可能是……」或简短写「执行路径记录不足，无法判断」。
- 「提前离场」类结论必须优先依据记录中的「提前离场记录」字段；若该段全为「无」或未出现有效内容，不得写道「明显扛单」「拒不止损」「未执行硬止损」等。
- 实际RR为负只说明结果相对于预期RR不利，不等同于「风控失灵」或「止损纪律崩溃」，除非记录里另有依据。
- 禁止用语：人身攻击、夸张定性（如「致命伤」「灾难」）；语气克制、对事不对人。
- 若有截图且你能辨认，再结合图讨论；看不清或无明确定位则明确说「无法从图确认」，不得虚构 K 线故事。
-
-【输出结构】
-1. 总体盈亏结构（紧扣笔数、盈亏数字与 RR，少形容词）
-2. 心态与执行（每笔 1–10 分 + 一句依据；依据必须对应记录字段）
-3. 行为标签（提前离场 / 乱开仓 / 扛单等）：仅在有字段或自述支撑时点名；否则写「记录未勾选或未描述，不作强加」
-4. 改进建议（最多 3 条，每条具体可执行）
-5. 图表（若有且可读）：结合价格行为简述；否则一两句说明无法看图分析
-
-交易记录：
-{trades_text}
-""".strip()
-    payload = {"model": AI_MODEL, "prompt": prompt, "stream": False, "options": {"temperature": 0.2}}
-    images = []
-    for p in image_paths or []:
-        b64 = _read_image_base64(p)
-        if b64:
-            images.append(b64)
-    if images:
-        payload["images"] = images
-    try:
-        r = requests.post(OLLAMA_API, json=payload, timeout=AI_TIMEOUT_SECONDS)
-        return r.json().get("response", "AI 生成失败")
-    except Exception as e:
-        return f"AI 调用失败：{str(e)}"
-
-
-def ai_short_advice(prompt_text):
-    prompt = f"""
-你是交易风控助理。请用中文给出**最多 3 条**提醒，要求：
- 每条不超过 25 个字
- 语气克制、具体、可执行
- 不要输出 Markdown，不要编号前缀以外的废话
-
-场景：
-{prompt_text}
-""".strip()
-    payload = {"model": AI_MODEL, "prompt": prompt, "stream": False, "options": {"temperature": 0.2}}
-    try:
-        r = requests.post(OLLAMA_API, json=payload, timeout=AI_TIMEOUT_SECONDS)
-        return (r.json().get("response") or "").strip()
-    except Exception:
-        return ""
-
-
 def _load_font(size):
    if not ImageFont:
        return None
@@ -1101,16 +1045,10 @@ JSON 字段：
  "note": ""
 }
 """.strip()
-    payload = {
-        "model": AI_MODEL,
-        "prompt": prompt,
-        "images": [image_b64],
-        "stream": False,
-        "options": {"temperature": 0.1},
-    }
    try:
-        r = requests.post(OLLAMA_API, json=payload, timeout=AI_TIMEOUT_SECONDS)
-        raw = r.json().get("response", "")
+        raw = ai_generate(prompt, images_b64=[image_b64], temperature=0.1)
+        if raw.startswith("AI 调用失败"):
+            return {}
        data = _extract_json_object(raw) or {}
        if not isinstance(data, dict):
            data = {}
@@ -5734,18 +5672,16 @@ def render_main_page(page="trade"):
        f"斐波：限价 @ E（SL/TP 为 H/L），可选移动保本｜趋势止损外侧 {KEY_TREND_STOP_OUTSIDE_PCT}%"
    )
    strategy_extra = {}
-    if page in ("strategy_trend", "strategy_roll"):
+    if page in ("strategy", "strategy_trend", "strategy_roll"):
        from strategy_ui import strategy_page_template_vars

        strategy_extra = strategy_page_template_vars(
-            conn, page, default_risk_percent=float(RISK_PERCENT)
+            conn,
+            "strategy",
+            default_risk_percent=float(RISK_PERCENT),
+            request_obj=request,
+            trend_cfg=app.extensions.get("strategy_trend_cfg"),
        )
-        if page == "strategy_trend":
-            cfg = app.extensions.get("strategy_trend_cfg")
-            if cfg:
-                from strategy_trend_register import load_trend_page_context
-
-                strategy_extra.update(load_trend_page_context(conn, request, cfg))
    conn.close()
    return render_template(
        "index.html",
@@ -7739,16 +7675,23 @@ except Exception as _hub_err:
    print(f"[hub_bridge] binance: {_hub_err}")


+@app.route("/strategy")
+@login_required
+def strategy_trading_page():
+    return render_main_page("strategy")
+
+
@app.route("/strategy/trend")
@login_required
 def strategy_trend_page():
-    return render_main_page("strategy_trend")
+    qs = request.query_string.decode()
+    return redirect(f"/strategy?{qs}" if qs else "/strategy")


@app.route("/strategy/roll")
@login_required
 def strategy_roll_page():
-    return render_main_page("strategy_roll")
+    return redirect("/strategy")


 from strategy_register import install_strategy_trading
@@ -218,7 +218,7 @@
    <div class="top-nav">
        <a href="/key_monitor" class="{% if page == 'key_monitor' %}active{% endif %}">关键位监控</a>
        <a href="/trade" class="{% if page == 'trade' %}active{% endif %}">实盘下单</a>
-        <a href="/strategy/trend" class="{% if page in ('strategy_trend', 'strategy_roll') %}active{% endif %}">策略交易</a>
+        <a href="/strategy" class="{% if page in ('strategy', 'strategy_trend', 'strategy_roll') %}active{% endif %}">策略交易</a>
        <a href="/records" class="{% if page == 'records' %}active{% endif %}">交易记录与复盘</a>
        <a href="/stats" class="{% if page == 'stats' %}active{% endif %}">统计分析</a>
    </div>
@@ -539,10 +539,8 @@
        </div>

        </div>
-        {% elif page == 'strategy_trend' %}
-        {% include 'strategy_trend_panel.html' %}
-        {% elif page == 'strategy_roll' %}
-        {% include 'strategy_roll_panel.html' %}
+        {% elif page in ('strategy', 'strategy_trend', 'strategy_roll') %}
+        {% include 'strategy_trading_page.html' %}
        {% endif %}