""" 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)