Add cognix.environment

modules/cog-interface.nix

@@ -71,6 +71,10 @@ in
       '';
       description = "Commands to run after gathering all the files for the container /src dir";
     };
+    environment = mkOption {
+      type = types.attrsOf types.anything;
+      default = {};
+    };
   };
   options.python-env = with lib; mkOption {
     type = types.submoduleWith {

modules/cog.nix

@@ -90,6 +90,8 @@ in {
       "${config.paths.projectRoot}"
                else "${config.paths.projectRoot}/${config.paths.package}";
 
+    cognix.environment.PYTHONUNBUFFERED = true;
+
     dockerTools.streamLayeredImage = {
       passthru.entirePackage = entirePackage;
       # glibc.out is needed for gpu
@@ -105,7 +107,7 @@ in {
         ] ++ resolvedSystemPackages;
       config = {
         Entrypoint = [ "${pkgs.tini}/bin/tini" "--" ];
-        Env = [ "PYTHONUNBUFFERED=1" ];
+        Env = lib.mapAttrsToList (name: val: "${name}=${toString val}") config.cognix.environment;
         EXPOSE = 5000;
         CMD = [ "python" "-m" "cog.server.http" ];
         WorkingDir = "/src";