feat(development/pi-coding-agent): wrapped the derivation with env variables

1 files changed, 12 insertions, 6 deletions

diff --git a/modules/development/pi-coding-agent/default.nix b/modules/development/pi-coding-agent/default.nix
index ec77686..01a83ac 100644
--- a/modules/development/pi-coding-agent/default.nix
+++ b/modules/development/pi-coding-agent/default.nix
@@ -14,16 +14,13 @@ in
 
     package = lib.mkOption {
       type = lib.types.package;
-      default = pkgs.pi;
+      readOnly = true;
+      default = config.programs.pi-coding-agent.package;
       description = "The Pi Coding Agent package to use.";
     };
   };
 
   config = lib.mkIf cfg.enable {
-    home.sessionVariables = {
-      PI_CODING_AGENT_DIR = configDir;
-    };
-
     home.file = {
       "${configDir}/skills".source = ./skills;
       "${configDir}/prompts".source = ./prompts;
@@ -33,6 +30,15 @@ in
     programs.pi-coding-agent = {
       enable = true;
       inherit configDir;
+      package = pkgs.symlinkJoin {
+        name = "pi-coding-agent-wrapped";
+        paths = [ pkgs.pi-coding-agent ];
+        nativeBuildInputs = [ pkgs.makeWrapper ];
+        postBuild = ''
+          wrapProgram $out/bin/pi \
+            --set PI_CODING_AGENT_DIR ${configDir}
+        '';
+      };
       keybindings = {
         "tui.editor.cursorUp" = [
           "up"
@@ -73,7 +79,7 @@ in
 
         retry = {
           enabled = true;
-          maxRetries = 3;
+          maxRetries = 30;
           baseDelayMs = 2000;
           maxAgentDelayMs = 60000;
         };