@allbluecn/web-app 0.12.1 → 0.13.0

package/src/server/serverFns/__tests__/ai-source-mode.test.ts

@@ -0,0 +1,119 @@
+// @vitest-environment node
+import { describe, expect, it, vi, beforeEach } from 'vitest'
+
+// ① createServerFn chain builder mock（照抄 workspace-kb-idor.test.ts:25-45）
+vi.mock('@tanstack/react-start', () => {
+  const createServerFn = () => {
+    const api: any = {
+      validator(schema: { parse: (d: unknown) => unknown }) {
+        api._validator = schema
+        return api
+      },
+      handler(fn: Function) {
+        return async (args: { data: unknown }) => {
+          let data = args?.data
+          if (api._validator && data !== undefined) {
+            data = api._validator.parse ? api._validator.parse(data) : data
+          }
+          return fn({ data })
+        }
+      },
+    }
+    return api
+  }
+  return { createServerFn }
+})
+vi.mock('@tanstack/react-start/server', () => ({
+  getRequest: () => ({}),
+}))
+
+// ② seam mock（照抄 tags.test.ts:92-120）
+const mocks = vi.hoisted(() => ({
+  currentSession: vi.fn(),
+}))
+
+vi.mock('@allbluecn/kernel', async (importOriginal) => {
+  const actual = await importOriginal<typeof import('@allbluecn/kernel')>()
+  const UnauthorizedError = class UnauthorizedError extends Error {
+    constructor(message = 'UNAUTHORIZED') { super(message); this.name = 'UnauthorizedError' }
+  }
+  return {
+    ...actual,
+    getServerSeam: () => ({
+      requireAuth: async () => {
+        const session = await mocks.currentSession()
+        if (!session?.user?.id) throw new UnauthorizedError()
+        return session
+      },
+    }),
+  }
+})
+
+// ③ source-mode mock（需在 hoisted 中定义以避免提升问题）
+const sourceMocks = vi.hoisted(() => ({
+  resolveSourceMode: vi.fn(),
+  setSourceMode: vi.fn(),
+}))
+
+vi.mock('@/server/ai-gateway/source-mode', () => ({
+  resolveSourceMode: sourceMocks.resolveSourceMode,
+  setSourceMode: sourceMocks.setSourceMode,
+}))
+
+import { setAiSourceModeFn, getAiSourceModeFn } from '@/plugins/core/prd/serverFns/ai/source-mode'
+
+describe('setAiSourceModeFn 权限闸', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+    delete process.env.DEPLOY_TARGET
+  })
+
+  it('hosted 模式仅 SUPER_ADMIN 可设（session.user.role）', async () => {
+    mocks.currentSession.mockResolvedValue({ user: { id: 'u1', role: 'USER' } })
+    await expect((setAiSourceModeFn as (args: unknown) => Promise<unknown>)({ data: { mode: 'hosted' } })).rejects.toThrow(/超级管理员|Forbidden/)
+    expect(sourceMocks.setSourceMode).not.toHaveBeenCalled()
+  })
+
+  it('SUPER_ADMIN 可设 hosted', async () => {
+    mocks.currentSession.mockResolvedValue({ user: { id: 'u1', role: 'SUPER_ADMIN' } })
+    await (setAiSourceModeFn as (args: unknown) => Promise<unknown>)({ data: { mode: 'hosted' } })
+    expect(sourceMocks.setSourceMode).toHaveBeenCalledWith({ userId: 'u1', isGuest: false }, 'hosted')
+  })
+
+  it('local-agent 在云端部署被拒（需桌面版）', async () => {
+    mocks.currentSession.mockResolvedValue({ user: { id: 'u1', role: 'USER' } })
+    await expect((setAiSourceModeFn as (args: unknown) => Promise<unknown>)({ data: { mode: 'local-agent' } })).rejects.toThrow(/桌面版|desktop/)
+    expect(sourceMocks.setSourceMode).not.toHaveBeenCalled()
+  })
+
+  it('local-agent 在 desktop 部署放行（含游客）', async () => {
+    process.env.DEPLOY_TARGET = 'desktop'
+    mocks.currentSession.mockResolvedValue({ user: { id: 'local-guest', role: 'GUEST' } })
+    await (setAiSourceModeFn as (args: unknown) => Promise<unknown>)({ data: { mode: 'local-agent' } })
+    expect(sourceMocks.setSourceMode).toHaveBeenCalledWith({ userId: 'local-guest', isGuest: true }, 'local-agent')
+  })
+
+  it('byok 任意登录态可设；未登录拒绝', async () => {
+    mocks.currentSession.mockResolvedValue({ user: { id: 'u1', role: 'USER' } })
+    await (setAiSourceModeFn as (args: unknown) => Promise<unknown>)({ data: { mode: 'byok' } })
+    expect(sourceMocks.setSourceMode).toHaveBeenCalled()
+    mocks.currentSession.mockResolvedValue({ user: null })
+    await expect((setAiSourceModeFn as (args: unknown) => Promise<unknown>)({ data: { mode: 'byok' } })).rejects.toThrow()
+  })
+})
+
+describe('getAiSourceModeFn', () => {
+  beforeEach(() => vi.clearAllMocks())
+
+  it('返回当前 mode + canHosted', async () => {
+    mocks.currentSession.mockResolvedValue({ user: { id: 'u1', role: 'SUPER_ADMIN' } })
+    sourceMocks.resolveSourceMode.mockResolvedValue('byok')
+    const result = await (getAiSourceModeFn as (args: unknown) => Promise<unknown>)({})
+    expect(result).toEqual({ mode: 'byok', canHosted: true })
+  })
+
+  it('未登录抛错', async () => {
+    mocks.currentSession.mockResolvedValue({ user: null })
+    await expect((getAiSourceModeFn as (args: unknown) => Promise<unknown>)({})).rejects.toThrow(/未登录|UNAUTHORIZED/)
+  })
+})

