mirror of
https://git.mirrors.martin98.com/https://github.com/open-webui/open-webui
synced 2025-06-20 16:21:18 +08:00
36 lines
809 B
TypeScript
36 lines
809 B
TypeScript
export const getOpenAIModels = async (
|
|
base_url: string = 'https://api.openai.com/v1',
|
|
api_key: string = ''
|
|
) => {
|
|
let error = null;
|
|
|
|
const res = await fetch(`${base_url}/models`, {
|
|
method: 'GET',
|
|
headers: {
|
|
'Content-Type': 'application/json',
|
|
Authorization: `Bearer ${api_key}`
|
|
}
|
|
})
|
|
.then(async (res) => {
|
|
if (!res.ok) throw await res.json();
|
|
return res.json();
|
|
})
|
|
.catch((error) => {
|
|
console.log(error);
|
|
error = `OpenAI: ${error?.error?.message ?? 'Network Problem'}`;
|
|
return null;
|
|
});
|
|
|
|
if (error) {
|
|
throw error;
|
|
}
|
|
|
|
let models = Array.isArray(res) ? res : res?.data ?? null;
|
|
|
|
console.log(models);
|
|
|
|
return models
|
|
.map((model) => ({ name: model.id, external: true }))
|
|
.filter((model) => (base_url.includes('openai') ? model.name.includes('gpt') : true));
|
|
};
|