@allbluecn/web-app 0.17.2 → 0.18.1

package/src/hooks/use-paged-groups.ts

@@ -0,0 +1,194 @@
+// SPDX-License-Identifier: AGPL-3.0-or-later
+// AllBlue - 理想之海
+// Copyright (C) 2026 AllBlue Contributors
+
+import { useCallback, useEffect, useMemo, useRef, useState } from "react";
+
+/**
+ * 服务端分组分页列表的共享状态机（usePagedStoryGroups / usePagedBugGroups 的公共底层）。
+ *
+ * 收口两域 hook 同构的部分：signature 防过期（参数变化整体重置、过期响应按 sig 丢弃）、
+ * 骨架最短展示（minLoadingMs）、组内取页的 loading/去重守卫、行内编辑乐观更新的
+ * 「sig 校验 → applyPatch → refresh/null/提交」三档派发。
+ *
+ * 域差异（分组查询形状、首屏脚手架、双层子组、patch 树操作、排序比较器）经回调注入，
+ * 由 use-paged-story-groups / use-paged-bug-groups 各自实现。
+ */
+export type PagedPatchResult<G> = G[] | "refresh" | null;
+
+export interface PagedPageResult<T> {
+  items: T[];
+  total?: number;
+  hasMore: boolean;
+}
+
+/** 取页目标容器的最小面（组或子组） */
+export interface PagedGroupContainer<T> {
+  items: T[];
+  total?: number;
+  hasMore?: boolean;
+  loading?: boolean;
+}
+
+export interface PagedGroupsCoreOptions<G, T, Req> {
+  /** 请求签名：变化即整体重置，过期响应按 sig 丢弃 */
+  sig: string;
+  /** false 时不发首屏请求（story 仅 list 视图启用；bug 恒 true） */
+  enabled?: boolean;
+  /** 首屏骨架最短展示时长（ms）；测试可传 0 跳过 */
+  minLoadingMs?: number;
+  /** 每次渲染的最新请求上下文（groupBy/pageSize/base 参数等），供回调经 ref 读取 */
+  request: Req;
+  /** 首屏加载：返回最终组节点数组；抛错时保留旧数据仅关 loading（提交时机与骨架最短展示由 core 掌握） */
+  loadInitial: () => Promise<G[]>;
+  /** 组内取页（skip 由 core 按已加载条数传入；take 从 request 读取） */
+  fetchPage: (req: Req, groupKey: string, subGroupKey: string | undefined, skip: number) => Promise<PagedPageResult<T>>;
+  /** 定位取页目标容器（组或子组）；未找到返回 null */
+  findContainer: (groups: G[], groupKey: string, subGroupKey: string | undefined) => PagedGroupContainer<T> | null;
+  /** 取页去重 key（story 双层为 `parent/sub`，bug 单层为 groupKey） */
+  containerLoadKey: (groupKey: string, subGroupKey: string | undefined) => string;
+  /** 容器 loading 状态切换（纯函数，返回新 groups） */
+  setLoading: (groups: G[], groupKey: string, subGroupKey: string | undefined, loading: boolean) => G[];
+  /** 取页成功合并（dedupe 与 total/totalCount 语义由域实现，纯函数） */
+  mergePage: (groups: G[], groupKey: string, subGroupKey: string | undefined, page: PagedPageResult<T>) => G[];
+  /** 行内编辑乐观更新（返回 "refresh" 时由 core 触发整体重置；null 表示未找到行） */
+  applyPatch: (groups: G[], req: Req, id: string, patch: unknown) => PagedPatchResult<G>;
+}
+
+/**
+ * @returns groups/isLoading 透传；loadMore 返回取页 Promise（调用方可据此解除
+ * loading 阻断，story 消费方忽略返回值行为不变）；dispatchPatch 供各域
+ * patchBug/patchStory 公开 API 转发。
+ */
+export function usePagedGroupsCore<G, T, Req>(
+  options: PagedGroupsCoreOptions<G, T, Req>,
+): {
+  groups: G[];
+  isLoading: boolean;
+  loadMore: (groupKey: string, subGroupKey?: string) => void | Promise<unknown>;
+  refresh: () => void;
+  dispatchPatch: (id: string, patch: unknown) => void;
+} {
+  const { sig, enabled = true, minLoadingMs: minLoadingMsOption, request } = options;
+
+  const [refreshTick, setRefreshTick] = useState(0);
+
+  const fullSig = useMemo(
+    () => JSON.stringify([sig, enabled, refreshTick]),
+    [sig, enabled, refreshTick],
+  );
+
+  const [state, setState] = useState<{ sig: string; groups: G[]; isLoading: boolean }>({
+    sig: "",
+    groups: [],
+    isLoading: true,
+  });
+
+  const stateRef = useRef(state);
+  useEffect(() => {
+    stateRef.current = state;
+  }, [state]);
+
+  // 回调经 ref 读取最新实现：loadMore/dispatchPatch 引用稳定（与两域原实现 [] deps 一致）
+  const optionsRef = useRef(options);
+  optionsRef.current = options;
+  const sigRef = useRef(fullSig);
+  sigRef.current = fullSig;
+
+  const loadingKeysRef = useRef<Set<string>>(new Set());
+
+  useEffect(() => {
+    if (!enabled) return;
+    let alive = true;
+    const startedAt = performance.now();
+    setState((s) => ({ ...s, isLoading: true }));
+
+    // 骨架最短展示：本地数据太快到达时保证加载动画至少完整起跳可见，消除「刚播一帧就消失」的闪烁
+    const minLoadingMs = minLoadingMsOption ?? 800;
+    const settleWithMinDelay = (settle: () => void) => {
+      const remaining = minLoadingMs - (performance.now() - startedAt);
+      if (remaining <= 0) {
+        settle();
+        return;
+      }
+      const timer = setTimeout(() => {
+        if (alive) settle();
+      }, remaining);
+      void timer;
+    };
+
+    (async () => {
+      try {
+        const groups = await optionsRef.current.loadInitial();
+        settleWithMinDelay(() => {
+          if (alive) setState({ sig: fullSig, groups, isLoading: false });
+        });
+      } catch {
+        settleWithMinDelay(() => {
+          if (alive) setState((s) => ({ ...s, isLoading: false }));
+        });
+      }
+    })();
+
+    return () => {
+      alive = false;
+    };
+    // 过滤/分组参数均编码进 fullSig，仅签名变化时重载
+  }, [fullSig]);
+
+  const loadMore = useCallback((groupKey: string, subGroupKey?: string): void | Promise<unknown> => {
+    const opts = optionsRef.current;
+    const currentSig = sigRef.current;
+    const snapshot = stateRef.current;
+    if (snapshot.sig !== currentSig) return;
+    const target = opts.findContainer(snapshot.groups, groupKey, subGroupKey);
+    if (!target || target.loading || !target.hasMore) return;
+    const loadKey = opts.containerLoadKey(groupKey, subGroupKey);
+    if (loadingKeysRef.current.has(loadKey)) return;
+    loadingKeysRef.current.add(loadKey);
+
+    setState((prev) => {
+      if (prev.sig !== currentSig) return prev;
+      return { ...prev, groups: opts.setLoading(prev.groups, groupKey, subGroupKey, true) };
+    });
+
+    // 返回取页 Promise：调用方（表格哨兵）据此判定何时解除 loading 阻断，
+    // 避免「首触后 loading 永久卡 true」的无限滚动死锁
+    return opts
+      .fetchPage(opts.request, groupKey, subGroupKey, target.items.length)
+      .then((page) => {
+        loadingKeysRef.current.delete(loadKey);
+        setState((prev) => {
+          if (prev.sig !== currentSig) return prev;
+          return { ...prev, groups: opts.mergePage(prev.groups, groupKey, subGroupKey, page) };
+        });
+      })
+      .catch(() => {
+        loadingKeysRef.current.delete(loadKey);
+        setState((prev) => {
+          if (prev.sig !== currentSig) return prev;
+          return { ...prev, groups: opts.setLoading(prev.groups, groupKey, subGroupKey, false) };
+        });
+      });
+  }, []);
+
+  const refresh = useCallback(() => setRefreshTick((t) => t + 1), []);
+
+  const dispatchPatch = useCallback((id: string, patch: unknown) => {
+    const opts = optionsRef.current;
+    const snapshot = stateRef.current;
+    if (snapshot.sig !== sigRef.current) {
+      refresh();
+      return;
+    }
+    const result = opts.applyPatch(snapshot.groups, opts.request, id, patch);
+    if (result === "refresh") {
+      refresh();
+      return;
+    }
+    if (result == null) return;
+    setState((prev) => (prev.sig !== snapshot.sig ? prev : { ...prev, groups: result }));
+  }, [refresh]);
+
+  return { groups: state.groups, isLoading: state.isLoading, loadMore, refresh, dispatchPatch };
+}

