27 lines
731 B
TypeScript
27 lines
731 B
TypeScript
import { useProviders } from './useProvider'
|
|
import { getStoreProviders } from './useStore'
|
|
|
|
export function useModel(id?: string, providerId?: string) {
|
|
const { providers } = useProviders()
|
|
const allModels = providers.map((p) => p.models).flat()
|
|
return allModels.find((m) => {
|
|
if (providerId) {
|
|
return m.id === id && m.provider === providerId
|
|
} else {
|
|
return m.id === id
|
|
}
|
|
})
|
|
}
|
|
|
|
export function getModel(id?: string, providerId?: string) {
|
|
const providers = getStoreProviders()
|
|
const allModels = providers.map((p) => p.models).flat()
|
|
return allModels.find((m) => {
|
|
if (providerId) {
|
|
return m.id === id && m.provider === providerId
|
|
} else {
|
|
return m.id === id
|
|
}
|
|
})
|
|
}
|