Jayden/AFFiNE-Mirror
1
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-02-12 12:28:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(server): copilot test condition (#11417)

Browse Source
This commit is contained in:
darkskygit
2025-04-02 14:59:22 +00:00
parent abfbf07712
commit 90c43356c1
1 changed files with 10 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
packages/backend/server/src/__tests__/copilot-provider.spec.ts
View File

@@ -2,7 +2,7 @@ import type { ExecutionContext, TestFn } from 'ava';
import ava from 'ava';
import { z } from 'zod';
import { ConfigModule } from '../base/config';
import { ServerFeature, ServerService } from '../core';
import { AuthService } from '../core/auth';
import { QuotaModule } from '../core/quota';
import { CopilotModule } from '../plugins/copilot';
@@ -36,13 +36,7 @@ type Tester = {
};
const test = ava as TestFn<Tester>;
const isCopilotConfigured =
!!process.env.COPILOT_OPENAI_API_KEY &&
!!process.env.COPILOT_FAL_API_KEY &&
!!process.env.COPILOT_PERPLEXITY_API_KEY &&
process.env.COPILOT_OPENAI_API_KEY !== '1' &&
process.env.COPILOT_FAL_API_KEY !== '1' &&
process.env.COPILOT_PERPLEXITY_API_KEY !== '1';
let isCopilotConfigured = false;
const runIfCopilotConfigured = test.macro(
async (
t,
@@ -59,30 +53,12 @@ const runIfCopilotConfigured = test.macro(
test.serial.before(async t => {
const module = await createTestingModule({
imports: [
ConfigModule.override({
copilot: {
providers: {
openai: {
apiKey: process.env.COPILOT_OPENAI_API_KEY || '',
},
fal: {
apiKey: process.env.COPILOT_FAL_API_KEY || '',
},
perplexity: {
apiKey: process.env.COPILOT_PERPLEXITY_API_KEY || '',
},
gemini: {
apiKey: process.env.COPILOT_GOOGLE_API_KEY || '',
},
},
},
}),
QuotaModule,
CopilotModule,
],
imports: [QuotaModule, CopilotModule],
});
const service = module.get(ServerService);
isCopilotConfigured = service.features.includes(ServerFeature.Copilot);
const auth = module.get(AuthService);
const prompt = module.get(PromptService);
const factory = module.get(CopilotProviderFactory);
@@ -357,10 +333,10 @@ const actions = [
.trim();
t.notThrows(() => {
z.object({
speaker: z.string(),
start: z.string(),
end: z.string(),
transcription: z.string(),
a: z.string(),
s: z.number(),
e: z.number(),
t: z.string(),
})
.array()
.parse(JSON.parse(cleaned));