package/src/hooks/use-scrolling.ts

@@ -16,7 +16,7 @@
 // along with this program. If not, see <https://www.gnu.org/licenses/>.
 
 import type { RefObject } from "react"
-import { useCallback, useEffect, useState } from "react"
+import { useEffect, useState } from "react"
 import { useDebouncedCallback } from "@tanstack/react-pacer"
 
 type ScrollTarget = RefObject<HTMLElement> | Window | null | undefined

package/src/hooks/use-toolbar-search.ts

@@ -0,0 +1,51 @@
+// SPDX-License-Identifier: AGPL-3.0-or-later
+// AllBlue - 理想之海
+// Copyright (C) 2026 AllBlue Contributors
+
+import { useEffect, useRef, useState } from "react";
+import { SEARCH_DEBOUNCE_MS, useDebouncedValue } from "@/lib/pacer";
+
+interface UseToolbarSearchOptions {
+  /** 外部受控搜索值（如 filters.search），变化时同步回显输入框 */
+  externalValue: string | undefined;
+  /** 防抖窗口结束后提交 transform 后的值；与外部值恒等时跳过提交 */
+  onSearch: (value: string | undefined) => void;
+  /** 提交前的值规整（如 trim）；默认空串转 undefined */
+  transform?: (value: string) => string | undefined;
+}
+
+/** 模块级默认 transform，保持依赖稳定 */
+const defaultTransform = (value: string): string | undefined => value || undefined;
+
+/**
+ * 列表工具栏搜索框统一 hook：收口 bug/testcase/story 三处
+ * localSearch + 防抖 + 外部值同步回显的重复实现（防抖经 @tanstack/react-pacer）。
+ */
+export function useToolbarSearch({
+  externalValue,
+  onSearch,
+  transform = defaultTransform,
+}: UseToolbarSearchOptions) {
+  const [localSearch, setLocalSearch] = useState(externalValue ?? "");
+  const [debouncedSearch] = useDebouncedValue(localSearch, { wait: SEARCH_DEBOUNCE_MS });
+
+  const externalRef = useRef(externalValue);
+  externalRef.current = externalValue;
+  const onSearchRef = useRef(onSearch);
+  onSearchRef.current = onSearch;
+
+  useEffect(() => {
+    setLocalSearch(externalValue ?? "");
+  }, [externalValue]);
+
+  useEffect(() => {
+    if (debouncedSearch === (externalRef.current ?? "")) return;
+    onSearchRef.current(transform(debouncedSearch));
+  }, [debouncedSearch, transform]);
+
+  const handleSearchChange = (value: string) => {
+    setLocalSearch(value);
+  };
+
+  return { localSearch, handleSearchChange };
+}

