aichat: add agents option

modules/programs/aichat.nix

@@ -11,7 +11,7 @@ let
 
   settingsFormat = pkgs.formats.yaml { };
 
-  inherit (pkgs.stdenv.hostPlatform) isLinux isDarwin;
+  inherit (pkgs.stdenv.hostPlatform) isDarwin;
 in
 {
   meta.maintainers = [
@@ -56,19 +56,53 @@ in
         for supported values.
       '';
     };
-  };
 
-  config = mkIf cfg.enable {
-    home.packages = mkIf (cfg.package != null) [ cfg.package ];
-
-    home.file."Library/Application Support/aichat/config.yaml" =
-      mkIf (cfg.settings != { } && (isDarwin && !config.xdg.enable))
-        {
-          source = settingsFormat.generate "aichat-config" cfg.settings;
+    agents = lib.mkOption {
+      type = lib.types.attrsOf settingsFormat.type;
+      default = { };
+      example = {
+        openai = {
+          model = "openai:gpt-4o";
+          temperature = 0.5;
+          top_p = 0.7;
+          use_tools = "fs,web_search";
+          agent_prelude = "default";
         };
 
-    xdg.configFile."aichat/config.yaml" = mkIf (cfg.settings != { } && (isLinux || config.xdg.enable)) {
-      source = settingsFormat.generate "aichat-config" cfg.settings;
+        llama = {
+          model = "llama3.2:latest";
+          temperature = 0.5;
+          use_tools = "web_search";
+        };
+      };
+      description = ''
+        Agent-specific configurations. See
+        <https://github.com/sigoden/aichat/wiki/Configuration-Guide#agent-specific>
+        for supported values.
+      '';
     };
   };
+
+  config =
+    let
+      aichatConfigPath =
+        if (isDarwin && !config.xdg.enable) then
+          "Library/Application Support/aichat"
+        else
+          "${lib.removePrefix config.home.homeDirectory config.xdg.configHome}/aichat";
+    in
+    mkIf cfg.enable {
+      home.packages = mkIf (cfg.package != null) [ cfg.package ];
+      home.file = {
+        "${aichatConfigPath}/config.yaml" = mkIf (cfg.settings != { }) {
+          source = settingsFormat.generate "aichat-config" cfg.settings;
+        };
+      }
+      // (lib.mapAttrs' (
+        k: v:
+        lib.nameValuePair "${aichatConfigPath}/agents/${k}/config.yaml" {
+          source = settingsFormat.generate "aichat-agent-${k}" v;
+        }
+      ) cfg.agents);
+    };
 }

tests/modules/programs/aichat/llama.yaml

@@ -0,0 +1,3 @@
+model: llama3.2:latest
+temperature: 0.5
+use_tools: web_search

tests/modules/programs/aichat/openai.yaml

@@ -0,0 +1,5 @@
+agent_prelude: default
+model: openai:gpt-4o
+temperature: 0.5
+top_p: 0.7
+use_tools: fs,web_search

tests/modules/programs/aichat/settings.nix

@@ -1,4 +1,3 @@
-{ ... }:
 {
   programs.aichat = {
     enable = true;
@@ -18,10 +17,34 @@
         }
       ];
     };
+
+    agents = {
+      openai = {
+        model = "openai:gpt-4o";
+        temperature = 0.5;
+        top_p = 0.7;
+        use_tools = "fs,web_search";
+        agent_prelude = "default";
+      };
+
+      llama = {
+        model = "llama3.2:latest";
+        temperature = 0.5;
+        use_tools = "web_search";
+      };
+    };
   };
+
   nmt.script = ''
     assertFileExists home-files/.config/aichat/config.yaml
     assertFileContent home-files/.config/aichat/config.yaml \
       ${./settings.yml}
+
+    assertFileExists home-files/.config/aichat/agents/openai/config.yaml
+    assertFileExists home-files/.config/aichat/agents/llama/config.yaml
+    assertFileContent home-files/.config/aichat/agents/openai/config.yaml \
+      ${./openai.yaml}
+    assertFileContent home-files/.config/aichat/agents/llama/config.yaml \
+      ${./llama.yaml}
   '';
 }