Imported as a linear vendor commit so forge rebases do not replay the upstream root tree. git-subtree-dir: third_party/pi git-subtree-split: d981de1229ef899957bbe968bc8dcda02a21f477
28 lines
920 B
TypeScript
28 lines
920 B
TypeScript
import type { Api, Model, ProviderId } from "./types.ts";
|
|
|
|
export type ModelGroups = Record<string, Record<string, object>>;
|
|
|
|
type ModelId<TGroups extends ModelGroups> = {
|
|
[TApi in keyof TGroups]: keyof TGroups[TApi];
|
|
}[keyof TGroups] &
|
|
string;
|
|
|
|
type ModelApi<TGroups extends ModelGroups, TModelId extends ModelId<TGroups>> = {
|
|
[TApi in keyof TGroups]: TModelId extends keyof TGroups[TApi] ? TApi : never;
|
|
}[keyof TGroups] &
|
|
Api;
|
|
|
|
export type ModelCatalog<TGroups extends ModelGroups, TProvider extends ProviderId> = {
|
|
[TModelId in ModelId<TGroups>]: Model<ModelApi<TGroups, TModelId>> & {
|
|
id: TModelId;
|
|
provider: TProvider;
|
|
};
|
|
};
|
|
|
|
export function flattenModelCatalog<const TProvider extends ProviderId, const TGroups extends ModelGroups>(
|
|
_provider: TProvider,
|
|
groups: TGroups,
|
|
): ModelCatalog<TGroups, TProvider> {
|
|
return Object.assign({}, ...Object.values(groups)) as ModelCatalog<TGroups, TProvider>;
|
|
}
|