Dokploy/dokploy
1
0
Fork 0
mirror of https://github.com/Dokploy/dokploy synced 2025-06-26 18:27:59 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: fallback to openai compatible provider if url host doesn't match

Browse Source
This commit is contained in:
Frost Ming
2025-03-05 16:12:46 +08:00
committed by GitHub
parent 6ff06576d0
commit 21c8b98f9c
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
packages/server/src/utils/ai/select-ai-provider.ts
View File

@@ -17,7 +17,7 @@ function getProviderName(apiUrl: string) {
if (apiUrl.includes("localhost:11434") || apiUrl.includes("ollama")) if (apiUrl.includes("localhost:11434") || apiUrl.includes("ollama"))
return "ollama"; return "ollama";
if (apiUrl.includes("api.deepinfra.com")) return "deepinfra"; if (apiUrl.includes("api.deepinfra.com")) return "deepinfra";
throw new Error(`Unsupported AI provider for URL: ${apiUrl}`); return "custom";
} }
export function selectAIProvider(config: { apiUrl: string; apiKey: string }) { export function selectAIProvider(config: { apiUrl: string; apiKey: string }) {
@@ -68,6 +68,12 @@ export function selectAIProvider(config: { apiUrl: string; apiKey: string }) {
apiKey: config.apiKey, apiKey: config.apiKey,
}); });
default: default:
throw new Error(`Unsupported AI provider: ${providerName}`); return createOpenAICompatible({
name: "custom",
baseURL: config.apiUrl,
headers: {
Authorization: `Bearer ${config.apiKey}`,
},
)};
} }
} }