fix: test case

toeverything · Apr 25, 2024 · 9145980 · 9145980
1 parent c983abc
commit 9145980
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 2 deletions.
diff --git a/packages/backend/server/src/config/affine.self.ts b/packages/backend/server/src/config/affine.self.ts
@@ -45,6 +45,7 @@ if (env.R2_OBJECT_STORAGE_ACCOUNT_ID) {
 
 AFFiNE.plugins.use('copilot', {
   openai: {},
+  fal: {},
 });
 AFFiNE.plugins.use('redis');
 AFFiNE.plugins.use('payment', {

diff --git a/packages/backend/server/src/plugins/copilot/types.ts b/packages/backend/server/src/plugins/copilot/types.ts
@@ -15,6 +15,7 @@ export interface CopilotConfig {
   openai: OpenAIClientOptions;
   fal: FalConfig;
   unsplashKey: string;
+  test: never;
 }
 
 export enum AvailableModels {

diff --git a/packages/backend/server/tests/copilot.e2e.ts b/packages/backend/server/tests/copilot.e2e.ts
@@ -30,7 +30,7 @@ import {
   createCopilotMessage,
   createCopilotSession,
   getHistories,
-  TestProvider,
+  MockCopilotTestProvider,
   textToEventStream,
 } from './utils/copilot';
 
@@ -52,6 +52,9 @@ test.beforeEach(async t => {
             openai: {
               apiKey: '1',
             },
+            fal: {
+              apiKey: '1',
+            },
           },
         },
       }),
@@ -75,7 +78,7 @@ test.beforeEach(async t => {
   const user = await signUp(app, 'test', 'darksky@affine.pro', '123456');
   token = user.token.token;
 
-  registerCopilotProvider(TestProvider);
+  registerCopilotProvider(MockCopilotTestProvider);
 
   await prompt.set(promptName, 'test', [
     { role: 'system', content: 'hello {{word}}' },

diff --git a/packages/backend/server/tests/copilot.spec.ts b/packages/backend/server/tests/copilot.spec.ts
@@ -38,6 +38,9 @@ test.beforeEach(async t => {
             openai: {
               apiKey: '1',
             },
+            fal: {
+              apiKey: '1',
+            },
           },
         },
       }),