52 lines
1.5 KiB
TypeScript
52 lines
1.5 KiB
TypeScript
import { schema } from '#triplit/schema';
|
|
import type { Entity } from '@triplit/client';
|
|
|
|
export type ModelWithProvider = Entity<typeof schema, 'models'> & {
|
|
provider: Entity<typeof schema, 'providers'>;
|
|
};
|
|
|
|
export type ProviderWithModels = Entity<typeof schema, 'providers'> & {
|
|
models: Entity<typeof schema, 'models'>[];
|
|
};
|
|
|
|
export const useModels = async () => {
|
|
const triplit = useTriplitClient();
|
|
|
|
const providersQuery = triplit
|
|
.query('providers')
|
|
.Include('models')
|
|
|
|
const { results: providers, unsubscribe } = await useQuery('providers', triplit, providersQuery);
|
|
|
|
console.log("GET PROVIDERS", providers.value);
|
|
|
|
// const enabledProvidersWithModels = computed<ProviderWithModels[]>(() => {
|
|
// if (!providers.value) return [];
|
|
|
|
// return (providers.value as unknown as ProviderWithModels[]).filter(
|
|
// (provider: ProviderWithModels) => provider.models && provider.models.length > 0
|
|
// );
|
|
// });
|
|
|
|
const allEnabledModels = computed<ModelWithProvider[]>(() => {
|
|
return providers.value?.flatMap((provider) =>
|
|
provider.models.map((model) => ({
|
|
...model,
|
|
provider,
|
|
}))
|
|
);
|
|
});
|
|
|
|
const getFirstAvailableModel = (): ModelWithProvider | null => {
|
|
if (allEnabledModels.value.length === 0) return null;
|
|
return allEnabledModels.value[0]!;
|
|
};
|
|
|
|
return {
|
|
providers,
|
|
allModels: allEnabledModels,
|
|
getFirstAvailableModel,
|
|
unsubscribe,
|
|
};
|
|
};
|