""" 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.config import AIConfig from app.ai.dashscope_client import DashScopeClient from app.database import get_connection from app.services import fdw_queries from app.trace.decorators import trace_service logger = logging.getLogger(__name__) APP_ID = "app1_chat" # 对话复用时限（天） _REUSE_DAYS = 3 class ChatService: """CHAT 模块业务逻辑。""" # ------------------------------------------------------------------ # CHAT-1: 对话历史列表 # ------------------------------------------------------------------ @trace_service("查询对话历史", "Get chat history") 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 # ------------------------------------------------------------------ # 对话复用 / 创建 # ------------------------------------------------------------------ @trace_service("查找或创建对话", "Get or create session") 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。同时生成 session_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, EXTRACT(EPOCH FROM created_at)::bigint """, (str(user_id), nickname, APP_ID, site_id, context_type, context_id), ) result = cur.fetchone() new_id = result[0] created_ts = result[1] # 生成 session_id 并回写（格式：conv_{id}_{timestamp}） session_id = f"conv_{new_id}_{created_ts}" cur.execute( """ UPDATE biz.ai_conversations SET session_id = %s WHERE id = %s """, (session_id, new_id), ) conn.commit() return new_id except Exception: conn.rollback() raise finally: conn.close() @trace_service("获取对话 session_id", "Get session ID") def get_session_id(self, chat_id: int) -> str | None: """获取对话的 session_id。无记录或字段为空时返回 None。""" conn = get_connection() try: with conn.cursor() as cur: cur.execute( "SELECT session_id FROM biz.ai_conversations WHERE id = %s", (chat_id,), ) row = cur.fetchone() return row[0] if row and row[0] else None finally: conn.close() # ------------------------------------------------------------------ # CHAT-2: 消息列表 # ------------------------------------------------------------------ @trace_service("查询消息列表", "Get messages") 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: 发送消息（同步回复） # ------------------------------------------------------------------ @trace_service("发送消息并获取回复", "Send message sync") 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 组装 # ------------------------------------------------------------------ @trace_service("构建引用卡片", "Build reference card") 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 # ------------------------------------------------------------------ # 标题生成 # ------------------------------------------------------------------ @trace_service("生成对话标题", "Generate title") 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]: """调用 DashScope Application API 获取非流式回复，返回 (reply_text, tokens_used)。 通过 Application.call() 调用 App1（通用对话），prompt 为最近历史拼接。 """ # CHANGE 2026-03-22 | BailianClient → DashScopeClient（P14 迁移收尾） client = _get_dashscope_client() ai_config = AIConfig.from_env() # 获取历史消息作为上下文（最近 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() # 拼接历史消息为 prompt 文本 recent = history[-20:] if len(history) > 20 else history prompt_parts: list[str] = [] for role, msg_content in recent: prompt_parts.append(f"[{role}]: {msg_content}") prompt = "\n".join(prompt_parts) # 通过 Application API 调用 App1 result, tokens_used, _session_id = await client.call_app( ai_config.app_id_1_chat, prompt, ) # 从返回结果提取文本回复 reply = result.get("text", "") if isinstance(result, dict) else str(result) return reply, tokens_used @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_dashscope_client() -> DashScopeClient: """从环境变量构建 DashScopeClient，缺失时报错。""" # CHANGE 2026-03-22 | BailianClient → DashScopeClient（P14 迁移收尾） ai_config = AIConfig.from_env() return DashScopeClient(api_key=ai_config.api_key, workspace_id=ai_config.workspace_id)