feat: 优化AI图表单位显示并新增登录验证码

Author: heyike

packages/mobile/src/lib/detectVisualization.ts

@@ -4,18 +4,29 @@
  */
 import type { AiVisualizationSpec } from '@/types'
 import {
+  formatChartMetricValue,
   fieldToLabel,
+  fmtCompactNum,
   fmtNum,
   fmtQtyWithUnit,
   fmtFieldValue,
   fmtNumRaw,
   getDisplayColumns,
+  normalizeChartMetricValue,
   isStrategySnapshotRows,
   NON_METRIC_NUMERIC_FIELDS,
   DETAIL_RECORD_FIELDS,
 } from '@shared/constants/ai-field-label'
 
-export { fieldToLabel, fmtNum, fmtQtyWithUnit, fmtFieldValue, fmtNumRaw }
+export {
+  fieldToLabel,
+  formatChartMetricValue,
+  fmtCompactNum,
+  fmtNum,
+  fmtQtyWithUnit,
+  fmtFieldValue,
+  fmtNumRaw,
+}
 
 const CHART_REQUEST_RE = /图表|可视化|柱状图|条形图|折线图|饼图|曲线图|趋势图|对比图/
 const COMPARISON_REQUEST_RE = /对比|比较|比一比|和.+比|相比|相较|排名|排行|名次|领先|落后|高低|头部|梯队|其它茶|其他茶/
@@ -128,7 +139,7 @@ export function prepareVisualizationRows(
     [xField]: xField === DERIVED_CATEGORY_FIELD
       ? buildDerivedCategoryLabel(row, duplicateProductNames)
       : String(row[xField] ?? ''),
-    ...(yField ? { [yField]: Number(row[yField]) || 0 } : {}),
+    ...(yField ? { [yField]: normalizeChartMetricValue(row[yField], yField, row, rows) } : {}),
   }))
 }
 

packages/mobile/src/pages/ai/components/AiChart.tsx

@@ -8,7 +8,12 @@ import {
   PieChart, Pie, Cell, Legend,
 } from 'recharts'
 import type { AiVisualizationSpec } from '@/types'
-import { fieldToLabel, fmtNum, fmtQtyWithUnit, fmtFieldValue, fmtNumRaw, prepareVisualizationRows } from '@/lib/detectVisualization'
+import {
+  fieldToLabel,
+  formatChartMetricValue,
+  fmtFieldValue,
+  prepareVisualizationRows,
+} from '@/lib/detectVisualization'
 import { getDisplayColumns, isSummarizableMetricField } from '@shared/constants/ai-field-label'
 
 const COLORS = ['#D4A853', '#52b788', '#74c69d', '#e76f51', '#457b9d', '#e9c46a', '#a8dadc', '#f4a261']
@@ -18,8 +23,14 @@ interface Props {
   spec: AiVisualizationSpec
 }
 
