39 lines
1.1 KiB
Nix
39 lines
1.1 KiB
Nix
{
|
|
lib,
|
|
writers,
|
|
writeText,
|
|
linkFarmFromDrvs,
|
|
}:
|
|
{
|
|
genModels =
|
|
configs:
|
|
let
|
|
name = lib.strings.sanitizeDerivationName (
|
|
builtins.concatStringsSep "_" ([ "local-ai-models" ] ++ (builtins.attrNames configs))
|
|
);
|
|
|
|
genModelFiles =
|
|
name: config:
|
|
let
|
|
templateName = type: name + "_" + type;
|
|
|
|
config' = lib.recursiveUpdate config (
|
|
{
|
|
inherit name;
|
|
}
|
|
// lib.optionalAttrs (lib.isDerivation config.parameters.model) {
|
|
parameters.model = config.parameters.model.name;
|
|
}
|
|
// lib.optionalAttrs (config ? template) {
|
|
template = builtins.mapAttrs (n: _: templateName n) config.template;
|
|
}
|
|
);
|
|
in
|
|
[ (writers.writeYAML "${name}.yaml" config') ]
|
|
++ lib.optional (lib.isDerivation config.parameters.model) config.parameters.model
|
|
++ lib.optionals (config ? template) (
|
|
lib.mapAttrsToList (n: writeText "${templateName n}.tmpl") config.template
|
|
);
|
|
in
|
|
linkFarmFromDrvs name (lib.flatten (lib.mapAttrsToList genModelFiles configs));
|
|
}
|