page-assist/src/libs/openai.ts

type Model = {
  id: string
  name?: string
  display_name?: string
  type: string
}

export const getAllOpenAIModels = async (baseUrl: string, apiKey?: string) => {
  try {
    const url = `${baseUrl}/models`
    const headers = apiKey
      ? {
        Authorization: `Bearer ${apiKey}`
      }
      : {}

    const res = await fetch(url, {
      headers
    })

    if (!res.ok) {
      return []
    }

    if (baseUrl === "https://api.together.xyz/v1") {
      const data = (await res.json()) as Model[]
      return data.map(model => ({
        id: model.id,
        name: model.display_name,
      }))
    }

    const data = (await res.json()) as { data: Model[] }

    return data.data
  } catch (e) {
    console.log(e)
    return []
  }
}