mirror of
https://github.com/open-webui/open-webui
synced 2025-01-11 05:18:13 +00:00
37 lines
845 B
TypeScript
37 lines
845 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((err) => {
|
|
console.log(err);
|
|
error = `OpenAI: ${err?.error?.message ?? 'Network Problem'}`;
|
|
return null;
|
|
});
|
|
|
|
if (error) {
|
|
throw error;
|
|
}
|
|
|
|
let models = Array.isArray(res) ? res : res?.data ?? null;
|
|
|
|
return models
|
|
.map((model) => ({ name: model.id, external: true }))
|
|
.filter((model) => (base_url.includes('openai') ? model.name.includes('gpt') : true))
|
|
.sort((a, b) => {
|
|
return a.name.localeCompare(b.name);
|
|
});
|
|
};
|