added more notes about continue configuration
This commit is contained in:
parent
77ab4781ac
commit
3f88b989d1
4 changed files with 5 additions and 1 deletions
44
modules/nixos-modules/ai.nix
Normal file
44
modules/nixos-modules/ai.nix
Normal file
|
@ -0,0 +1,44 @@
|
|||
{lib, ...}: {
|
||||
options.host = {
|
||||
ai = {
|
||||
enable = lib.mkEnableOption "should we use AI on this machine";
|
||||
models = lib.mkOption {
|
||||
type = lib.types.attrsOf (lib.types.submodule ({name, ...}: {
|
||||
option = {
|
||||
name = lib.mkOption {
|
||||
type = lib.types.str;
|
||||
default = name;
|
||||
};
|
||||
model = {
|
||||
type = lib.types.str;
|
||||
};
|
||||
provider = {
|
||||
type = lib.types.str;
|
||||
default = "ollama";
|
||||
};
|
||||
apiBase = {
|
||||
type = lib.types.str;
|
||||
default = null;
|
||||
};
|
||||
roles = {
|
||||
type = lib.types.listOf lib.types.enumOf [
|
||||
"chat"
|
||||
"autocomplete"
|
||||
"embed"
|
||||
"rerank"
|
||||
"edit"
|
||||
"apply"
|
||||
"summarize"
|
||||
];
|
||||
};
|
||||
};
|
||||
}));
|
||||
};
|
||||
};
|
||||
};
|
||||
|
||||
config = {
|
||||
# TODO: configure ollama to download any modules listed in options.host.ai.models.{name}.model if options.host.ai.models.{name}.apiBase is null
|
||||
# TODO: if we have any models that have a non null options.host.ai.models.{name}.apiBase then set services.ollama.enable to a lib.mkAfter true
|
||||
};
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue