7x31/ChatGPT-Next-Web
1
0
Fork 0
mirror of https://github.com/Yidadaa/ChatGPT-Next-Web.git synced 2025-08-19 22:18:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #4986 from ConnectAI-E/hotfix/cf-ai-gateway

Browse Source 
hotfix: using custom models, create custom provider
This commit is contained in:
LiuElric
2024-07-12 20:31:22 +08:00
committed by GitHub
parent e8088d6e38 9203870df5
commit 4ea8c0802a
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
app/utils/model.ts
View File

@@ -1,9 +1,9 @@
import { DEFAULT_MODELS } from "../constant"; import { DEFAULT_MODELS } from "../constant";
import { LLMModel } from "../client/api"; import { LLMModel } from "../client/api";
const customProvider = (modelName: string) => ({ const customProvider = (providerName: string) => ({
id: modelName, id: providerName.toLowerCase(),
providerName: "Custom", providerName: providerName,
providerType: "custom", providerType: "custom",
}); });
@@ -71,10 +71,13 @@ export function collectModelTable(
} }
// 2. if model not exists, create new model with available value // 2. if model not exists, create new model with available value
if (count === 0) { if (count === 0) {
const provider = customProvider(name); const [customModelName, customProviderName] = name.split("@");
modelTable[`${name}@${provider?.id}`] = { const provider = customProvider(
name, customProviderName || customModelName,
displayName: displayName || name, );
modelTable[`${customModelName}@${provider?.id}`] = {
name: customModelName,
displayName: displayName || customModelName,
available, available,
provider, // Use optional chaining provider, // Use optional chaining
}; };