feat: qwen

This commit is contained in:
Dogtiti
2024-07-07 21:59:56 +08:00
parent 7218f13783
commit 9bdd37bb63
8 changed files with 493 additions and 1 deletions

View File

@@ -14,6 +14,9 @@ export const ANTHROPIC_BASE_URL = "https://api.anthropic.com";
export const GEMINI_BASE_URL = "https://generativelanguage.googleapis.com/";
export const ALIBABA_BASE_URL =
"https://dashscope.aliyuncs.com/api/v1/services/aigc/text-generation/generation";
export enum Path {
Home = "/",
Chat = "/chat",
@@ -28,6 +31,7 @@ export enum ApiPath {
Azure = "/api/azure",
OpenAI = "/api/openai",
Anthropic = "/api/anthropic",
Alibaba = "/api/alibaba",
}
export enum SlotID {
@@ -71,12 +75,14 @@ export enum ServiceProvider {
Azure = "Azure",
Google = "Google",
Anthropic = "Anthropic",
Alibaba = "Alibaba",
}
export enum ModelProvider {
GPT = "GPT",
GeminiPro = "GeminiPro",
Claude = "Claude",
Qwen = "Qwen",
}
export const Anthropic = {
@@ -104,6 +110,10 @@ export const Google = {
ChatPath: (modelName: string) => `v1beta/models/${modelName}:generateContent`,
};
export const Alibaba = {
ChatPath: "chat/completions",
};
export const DEFAULT_INPUT_TEMPLATE = `{{input}}`; // input / time / model / lang
// export const DEFAULT_SYSTEM_TEMPLATE = `
// You are ChatGPT, a large language model trained by {{ServiceProvider}}.
@@ -173,6 +183,16 @@ const anthropicModels = [
"claude-3-5-sonnet-20240620",
];
const alibabaModes = [
"qwen-turbo",
"qwen-plus",
"qwen-max",
"qwen-max-0428",
"qwen-max-0403",
"qwen-max-0107",
"qwen-max-longcontext",
];
export const DEFAULT_MODELS = [
...openaiModels.map((name) => ({
name,
@@ -210,6 +230,15 @@ export const DEFAULT_MODELS = [
providerType: "anthropic",
},
})),
...alibabaModes.map((name) => ({
name,
available: true,
provider: {
id: "alibaba",
providerName: "Alibaba",
providerType: "alibaba",
},
})),
] as const;
export const CHAT_PAGE_SIZE = 15;