@openclawcity/become 1.0.0 → 1.0.2

package/dist/cli.js

@@ -1226,18 +1226,24 @@ function patchOpenClaw(config) {
   const raw = readFileSync4(OPENCLAW_CONFIG, "utf-8");
   const clawConfig = JSON.parse(raw);
   mkdirSync4(join4(homedir2(), ".become", "state"), { recursive: true });
-  writeFileSync4(BACKUP_PATH, raw, "utf-8");
+  if (!clawConfig.models?.providers?.become) {
+    writeFileSync4(BACKUP_PATH, raw, "utf-8");
+  }
+  const originalModel = clawConfig.agents?.defaults?.model?.primary ?? "";
+  const originalModelPath = join4(homedir2(), ".become", "state", "original_model.txt");
+  writeFileSync4(originalModelPath, originalModel, "utf-8");
+  const modelId = originalModel.includes("/") ? originalModel.split("/").slice(1).join("/") : originalModel;
   if (!clawConfig.models) clawConfig.models = {};
   if (!clawConfig.models.providers) clawConfig.models.providers = {};
   clawConfig.models.providers.become = {
-    api: "anthropic-messages",
+    api: config.llm_provider === "openai" ? "openai-completions" : "anthropic-messages",
     baseUrl: `http://127.0.0.1:${config.proxy_port}`,
-    apiKey: config.llm_api_key
+    apiKey: config.llm_api_key,
+    models: [
+      { id: modelId, name: `${modelId} via become` }
+    ]
   };
-  if (clawConfig.agents?.defaults?.model?.primary) {
-    const original = clawConfig.agents.defaults.model.primary;
-    clawConfig.models.providers.become._originalModel = original;
-    const modelId = original.includes("/") ? original.split("/").slice(1).join("/") : original;
+  if (originalModel) {
     clawConfig.agents.defaults.model.primary = `become/${modelId}`;
   }
   writeFileSync4(OPENCLAW_CONFIG, JSON.stringify(clawConfig, null, 2), "utf-8");
@@ -1248,17 +1254,46 @@ function patchOpenClaw(config) {
   }
 }
 function restoreOpenClaw() {
-  if (!existsSync4(BACKUP_PATH)) {
-    throw new Error("No backup found. Was become ever turned on?");
+  if (!existsSync4(OPENCLAW_CONFIG)) {
+    throw new Error(`OpenClaw config not found at ${OPENCLAW_CONFIG}`);
+  }
+  if (existsSync4(BACKUP_PATH)) {
+    const backup = readFileSync4(BACKUP_PATH, "utf-8");
+    const backupConfig = JSON.parse(backup);
+    if (!backupConfig.models?.providers?.become) {
+      writeFileSync4(OPENCLAW_CONFIG, backup, "utf-8");
+    } else {
+      manualRestore();
+    }
+  } else {
+    manualRestore();
   }
-  const backup = readFileSync4(BACKUP_PATH, "utf-8");
-  writeFileSync4(OPENCLAW_CONFIG, backup, "utf-8");
   try {
     execSync("openclaw gateway restart", { stdio: "pipe", timeout: 15e3 });
   } catch {
     console.log("Warning: Could not restart OpenClaw gateway. Restart it manually: openclaw gateway restart");
   }
 }
+function manualRestore() {
+  const raw = readFileSync4(OPENCLAW_CONFIG, "utf-8");
+  const config = JSON.parse(raw);
+  const originalModelPath = join4(homedir2(), ".become", "state", "original_model.txt");
+  if (existsSync4(originalModelPath)) {
+    const originalModel = readFileSync4(originalModelPath, "utf-8").trim();
+    if (originalModel && config.agents?.defaults?.model) {
+      config.agents.defaults.model.primary = originalModel;
+    }
+  }
+  if (config.models?.providers?.become) {
+    delete config.models.providers.become;
+  }
+  for (const provider of Object.values(config.models?.providers ?? {})) {
+    if (provider && typeof provider === "object" && "_originalModel" in provider) {
+      delete provider._originalModel;
+    }
+  }
+  writeFileSync4(OPENCLAW_CONFIG, JSON.stringify(config, null, 2), "utf-8");
+}
 
 // src/cli/adapter/ironclaw.ts
 import { readFileSync as readFileSync5, writeFileSync as writeFileSync5, existsSync as existsSync5, mkdirSync as mkdirSync5, copyFileSync } from "fs";