package/src/server/serverFns/__tests__/harness-plugins.test.ts

@@ -0,0 +1,140 @@
+// 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 { precheckPluginCode } from "../plugins/plugin-precheck";
+
+describe("precheckPluginCode", () => {
+  it("accepts valid plugin with both host and client code", () => {
+    const result = precheckPluginCode({
+      name: "my-plugin",
+      purpose: "A test plugin",
+      codeHost: "function apply(ctx) {}",
+      codeClient: "function apply(ctx) {}",
+    });
+    expect(result.ok).toBe(true);
+    expect(result.errors).toHaveLength(0);
+  });
+
+  it("accepts valid plugin with only host code", () => {
+    const result = precheckPluginCode({
+      name: "my-plugin",
+      purpose: "A test plugin",
+      codeHost: "function apply(ctx) {}",
+    });
+    expect(result.ok).toBe(true);
+  });
+
+  it("rejects empty name", () => {
+    const result = precheckPluginCode({
+      name: "",
+      purpose: "A test plugin",
+      codeHost: "function apply(ctx) {}",
+    });
+    expect(result.ok).toBe(false);
+    expect(result.errors).toContain("name is required");
+  });
+
+  it("rejects name over 64 characters", () => {
+    const result = precheckPluginCode({
+      name: "a".repeat(65),
+      purpose: "A test plugin",
+      codeHost: "function apply(ctx) {}",
+    });
+    expect(result.ok).toBe(false);
+    expect(result.errors).toContain("name must be at most 64 characters");
+  });
+
+  it("rejects empty purpose", () => {
+    const result = precheckPluginCode({
+      name: "my-plugin",
+      purpose: "",
+      codeHost: "function apply(ctx) {}",
+    });
+    expect(result.ok).toBe(false);
+    expect(result.errors).toContain("purpose is required");
+  });
+
+  it("rejects purpose over 512 characters", () => {
+    const result = precheckPluginCode({
+      name: "my-plugin",
+      purpose: "a".repeat(513),
+      codeHost: "function apply(ctx) {}",
+    });
+    expect(result.ok).toBe(false);
+    expect(result.errors).toContain("purpose must be at most 512 characters");
+  });
+
+  it("rejects when both codeHost and codeClient are empty", () => {
+    const result = precheckPluginCode({
+      name: "my-plugin",
+      purpose: "A test plugin",
+    });
+    expect(result.ok).toBe(false);
+    expect(result.errors).toContain(
+      "at least one of codeHost or codeClient is required"
+    );
+  });
+
+  it("rejects codeHost with syntax error", () => {
+    const result = precheckPluginCode({
+      name: "my-plugin",
+      purpose: "A test plugin",
+      codeHost: "function {{{ invalid syntax",
+    });
+    expect(result.ok).toBe(false);
+    expect(result.errors.some((e) => e.includes("codeHost syntax error"))).toBe(
+      true
+    );
+  });
+
+  it("rejects codeClient with syntax error", () => {
+    const result = precheckPluginCode({
+      name: "my-plugin",
+      purpose: "A test plugin",
+      codeClient: "function {{{ invalid syntax",
+    });
+    expect(result.ok).toBe(false);
+    expect(result.errors.some((e) => e.includes("codeClient syntax error"))).toBe(
+      true
+    );
+  });
+
+  it("rejects codeHost over 64000 characters", () => {
+    const result = precheckPluginCode({
+      name: "my-plugin",
+      purpose: "A test plugin",
+      codeHost: "a".repeat(64_001),
+    });
+    expect(result.ok).toBe(false);
+    expect(result.errors).toContain(
+      "codeHost must be at most 64000 characters"
+    );
+  });
+
+  it("rejects codeClient over 64000 characters", () => {
+    const result = precheckPluginCode({
+      name: "my-plugin",
+      purpose: "A test plugin",
+      codeClient: "a".repeat(64_001),
+    });
+    expect(result.ok).toBe(false);
+    expect(result.errors).toContain(
+      "codeClient must be at most 64000 characters"
+    );
+  });
+});

