feat: chat integration, tenant admin spec, backend chat service, miniprogram updates, DEMO moved to tmp, XCX-TEST removed, migrations & docs
This commit is contained in:
685
apps/backend/app/services/chat_service.py
Normal file
685
apps/backend/app/services/chat_service.py
Normal file
@@ -0,0 +1,685 @@
|
||||
"""
|
||||
CHAT 模块业务逻辑层。
|
||||
|
||||
封装对话管理、消息持久化、referenceCard 组装、标题生成等核心逻辑。
|
||||
路由层（xcx_chat.py）调用本服务完成 CHAT-1/2/3/4 端点的业务处理。
|
||||
|
||||
表依赖：
|
||||
- biz.ai_conversations — 对话会话（含 context_type/context_id/title/last_message 扩展字段）
|
||||
- biz.ai_messages — 消息记录（含 reference_card 扩展字段）
|
||||
- fdw_etl.v_dim_member — 会员信息（通过 ETL 直连）
|
||||
- fdw_etl.v_dws_member_consumption_summary / v_dwd_assistant_service_log — 消费指标
|
||||
|
||||
⚠️ P5 PRD 合规：
|
||||
- app_id 固定为 'app1_chat'
|
||||
- 用户消息发送时即写入 ai_messages（role=user）
|
||||
- 流式完成后完整 assistant 回复写入 ai_messages（role=assistant），含 tokens_used
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
from datetime import datetime
|
||||
from decimal import Decimal
|
||||
from typing import Any
|
||||
|
||||
from fastapi import HTTPException, status
|
||||
|
||||
from app.ai.bailian_client import BailianClient
|
||||
from app.database import get_connection
|
||||
from app.services import fdw_queries
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
APP_ID = "app1_chat"
|
||||
|
||||
# 对话复用时限（天）
|
||||
_REUSE_DAYS = 3
|
||||
|
||||
|
||||
class ChatService:
|
||||
"""CHAT 模块业务逻辑。"""
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# CHAT-1: 对话历史列表
|
||||
# ------------------------------------------------------------------
|
||||
|
||||
def get_chat_history(
|
||||
self,
|
||||
user_id: int,
|
||||
site_id: int,
|
||||
page: int,
|
||||
page_size: int,
|
||||
) -> tuple[list[dict], int]:
|
||||
"""查询对话历史列表，返回 (items, total)。
|
||||
|
||||
按 last_message_at 倒序，JOIN v_dim_member 获取 customerName。
|
||||
仅返回 app_id='app1_chat' 的对话。
|
||||
"""
|
||||
offset = (page - 1) * page_size
|
||||
conn = get_connection()
|
||||
try:
|
||||
with conn.cursor() as cur:
|
||||
# 总数
|
||||
cur.execute(
|
||||
"""
|
||||
SELECT COUNT(*)
|
||||
FROM biz.ai_conversations
|
||||
WHERE user_id = %s AND site_id = %s AND app_id = %s
|
||||
""",
|
||||
(str(user_id), site_id, APP_ID),
|
||||
)
|
||||
total = cur.fetchone()[0]
|
||||
|
||||
# 分页列表
|
||||
cur.execute(
|
||||
"""
|
||||
SELECT id, title, context_type, context_id,
|
||||
last_message, last_message_at, created_at
|
||||
FROM biz.ai_conversations
|
||||
WHERE user_id = %s AND site_id = %s AND app_id = %s
|
||||
ORDER BY COALESCE(last_message_at, created_at) DESC
|
||||
LIMIT %s OFFSET %s
|
||||
""",
|
||||
(str(user_id), site_id, APP_ID, page_size, offset),
|
||||
)
|
||||
columns = [desc[0] for desc in cur.description]
|
||||
rows = cur.fetchall()
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
# 组装结果，尝试获取 customerName
|
||||
items: list[dict] = []
|
||||
# 收集需要查询姓名的 customer context_id
|
||||
customer_ids: list[int] = []
|
||||
raw_items: list[dict] = []
|
||||
for row in rows:
|
||||
item = dict(zip(columns, row))
|
||||
raw_items.append(item)
|
||||
if item.get("context_type") == "customer" and item.get("context_id"):
|
||||
try:
|
||||
customer_ids.append(int(item["context_id"]))
|
||||
except (ValueError, TypeError):
|
||||
pass
|
||||
|
||||
# 批量查询客户姓名（FDW 降级：查询失败返回空映射）
|
||||
name_map: dict[int, str] = {}
|
||||
if customer_ids:
|
||||
try:
|
||||
biz_conn = get_connection()
|
||||
try:
|
||||
info_map = fdw_queries.get_member_info(biz_conn, site_id, customer_ids)
|
||||
for mid, info in info_map.items():
|
||||
name_map[mid] = info.get("nickname") or ""
|
||||
finally:
|
||||
biz_conn.close()
|
||||
except Exception:
|
||||
logger.warning("查询客户姓名失败，降级为空", exc_info=True)
|
||||
|
||||
for item in raw_items:
|
||||
customer_name: str | None = None
|
||||
if item.get("context_type") == "customer" and item.get("context_id"):
|
||||
try:
|
||||
customer_name = name_map.get(int(item["context_id"]))
|
||||
except (ValueError, TypeError):
|
||||
pass
|
||||
|
||||
# 生成标题
|
||||
title = self.generate_title(
|
||||
title=item.get("title"),
|
||||
customer_name=customer_name,
|
||||
conversation_id=item["id"],
|
||||
)
|
||||
|
||||
ts = item.get("last_message_at") or item.get("created_at")
|
||||
items.append({
|
||||
"id": item["id"],
|
||||
"title": title,
|
||||
"customer_name": customer_name,
|
||||
"last_message": item.get("last_message"),
|
||||
"timestamp": ts.isoformat() if isinstance(ts, datetime) else str(ts) if ts else "",
|
||||
"unread_count": 0,
|
||||
})
|
||||
|
||||
return items, total
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# 对话复用 / 创建
|
||||
# ------------------------------------------------------------------
|
||||
|
||||
def get_or_create_session(
|
||||
self,
|
||||
user_id: int,
|
||||
site_id: int,
|
||||
context_type: str,
|
||||
context_id: str | None,
|
||||
) -> int:
|
||||
"""按入口上下文查找或创建对话，返回 chat_id。
|
||||
|
||||
复用规则：
|
||||
- context_type='task': 同一 taskId 始终复用（无时限）
|
||||
- context_type='customer'/'coach': 最后消息 ≤ 3 天复用，> 3 天新建
|
||||
- context_type='general': 始终新建
|
||||
"""
|
||||
# general 入口始终新建
|
||||
if context_type == "general":
|
||||
return self._create_session(user_id, site_id, context_type, context_id)
|
||||
|
||||
conn = get_connection()
|
||||
try:
|
||||
with conn.cursor() as cur:
|
||||
if context_type == "task":
|
||||
# task 入口：始终复用（无时限）
|
||||
cur.execute(
|
||||
"""
|
||||
SELECT id FROM biz.ai_conversations
|
||||
WHERE user_id = %s AND site_id = %s
|
||||
AND context_type = 'task' AND context_id = %s
|
||||
ORDER BY created_at DESC LIMIT 1
|
||||
""",
|
||||
(str(user_id), site_id, context_id),
|
||||
)
|
||||
elif context_type in ("customer", "coach"):
|
||||
# customer/coach 入口：3 天时限复用
|
||||
cur.execute(
|
||||
"""
|
||||
SELECT id FROM biz.ai_conversations
|
||||
WHERE user_id = %s AND site_id = %s
|
||||
AND context_type = %s AND context_id = %s
|
||||
AND last_message_at > NOW() - INTERVAL '3 days'
|
||||
ORDER BY last_message_at DESC LIMIT 1
|
||||
""",
|
||||
(str(user_id), site_id, context_type, context_id),
|
||||
)
|
||||
else:
|
||||
# 未知类型，新建
|
||||
return self._create_session(user_id, site_id, context_type, context_id)
|
||||
|
||||
row = cur.fetchone()
|
||||
if row:
|
||||
return row[0]
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
# 未找到可复用对话，新建
|
||||
return self._create_session(user_id, site_id, context_type, context_id)
|
||||
|
||||
def _create_session(
|
||||
self,
|
||||
user_id: int,
|
||||
site_id: int,
|
||||
context_type: str,
|
||||
context_id: str | None,
|
||||
) -> int:
|
||||
"""创建新对话记录，返回 conversation_id。"""
|
||||
conn = get_connection()
|
||||
try:
|
||||
with conn.cursor() as cur:
|
||||
# 查询用户昵称
|
||||
cur.execute(
|
||||
"SELECT nickname FROM auth.users WHERE id = %s",
|
||||
(user_id,),
|
||||
)
|
||||
row = cur.fetchone()
|
||||
nickname = row[0] if row and row[0] else ""
|
||||
|
||||
cur.execute(
|
||||
"""
|
||||
INSERT INTO biz.ai_conversations
|
||||
(user_id, nickname, app_id, site_id, context_type, context_id)
|
||||
VALUES (%s, %s, %s, %s, %s, %s)
|
||||
RETURNING id
|
||||
""",
|
||||
(str(user_id), nickname, APP_ID, site_id, context_type, context_id),
|
||||
)
|
||||
new_id = cur.fetchone()[0]
|
||||
conn.commit()
|
||||
return new_id
|
||||
except Exception:
|
||||
conn.rollback()
|
||||
raise
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# CHAT-2: 消息列表
|
||||
# ------------------------------------------------------------------
|
||||
|
||||
def get_messages(
|
||||
self,
|
||||
chat_id: int,
|
||||
user_id: int,
|
||||
site_id: int,
|
||||
page: int,
|
||||
page_size: int,
|
||||
) -> tuple[list[dict], int, int]:
|
||||
"""查询消息列表，返回 (messages, total, chat_id)。
|
||||
|
||||
验证 chat_id 归属当前用户，按 created_at 正序。
|
||||
"""
|
||||
self._verify_ownership(chat_id, user_id, site_id)
|
||||
|
||||
offset = (page - 1) * page_size
|
||||
conn = get_connection()
|
||||
try:
|
||||
with conn.cursor() as cur:
|
||||
cur.execute(
|
||||
"SELECT COUNT(*) FROM biz.ai_messages WHERE conversation_id = %s",
|
||||
(chat_id,),
|
||||
)
|
||||
total = cur.fetchone()[0]
|
||||
|
||||
cur.execute(
|
||||
"""
|
||||
SELECT id, role, content, created_at, reference_card
|
||||
FROM biz.ai_messages
|
||||
WHERE conversation_id = %s
|
||||
ORDER BY created_at ASC
|
||||
LIMIT %s OFFSET %s
|
||||
""",
|
||||
(chat_id, page_size, offset),
|
||||
)
|
||||
columns = [desc[0] for desc in cur.description]
|
||||
rows = cur.fetchall()
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
messages = []
|
||||
for row in rows:
|
||||
item = dict(zip(columns, row))
|
||||
ref_card = item.get("reference_card")
|
||||
# reference_card 可能是 dict（psycopg2 自动解析 jsonb）或 str
|
||||
if isinstance(ref_card, str):
|
||||
try:
|
||||
ref_card = json.loads(ref_card)
|
||||
except (json.JSONDecodeError, TypeError):
|
||||
ref_card = None
|
||||
|
||||
created_at = item["created_at"]
|
||||
messages.append({
|
||||
"id": item["id"],
|
||||
"role": item["role"],
|
||||
"content": item["content"],
|
||||
"created_at": created_at.isoformat() if isinstance(created_at, datetime) else str(created_at),
|
||||
"reference_card": ref_card,
|
||||
})
|
||||
|
||||
return messages, total, chat_id
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# CHAT-3: 发送消息（同步回复）
|
||||
# ------------------------------------------------------------------
|
||||
|
||||
async def send_message_sync(
|
||||
self,
|
||||
chat_id: int,
|
||||
content: str,
|
||||
user_id: int,
|
||||
site_id: int,
|
||||
) -> dict:
|
||||
"""发送消息并获取同步 AI 回复。
|
||||
|
||||
流程：
|
||||
1. 验证 chatId 归属
|
||||
2. 存入用户消息（立即写入）
|
||||
3. 调用 AI 获取回复
|
||||
4. 存入 AI 回复
|
||||
5. 更新 session 的 last_message / last_message_at
|
||||
6. AI 失败时返回错误提示消息（HTTP 200）
|
||||
"""
|
||||
self._verify_ownership(chat_id, user_id, site_id)
|
||||
|
||||
# 1. 立即存入用户消息（P5 PRD 合规：发送时即写入）
|
||||
user_msg_id, user_created_at = self._save_message(chat_id, "user", content)
|
||||
|
||||
# 2. 调用 AI
|
||||
ai_reply_text: str
|
||||
tokens_used: int | None = None
|
||||
try:
|
||||
ai_reply_text, tokens_used = await self._call_ai(chat_id, content, user_id, site_id)
|
||||
except Exception as e:
|
||||
logger.error("AI 服务调用失败: %s", e, exc_info=True)
|
||||
ai_reply_text = "抱歉，AI 助手暂时无法回复，请稍后重试"
|
||||
|
||||
# 3. 存入 AI 回复
|
||||
ai_msg_id, ai_created_at = self._save_message(
|
||||
chat_id, "assistant", ai_reply_text, tokens_used=tokens_used,
|
||||
)
|
||||
|
||||
# 4. 更新 session 元数据
|
||||
self._update_session_metadata(chat_id, ai_reply_text)
|
||||
|
||||
return {
|
||||
"user_message": {
|
||||
"id": user_msg_id,
|
||||
"content": content,
|
||||
"created_at": user_created_at,
|
||||
},
|
||||
"ai_reply": {
|
||||
"id": ai_msg_id,
|
||||
"content": ai_reply_text,
|
||||
"created_at": ai_created_at,
|
||||
},
|
||||
}
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# referenceCard 组装
|
||||
# ------------------------------------------------------------------
|
||||
|
||||
def build_reference_card(
|
||||
self,
|
||||
customer_id: int,
|
||||
site_id: int,
|
||||
) -> dict | None:
|
||||
"""从 FDW 查询客户关键指标，组装 referenceCard。
|
||||
|
||||
⚠️ DWD-DOC 规则：金额用 items_sum 口径（ledger_amount），
|
||||
会员信息通过 member_id JOIN dim_member（scd2_is_current=1）。
|
||||
|
||||
FDW 查询失败时静默降级返回 None（不影响消息本身）。
|
||||
"""
|
||||
try:
|
||||
biz_conn = get_connection()
|
||||
try:
|
||||
# 客户姓名
|
||||
info_map = fdw_queries.get_member_info(biz_conn, site_id, [customer_id])
|
||||
if customer_id not in info_map:
|
||||
return None
|
||||
member_name = info_map[customer_id].get("nickname") or "未知客户"
|
||||
|
||||
# 余额
|
||||
balance: Decimal | None = None
|
||||
try:
|
||||
balance_map = fdw_queries.get_member_balance(biz_conn, site_id, [customer_id])
|
||||
balance = balance_map.get(customer_id)
|
||||
except Exception:
|
||||
logger.warning("referenceCard: 查询余额失败", exc_info=True)
|
||||
|
||||
# 近 30 天消费（items_sum 口径）
|
||||
consume_30d: Decimal | None = None
|
||||
try:
|
||||
consume_30d = self._get_consumption_30d(biz_conn, site_id, customer_id)
|
||||
except Exception:
|
||||
logger.warning("referenceCard: 查询近30天消费失败", exc_info=True)
|
||||
|
||||
# 近 30 天到店次数
|
||||
visit_count: int | None = None
|
||||
try:
|
||||
visit_count = self._get_visit_count_30d(biz_conn, site_id, customer_id)
|
||||
except Exception:
|
||||
logger.warning("referenceCard: 查询到店次数失败", exc_info=True)
|
||||
|
||||
finally:
|
||||
biz_conn.close()
|
||||
|
||||
# 格式化
|
||||
balance_str = f"¥{balance:,.2f}" if balance is not None else "—"
|
||||
consume_str = f"¥{consume_30d:,.2f}" if consume_30d is not None else "—"
|
||||
visit_str = f"{visit_count}次" if visit_count is not None else "—"
|
||||
|
||||
return {
|
||||
"type": "customer",
|
||||
"title": f"{member_name} — 消费概览",
|
||||
"summary": f"余额 {balance_str}，近30天消费 {consume_str}",
|
||||
"data": {
|
||||
"余额": balance_str,
|
||||
"近30天消费": consume_str,
|
||||
"到店次数": visit_str,
|
||||
},
|
||||
}
|
||||
|
||||
except Exception:
|
||||
logger.warning("referenceCard 组装失败，降级为 null", exc_info=True)
|
||||
return None
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# 标题生成
|
||||
# ------------------------------------------------------------------
|
||||
|
||||
def generate_title(
|
||||
self,
|
||||
title: str | None = None,
|
||||
customer_name: str | None = None,
|
||||
conversation_id: int | None = None,
|
||||
first_message: str | None = None,
|
||||
) -> str:
|
||||
"""生成对话标题：自定义标题 > 客户姓名 > 首条消息前 20 字。
|
||||
|
||||
结果始终非空。
|
||||
"""
|
||||
# 优先级 1：自定义标题
|
||||
if title and title.strip():
|
||||
return title.strip()
|
||||
|
||||
# 优先级 2：客户姓名
|
||||
if customer_name and customer_name.strip():
|
||||
return customer_name.strip()
|
||||
|
||||
# 优先级 3：首条消息前 20 字
|
||||
if first_message is None and conversation_id is not None:
|
||||
first_message = self._get_first_message(conversation_id)
|
||||
|
||||
if first_message and first_message.strip():
|
||||
text = first_message.strip()
|
||||
return text[:20] if len(text) > 20 else text
|
||||
|
||||
return "新对话"
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# 内部辅助方法
|
||||
# ------------------------------------------------------------------
|
||||
|
||||
def _verify_ownership(self, chat_id: int, user_id: int, site_id: int) -> None:
|
||||
"""验证对话归属当前用户，不属于时抛出 HTTP 403/404。"""
|
||||
conn = get_connection()
|
||||
try:
|
||||
with conn.cursor() as cur:
|
||||
cur.execute(
|
||||
"""
|
||||
SELECT user_id FROM biz.ai_conversations
|
||||
WHERE id = %s AND site_id = %s
|
||||
""",
|
||||
(chat_id, site_id),
|
||||
)
|
||||
row = cur.fetchone()
|
||||
if not row:
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_404_NOT_FOUND,
|
||||
detail="对话不存在",
|
||||
)
|
||||
if str(row[0]) != str(user_id):
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_403_FORBIDDEN,
|
||||
detail="无权访问此对话",
|
||||
)
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
def _save_message(
|
||||
self,
|
||||
conversation_id: int,
|
||||
role: str,
|
||||
content: str,
|
||||
tokens_used: int | None = None,
|
||||
reference_card: dict | None = None,
|
||||
) -> tuple[int, str]:
|
||||
"""写入消息记录，返回 (message_id, created_at ISO 字符串)。"""
|
||||
conn = get_connection()
|
||||
try:
|
||||
with conn.cursor() as cur:
|
||||
cur.execute(
|
||||
"""
|
||||
INSERT INTO biz.ai_messages
|
||||
(conversation_id, role, content, tokens_used, reference_card)
|
||||
VALUES (%s, %s, %s, %s, %s)
|
||||
RETURNING id, created_at
|
||||
""",
|
||||
(
|
||||
conversation_id,
|
||||
role,
|
||||
content,
|
||||
tokens_used,
|
||||
json.dumps(reference_card, ensure_ascii=False) if reference_card else None,
|
||||
),
|
||||
)
|
||||
row = cur.fetchone()
|
||||
conn.commit()
|
||||
msg_id = row[0]
|
||||
created_at = row[1]
|
||||
return msg_id, created_at.isoformat() if isinstance(created_at, datetime) else str(created_at)
|
||||
except Exception:
|
||||
conn.rollback()
|
||||
raise
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
def _update_session_metadata(self, chat_id: int, last_message: str) -> None:
|
||||
"""更新对话的 last_message 和 last_message_at。"""
|
||||
# 截断至 100 字
|
||||
truncated = last_message[:100] if len(last_message) > 100 else last_message
|
||||
conn = get_connection()
|
||||
try:
|
||||
with conn.cursor() as cur:
|
||||
cur.execute(
|
||||
"""
|
||||
UPDATE biz.ai_conversations
|
||||
SET last_message = %s, last_message_at = NOW()
|
||||
WHERE id = %s
|
||||
""",
|
||||
(truncated, chat_id),
|
||||
)
|
||||
conn.commit()
|
||||
except Exception:
|
||||
conn.rollback()
|
||||
raise
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
def _get_first_message(self, conversation_id: int) -> str | None:
|
||||
"""查询对话的首条 user 消息内容。"""
|
||||
conn = get_connection()
|
||||
try:
|
||||
with conn.cursor() as cur:
|
||||
cur.execute(
|
||||
"""
|
||||
SELECT content FROM biz.ai_messages
|
||||
WHERE conversation_id = %s AND role = 'user'
|
||||
ORDER BY created_at ASC LIMIT 1
|
||||
""",
|
||||
(conversation_id,),
|
||||
)
|
||||
row = cur.fetchone()
|
||||
return row[0] if row else None
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
async def _call_ai(
|
||||
self,
|
||||
chat_id: int,
|
||||
content: str,
|
||||
user_id: int,
|
||||
site_id: int,
|
||||
) -> tuple[str, int | None]:
|
||||
"""调用百炼 API 获取非流式回复，返回 (reply_text, tokens_used)。
|
||||
|
||||
构建历史消息上下文发送给 AI。
|
||||
"""
|
||||
bailian = _get_bailian_client()
|
||||
|
||||
# 获取历史消息作为上下文（最近 20 条）
|
||||
conn = get_connection()
|
||||
try:
|
||||
with conn.cursor() as cur:
|
||||
cur.execute(
|
||||
"""
|
||||
SELECT role, content FROM biz.ai_messages
|
||||
WHERE conversation_id = %s
|
||||
ORDER BY created_at ASC
|
||||
""",
|
||||
(chat_id,),
|
||||
)
|
||||
history = cur.fetchall()
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
# 构建消息列表
|
||||
messages: list[dict] = []
|
||||
# 取最近 20 条（含刚写入的 user 消息）
|
||||
recent = history[-20:] if len(history) > 20 else history
|
||||
for role, msg_content in recent:
|
||||
messages.append({"role": role, "content": msg_content})
|
||||
|
||||
# 如果没有 system 消息，添加默认 system prompt
|
||||
if not messages or messages[0]["role"] != "system":
|
||||
system_prompt = {
|
||||
"role": "system",
|
||||
"content": json.dumps(
|
||||
{"task": "你是台球门店的 AI 助手，根据用户的问题和当前页面上下文提供帮助。"},
|
||||
ensure_ascii=False,
|
||||
),
|
||||
}
|
||||
messages.insert(0, system_prompt)
|
||||
|
||||
# 非流式调用（chat_stream 用于 SSE，这里用 chat_stream 收集完整回复）
|
||||
full_parts: list[str] = []
|
||||
async for chunk in bailian.chat_stream(messages):
|
||||
full_parts.append(chunk)
|
||||
|
||||
reply = "".join(full_parts)
|
||||
# 流式模式不返回 tokens_used，按字符数估算
|
||||
estimated_tokens = len(reply)
|
||||
return reply, estimated_tokens
|
||||
|
||||
@staticmethod
|
||||
def _get_consumption_30d(conn: Any, site_id: int, member_id: int) -> Decimal | None:
|
||||
"""查询客户近 30 天消费金额（items_sum 口径）。
|
||||
|
||||
⚠️ DWD-DOC 规则 1: 使用 ledger_amount（items_sum 口径），禁用 consume_money。
|
||||
"""
|
||||
with fdw_queries._fdw_context(conn, site_id) as cur:
|
||||
cur.execute(
|
||||
"""
|
||||
SELECT COALESCE(SUM(ledger_amount), 0)
|
||||
FROM app.v_dwd_assistant_service_log
|
||||
WHERE tenant_member_id = %s
|
||||
AND is_delete = 0
|
||||
AND create_time >= (CURRENT_DATE - INTERVAL '30 days')::timestamptz
|
||||
""",
|
||||
(member_id,),
|
||||
)
|
||||
row = cur.fetchone()
|
||||
return Decimal(str(row[0])) if row and row[0] is not None else None
|
||||
|
||||
@staticmethod
|
||||
def _get_visit_count_30d(conn: Any, site_id: int, member_id: int) -> int | None:
|
||||
"""查询客户近 30 天到店次数。"""
|
||||
with fdw_queries._fdw_context(conn, site_id) as cur:
|
||||
cur.execute(
|
||||
"""
|
||||
SELECT COUNT(DISTINCT create_time::date)
|
||||
FROM app.v_dwd_assistant_service_log
|
||||
WHERE tenant_member_id = %s
|
||||
AND is_delete = 0
|
||||
AND create_time >= (CURRENT_DATE - INTERVAL '30 days')::timestamptz
|
||||
""",
|
||||
(member_id,),
|
||||
)
|
||||
row = cur.fetchone()
|
||||
return int(row[0]) if row and row[0] is not None else None
|
||||
|
||||
|
||||
# ── 模块级辅助函数 ──────────────────────────────────────────────
|
||||
|
||||
|
||||
def _get_bailian_client() -> BailianClient:
|
||||
"""从环境变量构建 BailianClient，缺失时报错。"""
|
||||
api_key = os.environ.get("BAILIAN_API_KEY")
|
||||
base_url = os.environ.get("BAILIAN_BASE_URL")
|
||||
model = os.environ.get("BAILIAN_MODEL")
|
||||
if not api_key or not base_url or not model:
|
||||
raise RuntimeError(
|
||||
"百炼 API 环境变量缺失，需要 BAILIAN_API_KEY、BAILIAN_BASE_URL、BAILIAN_MODEL"
|
||||
)
|
||||
return BailianClient(api_key=api_key, base_url=base_url, model=model)
|
||||
Reference in New Issue
Block a user