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

test: copilot unit & e2e test (#6649)

Browse Source 
fix CLOUD-31
This commit is contained in:
darkskygit
2024-04-26 09:43:35 +00:00
parent f015a11181
commit 850bbee629
12 changed files with 1145 additions and 134 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
packages/backend/server/src/plugins/copilot/providers/fal.ts
View File

@@ -41,6 +41,10 @@ export class FalProvider
return !!config.apiKey;
}
get type(): CopilotProviderType {
return FalProvider.type;
}
getCapabilities(): CopilotCapability[] {
return FalProvider.capabilities;
}

10
packages/backend/server/src/plugins/copilot/providers/openai.ts
View File

@@ -13,7 +13,7 @@ import {
PromptMessage,
} from '../types';
const DEFAULT_DIMENSIONS = 256;
export const DEFAULT_DIMENSIONS = 256;
const SIMPLE_IMAGE_URL_REGEX = /^(https?:\/\/|data:image\/)/;
@@ -59,6 +59,10 @@ export class OpenAIProvider
return !!config.apiKey;
}
get type(): CopilotProviderType {
return OpenAIProvider.type;
}
getCapabilities(): CopilotCapability[] {
return OpenAIProvider.capabilities;
}
@@ -67,7 +71,7 @@ export class OpenAIProvider
return this.availableModels.includes(model);
}
private chatToGPTMessage(
protected chatToGPTMessage(
messages: PromptMessage[]
): OpenAI.Chat.Completions.ChatCompletionMessageParam[] {
// filter redundant fields
@@ -92,7 +96,7 @@ export class OpenAIProvider
});
}
private checkParams({
protected checkParams({
messages,
embeddings,
model,