1

apps/backend/app/ai/prompts/app2_finance_prompt.py

@@ -0,0 +1,145 @@
+"""应用 2 财务洞察 Prompt 模板。
+
+构建包含当期和上期收入结构的完整 Prompt，供百炼 API 生成财务洞察。
+
+收入字段映射（严格遵守 items_sum 口径）：
+- table_fee = table_charge_money（台费）
+- assistant_pd = assistant_pd_money（陪打费）
+- assistant_cx = assistant_cx_money（超休费）
+- goods = goods_money（商品收入）
+- recharge = 充值 pay_amount settle_type=5（充值收入）
+
+禁止使用 consume_money，统一使用：
+items_sum = table_charge_money + goods_money + assistant_pd_money
+            + assistant_cx_money + electricity_money
+"""
+
+from __future__ import annotations
+
+import json
+
+
+def build_prompt(context: dict) -> list[dict]:
+    """构建 App2 财务洞察 Prompt 消息列表。
+
+    Args:
+        context: 包含以下字段：
+            - site_id: int，门店 ID
+            - time_dimension: str，时间维度编码
+            - current_data: dict，当期数据
+            - previous_data: dict，上期数据
+
+    Returns:
+        messages 列表（system + user），供 BailianClient.chat_json 调用
+    """
+    site_id = context.get("site_id", 0)
+    time_dimension = context.get("time_dimension", "")
+    current_data = context.get("current_data", {})
+    previous_data = context.get("previous_data", {})
+
+    system_content = _build_system_content(
+        site_id=site_id,
+        time_dimension=time_dimension,
+        current_data=current_data,
+        previous_data=previous_data,
+    )
+
+    user_content = (
+        f"请根据以上数据，为门店 {site_id} 生成 {_dimension_label(time_dimension)} 的财务洞察分析。"
+        "以 JSON 格式返回，包含 insights 数组，每项含 seq（序号）、title（标题）、body（正文）。"
+    )
+
+    return [
+        {"role": "system", "content": json.dumps(system_content, ensure_ascii=False)},
+        {"role": "user", "content": user_content},
+    ]
+
+
+def _build_system_content(
+    *,
+    site_id: int,
+    time_dimension: str,
+    current_data: dict,
+    previous_data: dict,
+) -> dict:
+    """构建 system prompt JSON 结构。"""
+    return {
+        "task": (
+            "你是台球门店的财务分析 AI 助手。"
+            "根据提供的当期和上期经营数据，生成结构化的财务洞察。"
+            "分析维度包括：收入结构变化、各收入项占比、环比趋势、异常波动。"
+            "输出 JSON 格式：{\"insights\": [{\"seq\": 1, \"title\": \"...\", \"body\": \"...\"}]}"
+        ),
+        "data": {
+            "site_id": site_id,
+            "time_dimension": time_dimension,
+            "time_dimension_label": _dimension_label(time_dimension),
+            "current_period": _build_period_data(current_data),
+            "previous_period": _build_period_data(previous_data),
+        },
+        "reference": {
+            "field_mapping": {
+                "items_sum": (
+                    "table_charge_money + goods_money + assistant_pd_money"
+                    " + assistant_cx_money + electricity_money"
+                ),
+                "table_fee": "table_charge_money（台费收入）",
+                "assistant_pd": "assistant_pd_money（陪打费）",
+                "assistant_cx": "assistant_cx_money（超休费）",
+                "goods": "goods_money（商品收入）",
+                "recharge": "充值 pay_amount（settle_type=5，充值收入）",
+                "electricity": "electricity_money（电费，当前未启用，全为 0）",
+            },
+            "rules": [
+                "统一使用 items_sum 口径计算营收总额",
+                "助教费用必须拆分为 assistant_pd_money（陪打）和 assistant_cx_money（超休）",
+                "支付渠道恒等式：balance_amount = recharge_card_amount + gift_card_amount",
+                "金额单位：元（CNY），保留两位小数",
+            ],
+        },
+    }
+
+
+def _build_period_data(data: dict) -> dict:
+    """构建单期数据结构，确保字段名遵守 items_sum 口径。"""
+    return {
+        # 收入结构（items_sum 口径）
+        "table_charge_money": data.get("table_charge_money", 0),
+        "goods_money": data.get("goods_money", 0),
+        "assistant_pd_money": data.get("assistant_pd_money", 0),
+        "assistant_cx_money": data.get("assistant_cx_money", 0),
+        "electricity_money": data.get("electricity_money", 0),
+        # 充值收入
+        "recharge_income": data.get("recharge_income", 0),
+        # 储值资产
+        "balance_pay": data.get("balance_pay", 0),
+        "recharge_card_pay": data.get("recharge_card_pay", 0),
+        "gift_card_pay": data.get("gift_card_pay", 0),
+        # 费用汇总
+        "discount_amount": data.get("discount_amount", 0),
+        "adjust_amount": data.get("adjust_amount", 0),
+        # 平台结算
+        "platform_settlement_amount": data.get("platform_settlement_amount", 0),
+        "groupbuy_pay_amount": data.get("groupbuy_pay_amount", 0),
+        # 汇总
+        "order_count": data.get("order_count", 0),
+        "member_count": data.get("member_count", 0),
+    }
+
+
+# 时间维度编码 → 中文标签
+_DIMENSION_LABELS: dict[str, str] = {
+    "this_month": "本月",
+    "last_month": "上月",
+    "this_week": "本周",
+    "last_week": "上周",
+    "last_3_months": "近三个月",
+    "this_quarter": "本季度",
+    "last_quarter": "上季度",
+    "last_6_months": "近六个月",
+}
+
+
+def _dimension_label(dimension: str) -> str:
+    """将时间维度编码转为中文标签。"""
+    return _DIMENSION_LABELS.get(dimension, dimension)