package/src/lib/__tests__/annotation-prompt.test.ts

@@ -24,8 +24,7 @@ vi.mock("@allbluecn/database", () => ({
 
 import { prisma } from "@allbluecn/database"
 import "@/server/ai-gateway/prompts/builtin"
-import { resolveSystemPrompt } from "@/server/ai-gateway/prompts/registry"
-import { buildFillPrompt, type FillComponentInfo } from "../../plugins/core/prd/lib/ai/annotation-prompt"
+import { buildFillPrompt, type FillComponentInfo } from "../../server/design/annotation-prompt"
 
 beforeEach(() => {
   vi.mocked(prisma.systemPromptTemplate.findUnique).mockResolvedValue(null as never)

package/src/lib/__tests__/datetime-local.test.ts

@@ -0,0 +1,86 @@
+// SPDX-License-Identifier: AGPL-3.0-or-later
+// AllBlue - 理想之海
+// Copyright (C) 2026 AllBlue Contributors
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program. If not, see <https://www.gnu.org/licenses/>.
+
+import { describe, it, expect } from 'vitest'
+import {
+  getMonthGrid,
+  isSameDay,
+  formatDateLocal,
+  parseDate,
+  buildTimelineRange,
+  getRangeFromDates,
+  getWeekGrid,
+} from '../datetime'
+
+describe('datetime 本地日历工具', () => {
+  it('getMonthGrid 返回 42 格且 2026-08 首日是周六', () => {
+    const grid = getMonthGrid(2026, 7) // 0-based month
+    expect(grid).toHaveLength(42)
+    expect(formatDateLocal(grid.find((g) => g.inMonth && g.date.getDate() === 1)!.date)).toBe(
+      '2026-08-01',
+    )
+  })
+
+  it('isSameDay 比较日期', () => {
+    expect(isSameDay(new Date(2026, 7, 1), new Date(2026, 7, 1))).toBe(true)
+    expect(isSameDay(new Date(2026, 7, 1), new Date(2026, 7, 2))).toBe(false)
+  })
+
+  it('formatDate/parseDate 往返', () => {
+    const d = parseDate('2026-08-15')
+    expect(formatDateLocal(d)).toBe('2026-08-15')
+  })
+
+  it('parseDate 兼容 ISO 字符串', () => {
+    const d = parseDate('2026-08-15T00:00:00.000Z')
+    expect(formatDateLocal(d)).toBe('2026-08-15')
+  })
+
+  it('buildTimelineRange 月模式生成月度刻度', () => {
+    const range = buildTimelineRange(new Date(2026, 0, 1), new Date(2026, 2, 1), 'month')
+    expect(range).toHaveLength(3)
+    expect(range[0].label).toBe('2026-01')
+  })
+
+  it('buildTimelineRange 周模式 label 为纯日期且 7 天递增', () => {
+    const range = buildTimelineRange(new Date(2026, 0, 1), new Date(2026, 1, 1), 'week')
+    expect(range[0].label).toBe('2026-01-01')
+    expect(range[1].label).toBe('2026-01-08')
+    expect(range.every((t) => /^\d{4}-\d{2}-\d{2}$/.test(t.label))).toBe(true)
+  })
+
+  it('getRangeFromDates 覆盖最早最晚日期', () => {
+    const { start, end } = getRangeFromDates([
+      new Date(2026, 5, 1),
+      new Date(2026, 7, 15),
+    ])
+    expect(start.getFullYear()).toBe(2026)
+    expect(end.getMonth()).toBe(7)
+  })
+
+  it('getWeekGrid 返回以周日开始的 7 天', () => {
+    const grid = getWeekGrid(2026, 7, 18) // 2026-08-18 周二
+    expect(grid).toHaveLength(7)
+    expect(formatDateLocal(grid[0].date)).toBe('2026-08-16') // 周日
+    expect(formatDateLocal(grid[6].date)).toBe('2026-08-22') // 周六
+  })
+
+  it('getWeekGrid 跨月时 inMonth 标记正确', () => {
+    const grid = getWeekGrid(2026, 6, 31) // 2026-07-31 周五，跨 8 月
+    expect(grid[6].inMonth).toBe(false)
+  })
+})

package/src/lib/__tests__/pgvector.test.ts

@@ -30,22 +30,22 @@ beforeEach(() => vi.clearAllMocks())
 describe('searchByVector', () => {
   it('builds correct pgvector similarity search SQL', async () => {
     mockQueryRawUnsafe.mockResolvedValue([])
-    await searchByVector([0.1, 0.2, 0.3], 'prd_pages', 5)
+    await searchByVector([0.1, 0.2, 0.3], 'design_pages', 5)
     expect(mockQueryRawUnsafe).toHaveBeenCalledOnce()
     const sql = mockQueryRawUnsafe.mock.calls[0][0] as string
     expect(sql).toContain('<=>')
-    expect(sql).toContain('"prd_pages"')
+    expect(sql).toContain('"design_pages"')
   })
 
   it('passes default limit of 10', async () => {
     mockQueryRawUnsafe.mockResolvedValue([])
-    await searchByVector([0.1], 'prd_pages')
+    await searchByVector([0.1], 'design_pages')
     expect(mockQueryRawUnsafe.mock.calls[0][3]).toBe(10)
   })
 
   it('converts embedding array to bracket format', async () => {
     mockQueryRawUnsafe.mockResolvedValue([])
-    await searchByVector([0.1, 0.2], 'prd_pages')
+    await searchByVector([0.1, 0.2], 'design_pages')
     const embStr = mockQueryRawUnsafe.mock.calls[0][1] as string
     expect(embStr).toBe('[0.1,0.2]')
   })
@@ -57,24 +57,24 @@ describe('searchByVector', () => {
 
   it('accepts allowed table names', async () => {
     mockQueryRawUnsafe.mockResolvedValue([])
-    await expect(searchByVector([0.1], 'prd_pages')).resolves.toEqual([])
+    await expect(searchByVector([0.1], 'design_pages')).resolves.toEqual([])
   })
 })
 
 describe('insertEmbedding', () => {
   it('builds correct INSERT SQL without extraFields', async () => {
     mockQueryRawUnsafe.mockResolvedValue(undefined)
-    await insertEmbedding('prd_pages', 'doc-1', 'some content', [0.1, 0.2])
+    await insertEmbedding('design_pages', 'doc-1', 'some content', [0.1, 0.2])
     const sql = mockQueryRawUnsafe.mock.calls[0][0] as string
-    expect(sql).toContain('INSERT INTO "prd_pages"')
+    expect(sql).toContain('INSERT INTO "design_pages"')
     expect(mockQueryRawUnsafe.mock.calls[0][1]).toBe('doc-1')
   })
 
   it('includes extra fields using parameterized placeholders', async () => {
     mockQueryRawUnsafe.mockResolvedValue(undefined)
-    await insertEmbedding('prd_pages', 'id-1', 'c', [0.5], { prdPageId: 'page-1', chunkIndex: 0 })
+    await insertEmbedding('design_pages', 'id-1', 'c', [0.5], { designPageId: 'page-1', chunkIndex: 0 })
     const sql = mockQueryRawUnsafe.mock.calls[0][0] as string
-    expect(sql).toContain('"prdPageId"')
+    expect(sql).toContain('"designPageId"')
     expect(sql).toContain('$1')
     expect(sql).toContain('$5::vector')
   })
@@ -87,7 +87,7 @@ describe('insertEmbedding', () => {
   it('extraFields values are not concatenated into SQL string', async () => {
     mockQueryRawUnsafe.mockResolvedValue(undefined)
     const maliciousValue = "'); DROP TABLE users; --"
-    await insertEmbedding('prd_pages', 'id', 'c', [0.1], { name: maliciousValue })
+    await insertEmbedding('design_pages', 'id', 'c', [0.1], { name: maliciousValue })
     const sql = mockQueryRawUnsafe.mock.calls[0][0] as string
     expect(sql).not.toContain(maliciousValue)
   })

package/src/lib/__tests__/provider-router.test.ts

@@ -17,15 +17,15 @@
 
 import { describe, it, expect, vi, beforeEach, beforeAll } from 'vitest'
 
-const { mockConfigFindMany, mockProviderFindUnique } = vi.hoisted(() => ({
+const { mockConfigFindMany, mockProviderFindMany } = vi.hoisted(() => ({
   mockConfigFindMany: vi.fn(),
-  mockProviderFindUnique: vi.fn(),
+  mockProviderFindMany: vi.fn(),
 }))
 
 vi.mock('@allbluecn/database', () => ({
   prisma: {
     aiProviderConfig: { findMany: mockConfigFindMany },
-    aiProvider: { findUnique: mockProviderFindUnique },
+    aiProvider: { findMany: mockProviderFindMany },
   },
 }))
 
@@ -93,7 +93,7 @@ describe('resolveDefaultProviderConfig', () => {
         models: ['gpt-4o'],
       },
     ])
-    mockProviderFindUnique.mockResolvedValue(provider)
+    mockProviderFindMany.mockResolvedValue([provider])
 
     const cfg = await resolveDefaultProviderConfig('user-1')
     expect(cfg).toEqual({
@@ -110,7 +110,7 @@ describe('resolveDefaultProviderConfig', () => {
     mockConfigFindMany.mockResolvedValue([
       { providerId: 'openai', enabled: true, apiKey: null, models: ['gpt-4o'] },
     ])
-    mockProviderFindUnique.mockResolvedValue(provider)
+    mockProviderFindMany.mockResolvedValue([provider])
 
     const cfg = await resolveDefaultProviderConfig('user-1')
     expect(cfg).toBeNull()
@@ -125,7 +125,7 @@ describe('resolveDefaultProviderConfig', () => {
         models: ['x'],
       },
     ])
-    mockProviderFindUnique.mockResolvedValue(null)
+    mockProviderFindMany.mockResolvedValue([])
 
     const cfg = await resolveDefaultProviderConfig('user-1')
     expect(cfg).toBeNull()
@@ -140,7 +140,7 @@ describe('resolveDefaultProviderConfig', () => {
         models: [],
       },
     ])
-    mockProviderFindUnique.mockResolvedValue(provider)
+    mockProviderFindMany.mockResolvedValue([provider])
 
     const cfg = await resolveDefaultProviderConfig('user-1')
     expect(cfg?.model).toBe('gpt-4o')
@@ -150,6 +150,8 @@ describe('resolveDefaultProviderConfig', () => {
     mockConfigFindMany.mockResolvedValue([])
     const cfg = await resolveDefaultProviderConfig('user-1')
     expect(cfg).toBeNull()
+    // 短路：不发起 provider 预取
+    expect(mockProviderFindMany).not.toHaveBeenCalled()
   })
 })
 
@@ -164,7 +166,7 @@ describe('resolveAiConfigForModel', () => {
         models: ['gpt-4o'],
       },
     ])
-    mockProviderFindUnique.mockResolvedValue(provider)
+    mockProviderFindMany.mockResolvedValue([provider])
 
     const cfg = await resolveAiConfigForModel('user-1', 'gpt-4o')
     expect(cfg).toEqual({
@@ -175,6 +177,12 @@ describe('resolveAiConfigForModel', () => {
       model: 'gpt-4o',
       isLocal: false,
     })
+    // DB-01 契约：provider 元数据一次 findMany in 预取（替代逐 config findUnique），select 白名单
+    expect(mockProviderFindMany).toHaveBeenCalledTimes(1)
+    expect(mockProviderFindMany).toHaveBeenCalledWith({
+      where: { providerId: { in: ['openai'] } },
+      select: { providerId: true, adapterKind: true, models: true, isLocal: true },
+    })
   })
 
   it('returns null when model not in enabledModels', async () => {
@@ -186,7 +194,7 @@ describe('resolveAiConfigForModel', () => {
         models: ['gpt-4o'],
       },
     ])
-    mockProviderFindUnique.mockResolvedValue(provider)
+    mockProviderFindMany.mockResolvedValue([provider])
 
     const cfg = await resolveAiConfigForModel('user-1', 'claude-3')
     expect(cfg).toBeNull()
@@ -201,7 +209,7 @@ describe('resolveAiConfigForModel', () => {
         models: [],
       },
     ])
-    mockProviderFindUnique.mockResolvedValue(provider)
+    mockProviderFindMany.mockResolvedValue([provider])
 
     const cfg = await resolveAiConfigForModel('user-1', 'gpt-4o-mini')
     expect(cfg?.model).toBe('gpt-4o-mini')
@@ -220,7 +228,7 @@ describe("resolveDefaultProviderConfig — 加密 apiKey", () => {
         models: ["gpt-4o"],
       },
     ])
-    mockProviderFindUnique.mockResolvedValue(provider)
+    mockProviderFindMany.mockResolvedValue([provider])
 
     const cfg = await resolveDefaultProviderConfig("user-1")
     expect(cfg?.apiKey).toBe("sk-encrypted-real")
@@ -236,7 +244,7 @@ describe("resolveDefaultProviderConfig — 加密 apiKey", () => {
         models: ["gpt-4o"],
       },
     ])
-    mockProviderFindUnique.mockResolvedValue(provider)
+    mockProviderFindMany.mockResolvedValue([provider])
 
     const cfg = await resolveDefaultProviderConfig("user-2")
     expect(cfg).toBeNull()

package/src/lib/__tests__/rag.test.ts

@@ -74,13 +74,13 @@ describe('indexDocument', () => {
     expect(mockInsertEmbedding.mock.calls[0][0]).toBe('prototype_embeddings')
   })
 
-  it('passes extra fields with prdPageId and chunkIndex', async () => {
+  it('passes extra fields with designPageId and chunkIndex', async () => {
     mockInsertEmbedding.mockResolvedValue(undefined)
 
     await indexDocument('doc-42', 'Content.')
 
     const extraFields = mockInsertEmbedding.mock.calls[0][4] as Record<string, unknown>
-    expect(extraFields.prdPageId).toBe('doc-42')
+    expect(extraFields.designPageId).toBe('doc-42')
     expect(extraFields.chunkIndex).toBe(0)
   })
 })

package/src/lib/ai/use-search-capability.ts

@@ -17,7 +17,7 @@
 
 import { useMemo } from "react"
 import { useQuery } from "@tanstack/react-query"
-import { listAiProvidersFn } from "@/server/serverFns/ai/providers"
+import { listAiProvidersFn } from "@/server/serverFns/design/ai/providers"
 import { getSearchCapabilityFn } from "@/server/serverFns/ai/search-capability"
 
 const NATIVE_SEARCH_KINDS = new Set([