48 lines
1.5 KiB
Nix
48 lines
1.5 KiB
Nix
{ lib, config, pkgs, flakeRoot, ... }:
|
|
|
|
let
|
|
# Path to the config file (relative to your flake or Home Manager root)
|
|
ollamaConfPath = flakeRoot + "/assets/conf/apps/ai/ollama/ollama.conf";
|
|
# Read and parse the config file (assuming it's in shell variable format)
|
|
ollamaConf = builtins.readFile ollamaConfPath;
|
|
|
|
# Extract values using regex (adjust if your format differs)
|
|
extractValue = key:
|
|
let
|
|
match = builtins.match ("(?m)^${key}=\"([^\"]+)\"$") ollamaConf;
|
|
in
|
|
if match != null && match != [] then
|
|
# Return the first capture group of the first match
|
|
builtins.elemAt (builtins.elemAt match 0) 1
|
|
else
|
|
null;
|
|
|
|
ollamaHost = extractValue "OLLAMA_HOST";
|
|
ollamaDefaultModel = extractValue "OLLAMA_DEFAULT_MODEL";
|
|
in
|
|
{
|
|
# Install Ollama
|
|
home.packages = with pkgs; [
|
|
ollama
|
|
];
|
|
|
|
# Configure Ollama environment variables
|
|
home.sessionVariables = {
|
|
OLLAMA_HOST = if ollamaHost != null then ollamaHost else "http://127.0.0.1:11434";
|
|
OLAMA_DEFAULT_MODEL = if ollamaDefaultModel != null then ollamaDefaultModel else "codellama:70b";
|
|
};
|
|
|
|
# Optional: Start Ollama service (if using NixOS)
|
|
systemd.user.services.ollama = {
|
|
description = "Ollama service";
|
|
wantedBy = [ "multi-user.target" ];
|
|
serviceConfig = {
|
|
ExecStart = "${pkgs.ollama}/bin/ollama serve";
|
|
Restart = "on-failure";
|
|
Environment = [
|
|
"OLLAMA_HOST=${if ollamaHost != null then ollamaHost else "http://127.0.0.1:11434"}"
|
|
];
|
|
};
|
|
};
|
|
}
|