-function tooltipFormatter(value: number | string, name: string): [number | string, string] {
-  return [typeof value === 'number' ? fmtNumRaw(value) : value, fieldToLabel(name)]
+function createMetricTickFormatter(fieldName: string, rows: Record<string, unknown>[]) {
+  return (value: number | string) => formatChartMetricValue(value, fieldName, rows)
+}
+
+function createTooltipFormatter(fieldName: string, rows: Record<string, unknown>[]) {
+  return (value: number | string, name: string): [string, string] => (
+    [formatChartMetricValue(value, fieldName, rows), fieldToLabel(name)]
+  )
 }
 
 function formatCellValue(field: string, value: unknown): string {
@@ -43,7 +54,7 @@ function Summary({ type, rows, spec }: { type: string; rows: Record<string, unkn
     const minIdx = vals.indexOf(minVal)
     const maxName = String(chartRows[maxIdx]?.[spec.xField] ?? '')
     const minName = String(chartRows[minIdx]?.[spec.xField] ?? '')
-    text = `共 ${rows.length} 项，${yLabel}合计 ${fmtQtyWithUnit(total, spec.yField!, rows[0] ?? {}, rows)}。最高「${maxName}」${fmtQtyWithUnit(maxVal, spec.yField!, rows[maxIdx] ?? {}, rows)}，最低「${minName}」${fmtQtyWithUnit(minVal, spec.yField!, rows[minIdx] ?? {}, rows)}。`
+    text = `共 ${rows.length} 项，${yLabel}合计 ${formatChartMetricValue(total, spec.yField!, rows)}。最高「${maxName}」${formatChartMetricValue(maxVal, spec.yField!, rows)}，最低「${minName}」${formatChartMetricValue(minVal, spec.yField!, rows)}。`
   }
 
   if (type === 'line' && spec.xField && spec.yField) {
@@ -53,7 +64,7 @@ function Summary({ type, rows, spec }: { type: string; rows: Record<string, unkn
     const avg = total / vals.length
     const firstX = String(chartRows[0]?.[spec.xField] ?? '')
     const lastX = String(chartRows[chartRows.length - 1]?.[spec.xField] ?? '')
-    text = `${firstX}→${lastX} 共 ${rows.length} 个周期，${yLabel}均值 ${fmtQtyWithUnit(avg, spec.yField!, rows[0] ?? {}, rows)}，合计 ${fmtQtyWithUnit(total, spec.yField!, rows[0] ?? {}, rows)}。`
+    text = `${firstX}→${lastX} 共 ${rows.length} 个周期，${yLabel}均值 ${formatChartMetricValue(avg, spec.yField!, rows)}，合计 ${formatChartMetricValue(total, spec.yField!, rows)}。`
   }
 
   if (type === 'pie' && spec.nameField && spec.valueField) {
@@ -65,7 +76,7 @@ function Summary({ type, rows, spec }: { type: string; rows: Record<string, unkn
     const total = data.reduce((s, d) => s + d.value, 0)
     const top = data[0]
     const topPct = total > 0 ? ((top.value / total) * 100).toFixed(1) : '0'
-    text = `共 ${data.length} 类，${yLabel}合计 ${fmtQtyWithUnit(total, spec.valueField!, rows[0] ?? {}, rows)}。占比最高「${top.name}」${topPct}%。`
+    text = `共 ${data.length} 类，${yLabel}合计 ${formatChartMetricValue(total, spec.valueField!, rows)}。占比最高「${top.name}」${topPct}%。`
   }
 
   if (type === 'table') {
@@ -76,7 +87,7 @@ function Summary({ type, rows, spec }: { type: string; rows: Record<string, unkn
       if (isSummarizableMetricField(col, rawValues)) {
         const vals = rawValues.map((value) => Number(value)).filter((v) => !isNaN(v) && v !== 0)
         const colTotal = vals.reduce((a, b) => a + b, 0)
-        numSummaries.push(`${fieldToLabel(col)} ${fmtQtyWithUnit(colTotal, col, rows[0] ?? {}, rows)}`)
+        numSummaries.push(`${fieldToLabel(col)} ${formatChartMetricValue(colTotal, col, rows)}`)
       }
     }
     text = `共 ${rows.length} 条记录。${numSummaries.length > 0 ? '合计：' + numSummaries.join('，') + '。' : ''}`
@@ -130,8 +141,8 @@ function BarViz({ rows, xField, yField }: { rows: Record<string, unknown>[]; xFi
       <BarChart data={data} margin={{ top: 6, right: 8, left: -16, bottom: 36 }}>
         <CartesianGrid strokeDasharray="3 3" stroke="hsl(var(--border))" />
         <XAxis dataKey={xField} tick={{ fontSize: 10 }} angle={-30} textAnchor="end" interval={0} />
-        <YAxis tick={{ fontSize: 10 }} />
-        <Tooltip formatter={tooltipFormatter} contentStyle={{ fontSize: 12 }} />
+        <YAxis tick={{ fontSize: 10 }} tickFormatter={createMetricTickFormatter(yField, rows)} />
+        <Tooltip formatter={createTooltipFormatter(yField, rows)} contentStyle={{ fontSize: 12 }} />
         <Bar dataKey={yField} name={fieldToLabel(yField)} fill="#D4A853" radius={[3, 3, 0, 0]} maxBarSize={40} />
       </BarChart>
     </ResponsiveContainer>
@@ -146,8 +157,8 @@ function LineViz({ rows, xField, yField }: { rows: Record<string, unknown>[]; xF
       <LineChart data={data} margin={{ top: 6, right: 8, left: -16, bottom: 36 }}>
         <CartesianGrid strokeDasharray="3 3" stroke="hsl(var(--border))" />
         <XAxis dataKey={xField} tick={{ fontSize: 10 }} angle={-30} textAnchor="end" interval={0} />
-        <YAxis tick={{ fontSize: 10 }} />
-        <Tooltip formatter={tooltipFormatter} contentStyle={{ fontSize: 12 }} />
+        <YAxis tick={{ fontSize: 10 }} tickFormatter={createMetricTickFormatter(yField, rows)} />
+        <Tooltip formatter={createTooltipFormatter(yField, rows)} contentStyle={{ fontSize: 12 }} />
         <Line
           type="monotone"
           dataKey={yField}
@@ -164,8 +175,8 @@ function LineViz({ rows, xField, yField }: { rows: Record<string, unknown>[]; xF
 
 // ─── 饼图 ─────────────────────────────────────────────────────────────────────
 function PieViz({ rows, nameField, valueField }: { rows: Record<string, unknown>[]; nameField: string; valueField: string }) {
-  const data = prepareVisualizationRows(rows, nameField)
-    .map((r) => ({ name: String(r[nameField] ?? ''), value: Number(r[valueField]) || 0 }))
+  const data = prepareVisualizationRows(rows, nameField, valueField)
+    .map((r) => ({ ...r, name: String(r[nameField] ?? ''), value: Number(r[valueField]) || 0 }))
     .filter((d) => d.value > 0)
   const total = data.reduce((s, d) => s + d.value, 0)
 
@@ -187,7 +198,7 @@ function PieViz({ rows, nameField, valueField }: { rows: Record<string, unknown>
             <Cell key={i} fill={COLORS[i % COLORS.length]} />
           ))}
         </Pie>
-        <Tooltip formatter={(v: number) => [fmtNum(v)]} />
+        <Tooltip formatter={(value: number | string) => [formatChartMetricValue(value, valueField, rows), fieldToLabel(valueField)]} />
         <Legend iconSize={10} wrapperStyle={{ fontSize: 11 }} />
       </PieChart>
     </ResponsiveContainer>

packages/server/src/modules/ai/ai-sql.service.ts

@@ -106,11 +106,31 @@ export class AiSqlService {
       nextSql = nextSql.replace(/join\s+customer\b/gi, 'LEFT JOIN customer');
     }
 
+    nextSql = this.rewriteCreatedAtToChinaTime(nextSql);
+
     return nextSql
       .replace(/\s+LIMIT\s+100\s+LIMIT\s+100/gi, ' LIMIT 100')
       .replace(/ORDER\s+BY\s+created_at\s+DESC\s+LIMIT\s+(\d+)\s+LIMIT\s+\1/gi, 'ORDER BY created_at DESC LIMIT $1')
   }
 
+  private rewriteCreatedAtToChinaTime(sql: string) {
+    let nextSql = sql;
+
+    // 数据库里的 created_at 目前按 UTC 存储，AI 常用 DATE(created_at)=DATE('now','localtime')
+    // 会把北京时间凌晨的订单误判成前一天，因此统一把 created_at 比较/展示口径转成北京时间。
+    nextSql = nextSql.replace(
+      /DATE\(\s*((?!datetime\()[^()]*?\bcreated_at\b[^()]*)\s*\)/gi,
+      (_matched, columnExpr: string) => `DATE(datetime(${columnExpr.trim()}, '+8 hours'))`,
+    );
+
+    nextSql = nextSql.replace(
+      /strftime\(\s*'(%Y-%m(?:-%d)?)'\s*,\s*((?!datetime\()[^()]*?\bcreated_at\b[^()]*)\s*\)/gi,
+      (_matched, format: string, columnExpr: string) => `strftime('${format}', datetime(${columnExpr.trim()}, '+8 hours'))`,
+    );
+
+    return nextSql;
+  }
+
   private normalizeRowDateTimes(row: Record<string, unknown>) {
     return Object.fromEntries(
       Object.entries(row).map(([key, value]) => [key, this.normalizeDateTimeValue(key, value)]),

packages/server/src/modules/ai/ai.service.ts

@@ -575,7 +575,7 @@ export class AiService implements OnModuleInit, OnModuleDestroy {
     const emit = (event: string, data: unknown) => emitter?.(event, data);
     // ── 1. 并行获取上下文 + AI 配置 ─────────────────────────────────────────
     const [structuredContext, availability] = await Promise.all([
-      useRecentStructuredContext ? this.getRecentStructuredContext(user.sub) : Promise.resolve({}),
+      useRecentStructuredContext ? this.getRecentStructuredContext(user.sub, currentSessionId) : Promise.resolve({}),
       this.aiConfigService.getAvailability(),
     ]);
 
@@ -2643,36 +2643,39 @@ export class AiService implements OnModuleInit, OnModuleDestroy {
     return '我刚刚没有成功查到这条数据。请换一种更具体的问法再试一次，比如带上订单号、客户名、供应商名或时间范围。';
   }
 
-  private async getRecentStructuredContext(userId: number): Promise<AiStructuredContext> {
+  private async getRecentStructuredContext(userId: number, sessionId?: string): Promise<AiStructuredContext> {
+    if (!sessionId) {
+      return {};
+    }
+
     const conversations = await this.aiConversationRepository.find({
-      where: { userId },
+      where: { userId, sessionId },
       order: { id: 'DESC' },
       take: 6,
     });
 
-    return conversations.reduce<AiStructuredContext>((merged, conversation) => {
-      if (!conversation.contextJson) {
-        return merged;
-      }
+    const latestConversationWithContext = conversations.find((conversation) => Boolean(conversation.contextJson?.trim()));
+    if (!latestConversationWithContext?.contextJson) {
+      return {};
+    }
 
-      try {
-        const context = JSON.parse(conversation.contextJson) as AiStructuredContext;
-        return {
-          orderNos: this.mergeContextList(merged.orderNos, context.orderNos),
-          returnNos: this.mergeContextList(merged.returnNos, context.returnNos),
-          refundNos: this.mergeContextList(merged.refundNos, context.refundNos),
-          exchangeNos: this.mergeContextList(merged.exchangeNos, context.exchangeNos),
-          customerNames: this.mergeContextList(merged.customerNames, context.customerNames),
-          customerContacts: this.mergeContextList(merged.customerContacts, context.customerContacts),
-          customerPhones: this.mergeContextList(merged.customerPhones, context.customerPhones),
-          supplierNames: this.mergeContextList(merged.supplierNames, context.supplierNames),
-          productNames: this.mergeContextList(merged.productNames, context.productNames),
-          reasonCodes: this.mergeContextList(merged.reasonCodes, context.reasonCodes),
-        };
-      } catch {
-        return merged;
-      }
-    }, {});
+    try {
+      const context = JSON.parse(latestConversationWithContext.contextJson) as AiStructuredContext;
+      return {
+        orderNos: context.orderNos?.slice(0, 6),
+        returnNos: context.returnNos?.slice(0, 6),
+        refundNos: context.refundNos?.slice(0, 6),
+        exchangeNos: context.exchangeNos?.slice(0, 6),
+        customerNames: context.customerNames?.slice(0, 6),
+        customerContacts: context.customerContacts?.slice(0, 6),
+        customerPhones: context.customerPhones?.slice(0, 6),
+        supplierNames: context.supplierNames?.slice(0, 6),
+        productNames: context.productNames?.slice(0, 6),
+        reasonCodes: context.reasonCodes?.slice(0, 6),
+      };
+    } catch {
+      return {};
+    }
   }
 
   private mergeContextList(current: string[] | undefined, next: string[] | undefined) {