hfpi: follow the Space's default model and model list on every boot, keeping the user's own choices

sandbox/hfpi.mjs

@@ -39,6 +39,8 @@ const AGENT_SAVE_SECONDS = Number(process.env.HFPI_AGENT_SAVE_SECONDS ?? 20);
 const PROJECT_SAVE_SECONDS = Number(process.env.HFPI_PROJECT_SAVE_SECONDS ?? 120);
 const NAME_RE = /^[A-Za-z0-9][A-Za-z0-9._-]{0,63}$/;
 const MANAGED_MARK = "<!-- managed by hfpi: delete this line to keep your own edits -->";
+const DEFAULT_MODEL = "deepseek-ai/DeepSeek-V4.1-Flash";
+const LEGACY_DEFAULT_MODEL = "zai-org/GLM-5.3"; // the default before hfpi-models.json recorded it
 
 // Regenerable directories are not saved: they make archives 20x larger and slower.
 // Python virtualenvs are found by their pyvenv.cfg, Rust/pytest caches by CACHEDIR.TAG.
@@ -408,6 +410,32 @@ function restoreProjects() {
 
 // ---------------------------------------------------------------- pi configuration
 
+// The Space's default model and model list follow the Space on every boot, without
+// undoing what the user chose in Pi Web: `applied` is what hfpi set last time (none
+// in workspaces from before this record, when the default was LEGACY_DEFAULT_MODEL).
+// Returns the new record. Mutates `settings`.
+function mergeModelSettings(settings, applied, defaultModel, spaceModels) {
+  const before = applied || { defaultModel: settings.defaultProvider ? LEGACY_DEFAULT_MODEL : null, enabledModels: [] };
+  if (!settings.defaultProvider) {
+    settings.defaultThinkingLevel = settings.defaultThinkingLevel || "medium";
+  }
+  if (!settings.defaultProvider || (settings.defaultProvider === "huggingface" && settings.defaultModel === before.defaultModel)) {
+    settings.defaultProvider = "huggingface";
+    settings.defaultModel = defaultModel;
+  }
+  if (spaceModels.length) {
+    if (!settings.enabledModels) {
+      settings.enabledModels = [...spaceModels];
+    } else {
+      // Drop what the Space dropped, add what it added; the user's own changes stay.
+      const kept = settings.enabledModels.filter((m) => spaceModels.includes(m) || !before.enabledModels.includes(m));
+      const added = spaceModels.filter((m) => !kept.includes(m) && !before.enabledModels.includes(m));
+      settings.enabledModels = [...kept, ...added];
+    }
+  }
+  return { defaultModel, enabledModels: spaceModels };
+}
+
 function configurePi() {
   fs.mkdirSync(AGENT_DIR, { recursive: true });
 
@@ -427,17 +455,13 @@ function configurePi() {
   writeJson(modelsPath, models);
 
   const settingsPath = path.join(AGENT_DIR, "settings.json");
+  const managedPath = path.join(AGENT_DIR, "hfpi-models.json");
   const settings = readJson(settingsPath, {}) || {};
-  if (!settings.defaultProvider) {
-    settings.defaultProvider = "huggingface";
-    settings.defaultModel = process.env.HFPI_DEFAULT_MODEL || "zai-org/GLM-5.3";
-    settings.defaultThinkingLevel = settings.defaultThinkingLevel || "medium";
-  }
-  if (!settings.enabledModels && process.env.HFPI_ENABLED_MODELS) {
-    settings.enabledModels = process.env.HFPI_ENABLED_MODELS.split(",").map((s) => s.trim()).filter(Boolean);
-  }
+  const spaceModels = (process.env.HFPI_ENABLED_MODELS || "").split(",").map((s) => s.trim()).filter(Boolean);
+  const applied = mergeModelSettings(settings, readJson(managedPath, null), process.env.HFPI_DEFAULT_MODEL || DEFAULT_MODEL, spaceModels);
   settings.enableInstallTelemetry = false;
   writeJson(settingsPath, settings);
+  writeJson(managedPath, applied);
 
   // Global context about this environment, rewritten each boot unless the user
   // removed the managed marker to keep their own version.