62 lines
2.0 KiB
Nix
62 lines
2.0 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;
|
|
|
|
# Split the file into lines and filter out comments/empty lines
|
|
lines = lib.filter (line: line != "" && ! (lib.hasPrefix "#" line))
|
|
(lib.splitString "\n" ollamaConf);
|
|
|
|
# Extract the last value for a given key
|
|
extractValue = key:
|
|
let
|
|
# Find all lines matching the key
|
|
matches = lib.map (
|
|
line: builtins.match ("^${key}=\"([^\"]+)\"$") line
|
|
) lines;
|
|
# Filter out null matches and get the last one
|
|
nonNullMatches = lib.filter (match: match != null && match != []) matches;
|
|
# Extract the first capture group from the last match
|
|
lastMatch = if nonNullMatches != [] then
|
|
builtins.elemAt nonNullMatches (lib.length nonNullMatches - 1)
|
|
else
|
|
null;
|
|
in
|
|
if lastMatch != null then
|
|
# The first capture group is the second element (index 1) of the match
|
|
builtins.elemAt (builtins.elemAt lastMatch 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"}"
|
|
];
|
|
};
|
|
};
|
|
}
|