diff --git a/packages/backend/server/src/__tests__/copilot-provider.e2e.ts b/packages/backend/server/src/__tests__/copilot-provider.e2e.ts index d00f8c2805..5dc3ba4e2f 100644 --- a/packages/backend/server/src/__tests__/copilot-provider.e2e.ts +++ b/packages/backend/server/src/__tests__/copilot-provider.e2e.ts @@ -1,6 +1,7 @@ import { randomUUID } from 'node:crypto'; import { createRandomAIUser } from '@affine-test/kit/utils/cloud'; +import { PrismaClient } from '@prisma/client'; import type { ExecutionContext, TestFn } from 'ava'; import ava from 'ava'; @@ -47,13 +48,8 @@ const runIfCopilotConfigured = test.macro( ); export const runPrisma = async ( - cb: ( - prisma: InstanceType< - typeof import('../../node_modules/@prisma/client').PrismaClient - > - ) => Promise + cb: (prisma: PrismaClient) => Promise ): Promise => { - const { PrismaClient } = await import('../../node_modules/@prisma/client'); const client = new PrismaClient(); await client.$connect(); try {