package/src/server/serverFns/__tests__/knowledge-index.test.ts

@@ -53,9 +53,12 @@ vi.mock('@allbluecn/database', () => ({
   },
 }))
 
-vi.mock('@/server/ai-gateway', () => ({
+vi.mock('@/server/ai-gateway/embedding', () => ({
   generateEmbedding: vi.fn(),
   chunkText: vi.fn((t: string) => [t]),
+}))
+
+vi.mock('@/server/ai-gateway/provider-router', () => ({
   resolveDefaultProviderConfig: (...args: unknown[]) => m.resolveDefaultProviderConfig(...args),
 }))
 
@@ -63,9 +66,12 @@ vi.mock('@/server/auth-helpers', () => ({
   requireAuth: vi.fn(),
 }))
 
-vi.mock('@/lib/pgvector', () => ({
-  insertEmbedding: vi.fn(),
-  deleteEmbeddings: vi.fn(),
+vi.mock('@/lib/vector-store', () => ({
+  getVectorStore: vi.fn().mockResolvedValue({
+    insertEmbedding: vi.fn(),
+    deleteEmbeddings: vi.fn(),
+    searchByVector: vi.fn(),
+  }),
 }))
 
 import { extractText, resolveAiConfig } from '../knowledge-index-utils'
@@ -173,6 +179,38 @@ describe('extractText', () => {
     expect(extractText({ markdown: '# 标题' })).toBe('# 标题')
     expect(extractText({ html: '<p>正文</p>' })).toBe('正文')
     expect(extractText({})).toBe('')
+  })
+
+  it('从 TipTap JSON doc 提取文本', () => {
+    const doc = {
+      type: 'doc',
+      content: [
+        { type: 'paragraph', content: [{ type: 'text', text: 'Hello' }] },
+        { type: 'heading', attrs: { level: 1 }, content: [{ type: 'text', text: 'World' }] },
+      ],
+    }
+    expect(extractText({ doc })).toBe('HelloWorld')
+  })
+
+  it('从 content.text 提取', () => {
+    expect(extractText({ text: 'plain text' })).toBe('plain text')
+  })
+
+  it('从字符串 content 提取', () => {
+    expect(extractText('raw string')).toBe('raw string')
+  })
+
+  it('直接从 ProseMirror doc 提取', () => {
+    const doc = { type: 'doc', content: [{ type: 'text', text: 'Direct' }] }
+    expect(extractText(doc)).toBe('Direct')
+  })
+
+  it('markdown 优先级最高', () => {
+    expect(extractText({ markdown: '# Title', html: '<h1>Title</h1>' })).toBe('# Title')
+  })
+
+  it('null/undefined 返回空字符串', () => {
     expect(extractText(null)).toBe('')
+    expect(extractText(undefined)).toBe('')
   })
 })

package/src/server/serverFns/__tests__/knowledge-search.test.ts

@@ -0,0 +1,262 @@
+// 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, vi, beforeEach } from 'vitest'
+
+vi.mock('@tanstack/react-start', () => ({
+  createServerFn: (_config: unknown) => {
+    const api: {
+      _validator: unknown
+      validator(schema: unknown): typeof api
+      handler(fn: (args: { data: unknown }) => Promise<unknown>): (args: unknown) => Promise<unknown>
+    } = {
+      _validator: undefined,
+      validator(schema: unknown) {
+        this._validator = schema
+        return this
+      },
+      handler(fn: (args: { data: unknown }) => Promise<unknown>) {
+        return async (args: unknown) => fn({ data: (args as { data?: unknown } | null | undefined)?.data })
+      },
+    }
+    return api
+  },
+}))
+
+vi.mock('@tanstack/react-start/server', () => ({
+  getRequest: vi.fn(),
+}))
+
+const m = vi.hoisted(() => {
+  const prismaMock = {
+    knowledgeBase: { findMany: vi.fn() },
+    knowledgeDocument: { findMany: vi.fn() },
+    articleFavorite: { findMany: vi.fn() },
+    legalRegulation: { findMany: vi.fn() },
+    industryCompliance: { findMany: vi.fn() },
+    dictionaryTerm: { findMany: vi.fn() },
+    websiteBookmark: { findMany: vi.fn() },
+  }
+  // serverFn handler 内是动态 import('@allbluecn/database')——inline deps 下
+  // vi.mock 拦不到该形态（说明符经 vite 改写后 miss registry），会加载真实
+  // 模块。借 database 包自身的 globalThis.prisma 单例缓存注入 mock：真实
+  // index.ts 顶层 globalForPrisma.prisma ?? await createPrismaClient() 命中
+  // 注入值即跳过连库，行为与 vi.mock 完全一致。
+  ;(globalThis as { prisma?: unknown }).prisma = prismaMock
+  return prismaMock
+})
+
+vi.mock('@allbluecn/database', () => ({
+  prisma: {
+    knowledgeBase: m.knowledgeBase,
+    knowledgeDocument: m.knowledgeDocument,
+    articleFavorite: m.articleFavorite,
+    legalRegulation: m.legalRegulation,
+    industryCompliance: m.industryCompliance,
+    dictionaryTerm: m.dictionaryTerm,
+    websiteBookmark: m.websiteBookmark,
+  },
+}))
+
+const ai = vi.hoisted(() => ({
+  generateEmbedding: vi.fn(),
+  rerankCandidates: vi.fn(),
+  RECALL_MULTIPLIER: 4,
+}))
+
+vi.mock('@/server/ai-gateway/embedding', () => ({
+  generateEmbedding: ai.generateEmbedding,
+  chunkText: vi.fn((t: string) => [t]),
+}))
+
+vi.mock('@/server/ai-gateway/rerank', () => ({
+  rerankCandidates: ai.rerankCandidates,
+  RECALL_MULTIPLIER: ai.RECALL_MULTIPLIER,
+}))
+
+vi.mock('@/server/auth-helpers', () => ({
+  requireAuth: vi.fn(),
+}))
+
+vi.mock('@/server/permissions', () => ({
+  requirePermission: vi.fn(),
+}))
+
+vi.mock('@/server/audit', () => ({
+  auditLog: vi.fn(),
+}))
+
+const vector = vi.hoisted(() => ({
+  searchByVector: vi.fn(),
+  insertEmbedding: vi.fn(),
+  deleteEmbeddings: vi.fn(),
+}))
+
+vi.mock('@/lib/vector-store', () => ({
+  getVectorStore: vi.fn(async () => ({
+    searchByVector: vector.searchByVector,
+    insertEmbedding: vector.insertEmbedding,
+    deleteEmbeddings: vector.deleteEmbeddings,
+  })),
+}))
+
+const ki = vi.hoisted(() => ({
+  resolveAiConfig: vi.fn(),
+  KB_EMBEDDINGS_TABLE: 'knowledge_embeddings' as const,
+  indexDocumentById: vi.fn(),
+}))
+
+vi.mock('@/server/serverFns/knowledge-index-utils', () => ({
+  resolveAiConfig: ki.resolveAiConfig,
+  KB_EMBEDDINGS_TABLE: ki.KB_EMBEDDINGS_TABLE,
+  indexDocumentById: ki.indexDocumentById,
+  extractText: vi.fn(),
+}))
+
+import { searchDocumentsFn } from '../knowledge'
+import { requireAuth } from '@/server/auth-helpers'
+
+const SESSION = { user: { id: 'user-1' } } as unknown as Awaited<ReturnType<typeof requireAuth>>
+
+describe('searchDocumentsFn 向量检索 + 三层降级', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+    vi.mocked(requireAuth).mockResolvedValue(SESSION)
+    m.knowledgeBase.findMany.mockResolvedValue([{ id: 'kb-1', name: '我的知识库' }])
+    m.articleFavorite.findMany.mockResolvedValue([])
+    m.legalRegulation.findMany.mockResolvedValue([])
+    m.industryCompliance.findMany.mockResolvedValue([])
+    m.dictionaryTerm.findMany.mockResolvedValue([])
+    m.websiteBookmark.findMany.mockResolvedValue([])
+  })
+
+  it('有 AI key 时走向量检索 + rerank', async () => {
+    ki.resolveAiConfig.mockResolvedValue({ apiKey: 'sk-xxx', baseUrl: 'https://api.openai.com/v1' })
+    ai.generateEmbedding.mockResolvedValue([0.1, 0.2, 0.3])
+    vector.searchByVector.mockResolvedValue([
+      { id: 'doc-1-0', content: 'chunk text', similarity: 0.9 },
+    ])
+    ai.rerankCandidates.mockResolvedValue([
+      { id: 'doc-1-0', content: 'chunk text', similarity: 0.95 },
+    ])
+    m.knowledgeDocument.findMany.mockResolvedValue([
+      { id: 'doc-1', title: '文档1', kbId: 'kb-1', type: 'doc', icon: null },
+    ])
+
+    const result = await searchDocumentsFn({ data: { query: 'test', currentKbId: 'kb-1' } })
+
+    expect(ki.resolveAiConfig).toHaveBeenCalledWith('kb-1')
+    expect(ai.generateEmbedding).toHaveBeenCalledWith('test', { apiKey: 'sk-xxx', baseUrl: 'https://api.openai.com/v1' })
+    expect(vector.searchByVector).toHaveBeenCalledWith(
+      [0.1, 0.2, 0.3],
+      'knowledge_embeddings',
+      40,
+      'kbId',
+      ['kb-1'],
+    )
+    expect(ai.rerankCandidates).toHaveBeenCalled()
+    expect(m.knowledgeDocument.findMany).toHaveBeenCalledWith(
+      expect.objectContaining({
+        where: expect.objectContaining({ id: { in: ['doc-1'] } }),
+      }),
+    )
+    expect(result).toEqual(
+      expect.arrayContaining([
+        expect.objectContaining({ id: 'doc-1', title: '文档1' }),
+      ]),
+    )
+  })
+
+  it('无 AI key 时降级为文本匹配（title contains）', async () => {
+    ki.resolveAiConfig.mockResolvedValue({})
+
+    m.knowledgeDocument.findMany.mockResolvedValue([
+      { id: 'doc-2', title: '文本匹配文档', kbId: 'kb-1', type: 'doc', icon: null },
+    ])
+
+    const result = await searchDocumentsFn({ data: { query: 'test', currentKbId: 'kb-1' } })
+
+    expect(ai.generateEmbedding).not.toHaveBeenCalled()
+    expect(vector.searchByVector).not.toHaveBeenCalled()
+    expect(m.knowledgeDocument.findMany).toHaveBeenCalledWith(
+      expect.objectContaining({
+        where: expect.objectContaining({
+          kbId: { in: ['kb-1'] },
+          title: { contains: 'test', mode: 'insensitive' },
+        }),
+      }),
+    )
+    expect(result).toEqual(
+      expect.arrayContaining([
+        expect.objectContaining({ id: 'doc-2', title: '文本匹配文档' }),
+      ]),
+    )
+  })
+
+  it('embedding 失败时降级为文本匹配', async () => {
+    ki.resolveAiConfig.mockResolvedValue({ apiKey: 'sk-xxx', baseUrl: 'https://api.openai.com/v1' })
+    ai.generateEmbedding.mockRejectedValue(new Error('embedding failed'))
+
+    m.knowledgeDocument.findMany.mockResolvedValue([
+      { id: 'doc-3', title: '降级文档', kbId: 'kb-1', type: 'doc', icon: null },
+    ])
+
+    const result = await searchDocumentsFn({ data: { query: 'test', currentKbId: 'kb-1' } })
+
+    expect(ai.generateEmbedding).toHaveBeenCalled()
+    expect(vector.searchByVector).not.toHaveBeenCalled()
+    expect(m.knowledgeDocument.findMany).toHaveBeenCalledWith(
+      expect.objectContaining({
+        where: expect.objectContaining({
+          title: { contains: 'test', mode: 'insensitive' },
+        }),
+      }),
+    )
+    expect(result).toEqual(
+      expect.arrayContaining([
+        expect.objectContaining({ id: 'doc-3', title: '降级文档' }),
+      ]),
+    )
+  })
+
+  it('rerank 返回 null 时用原始向量结果（降级不阻塞）', async () => {
+    ki.resolveAiConfig.mockResolvedValue({ apiKey: 'sk-xxx', baseUrl: 'https://api.openai.com/v1' })
+    ai.generateEmbedding.mockResolvedValue([0.1, 0.2, 0.3])
+    vector.searchByVector.mockResolvedValue([
+      { id: 'doc-1-0', content: 'chunk text', similarity: 0.9 },
+    ])
+    ai.rerankCandidates.mockResolvedValue(null)
+    m.knowledgeDocument.findMany.mockResolvedValue([
+      { id: 'doc-1', title: '文档1', kbId: 'kb-1', type: 'doc', icon: null },
+    ])
+
+    const result = await searchDocumentsFn({ data: { query: 'test', currentKbId: 'kb-1' } })
+
+    expect(ai.rerankCandidates).toHaveBeenCalled()
+    expect(m.knowledgeDocument.findMany).toHaveBeenCalledWith(
+      expect.objectContaining({
+        where: expect.objectContaining({ id: { in: ['doc-1'] } }),
+      }),
+    )
+    expect(result).toEqual(
+      expect.arrayContaining([
+        expect.objectContaining({ id: 'doc-1' }),
+      ]),
+    )
+  })
+})
+

package/src/server/serverFns/activity.ts

@@ -17,7 +17,6 @@
 
 import { createServerFn } from '@tanstack/react-start'
 import { getRequest } from '@tanstack/react-start/server'
-import { prisma } from '@allbluecn/database'
 import { requireAuth } from '@/server/auth-helpers'
 import { startOfDay, startOfMonth, subMonths, eachDayOfInterval } from 'date-fns'
 import { formatInTimeZone, toZonedTime, fromZonedTime } from 'date-fns-tz'
@@ -53,6 +52,7 @@ function startOfMonthInTimeZone(date: Date, timezone: string): Date {
 export const getActivityDataFn = createServerFn({ method: 'GET' })
   .validator((data: { timezone?: string }) => data)
   .handler(async ({ data }) => {
+    const { prisma } = await import('@allbluecn/database')
     const request = getRequest()
     const session = await requireAuth(request)
     const userId = session.user!.id!

package/src/server/serverFns/agents/agent-crud.ts

@@ -18,7 +18,6 @@
 import { createServerFn } from "@tanstack/react-start"
 import { getRequest } from "@tanstack/react-start/server"
 import { z } from "zod"
-import { prisma } from "@allbluecn/database"
 import { AGENT_TOOL_KEYS } from "@/server/ai-gateway/tools"
 import { getCurrentSession } from "@/lib/session"
 import { requireFeature } from "@/server/entitlements"
@@ -49,6 +48,7 @@ const agentInputSchema = z.object({
 type AgentInput = z.infer<typeof agentInputSchema>
 
 export const listAgentsFn = createServerFn({ method: "GET" }).handler(async () => {
+    const { prisma } = await import('@allbluecn/database')
   const request = getRequest()
   const userId = await requireSessionUser(request)
 
@@ -69,6 +69,7 @@ export const listAgentsFn = createServerFn({ method: "GET" }).handler(async () =
     greeting: a.greeting,
     suggestedPrompts: a.suggestedPrompts as string[] | null,
     knowledgeBaseIds: a.knowledgeBaseIds as string[],
+    isDefault: a.isDefault,
     createdAt: a.createdAt,
     updatedAt: a.updatedAt,
   }))
@@ -77,6 +78,7 @@ export const listAgentsFn = createServerFn({ method: "GET" }).handler(async () =
 export const createAgentFn = createServerFn({ method: "POST" })
   .validator(agentInputSchema)
   .handler(async ({ data }) => {
+    const { prisma } = await import('@allbluecn/database')
     const request = getRequest()
     const userId = await requireSessionUser(request)
 
@@ -111,6 +113,7 @@ export const createAgentFn = createServerFn({ method: "POST" })
       greeting: agent.greeting,
       suggestedPrompts: agent.suggestedPrompts as string[] | null,
       knowledgeBaseIds: agent.knowledgeBaseIds as string[],
+      isDefault: agent.isDefault,
       createdAt: agent.createdAt,
       updatedAt: agent.updatedAt,
     }
@@ -121,6 +124,7 @@ const getAgentSchema = z.object({ id: z.string() })
 export const getAgentFn = createServerFn({ method: "GET" })
   .validator(getAgentSchema)
   .handler(async ({ data }) => {
+    const { prisma } = await import('@allbluecn/database')
     const request = getRequest()
     const userId = await requireSessionUser(request)
 
@@ -141,6 +145,7 @@ export const getAgentFn = createServerFn({ method: "GET" })
       greeting: agent.greeting,
       suggestedPrompts: agent.suggestedPrompts as string[] | null,
       knowledgeBaseIds: agent.knowledgeBaseIds as string[],
+      isDefault: agent.isDefault,
       createdAt: agent.createdAt,
       updatedAt: agent.updatedAt,
     }
@@ -188,6 +193,7 @@ function buildAgentUpdatePayload(input: AgentUpdateInput): Record<string, unknow
 export const updateAgentFn = createServerFn({ method: "POST" })
   .validator(agentUpdateSchema)
   .handler(async ({ data }) => {
+    const { prisma } = await import('@allbluecn/database')
     const request = getRequest()
     const userId = await requireSessionUser(request)
 
@@ -212,6 +218,7 @@ const deleteAgentSchema = z.object({ id: z.string() })
 export const deleteAgentFn = createServerFn({ method: "POST" })
   .validator(deleteAgentSchema)
   .handler(async ({ data }) => {
+    const { prisma } = await import('@allbluecn/database')
     const request = getRequest()
     const userId = await requireSessionUser(request)