opencode: option to override model used by specialized (sub)agents

Author: YPares

lib/mkRiglib.nix

@@ -19,7 +19,7 @@ let
       );
   };
 
-  # Override a derivation so its always built locally, and not queried from remote substituters
+  # Override a derivation so it is always built locally, and not queried from remote substituters
   alwaysLocal =
     x:
     if pkgs.lib.isDerivation x then

riglets/models.nix

@@ -1,19 +1,8 @@
 _:
-{ lib, ... }:
+{ lib, riglib, ... }:
 {
   options.models = with lib.types; {
-    default = {
-      providerId = lib.mkOption {
-        type = nullOr str;
-        description = "Model provider ID. Can be ignored depending on the entrypoint";
-        default = null;
-      };
-      modelId = lib.mkOption {
-        type = nullOr str;
-        description = "Model ID";
-        default = null;
-      };
-    };
+    default = riglib.providerAndModel;
 
     providers = {
       disabled = lib.mkOption {
@@ -31,7 +20,7 @@ _:
   };
 
   config.riglets.models.meta = {
-    description = "Pre-select a specific model for a rig (to be imported by compatible entrypoints)";
+    description = "Pre-select default model for a rig (to be imported by compatible entrypoints)";
     status = "stable";
     version = "0.1.0";
     intent = "base";

riglets/opencode.nix

@@ -32,6 +32,8 @@ in
     let
       manifestPath = rig.manifest.override { shownDocRoot = "$RIG_DOCS"; };
 
+      toModelId = x: "${x.providerId}/${x.modelId}";
+
       # OpenCode config with permissions and MCP servers
       opencodeConfigJson = riglib.toJSON (
         {
@@ -112,24 +114,23 @@ in
                 If models.default.modelId is set, then models.default.providerId must be too
               ''
             else
-              "${config.models.default.providerId}/${config.models.default.modelId}";
+              toModelId config.models.default;
         }
         // lib.optionalAttrs (config.models.providers.disabled != null) {
           disabled_providers = config.models.providers.disabled;
         }
         // lib.optionalAttrs (config.models.providers.enabled != null) {
           enabled_providers = config.models.providers.enabled;
         }
+        // lib.optionalAttrs (config.models.specialized != { }) {
+          agent = lib.mapAttrs (_agentName: toModelId) config.models.specialized;
+        }
       );
     in
     # Return a folder derivation with bin/ subfolder
     pkgs.writeShellScriptBin "opencode" ''
       set -euo pipefail
 
-      warn() {
-        printf "\033[0;33m%s\n\033[0m" "$1" >&2
-      }
-
       export PATH="${rig.toolRoot}/bin:$PATH"
       export RIG_DOCS="${rig.docRoot}"
       # Exported for convenience