@4yi/cli 0.1.2 → 0.1.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@4yi/cli",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "description": "4YI command-line launcher for OAuth login and OpenCode runtime",
   "type": "module",
   "bin": {

package/src/opencode.mjs

@@ -6,18 +6,43 @@ import { pathsForHome, ensureDir } from "./config.mjs";
 import { requestJson } from "./http.mjs";
 
 const DEFAULT_OPENCODE_PACKAGE = "opencode-ai";
+const DEFAULT_CONTEXT_LIMIT = 200000;
+const DEFAULT_OUTPUT_LIMIT = 8192;
+
+function isClaudeModel(model) {
+  const id = model?.id || "";
+  const name = model?.display_name || "";
+  return /claude/i.test(id) || /claude/i.test(name);
+}
+
+function modelContextLimit(model) {
+  return model.context_length || model.context_limit || model.input_limit || DEFAULT_CONTEXT_LIMIT;
+}
+
+function modelOutputLimit(model) {
+  return model.output_limit || model.max_output_tokens || model.max_tokens || DEFAULT_OUTPUT_LIMIT;
+}
 
 export function buildOpenCodeConfig({ modelConfig, tokenEnv = "FOURYI_CLI_TOKEN", orgEnv = "FOURYI_ORG_ID" }) {
   const models = {};
-  for (const model of modelConfig.models || []) {
+  const claudeModels = (modelConfig.models || []).filter(isClaudeModel);
+  for (const model of claudeModels) {
     models[model.id] = {
       name: model.display_name || model.id,
-      ...(model.context_length ? { limit: { context: model.context_length } } : {}),
+      limit: {
+        context: modelContextLimit(model),
+        output: modelOutputLimit(model),
+      },
     };
   }
+  const defaultModel = claudeModels.some((model) => model.id === modelConfig.default_model)
+    ? modelConfig.default_model
+    : claudeModels[0]?.id;
+
   return {
     "$schema": "https://opencode.ai/config.json",
-    model: modelConfig.default_model ? `4yi/${modelConfig.default_model}` : undefined,
+    model: defaultModel ? `4yi/${defaultModel}` : undefined,
+    enabled_providers: ["4yi"],
     provider: {
       "4yi": {
         name: "4YI",
@@ -35,6 +60,20 @@ export function buildOpenCodeConfig({ modelConfig, tokenEnv = "FOURYI_CLI_TOKEN"
   };
 }
 
+export function opencodeEnv({ home = os.homedir(), token, orgId = "", configFile, baseEnv = process.env } = {}) {
+  const xdgRoot = path.join(pathsForHome(home).opencodeConfigDir, "xdg");
+  return {
+    ...baseEnv,
+    FOURYI_CLI_TOKEN: token,
+    FOURYI_ORG_ID: orgId,
+    OPENCODE_CONFIG: configFile,
+    XDG_CONFIG_HOME: path.join(xdgRoot, "config"),
+    XDG_DATA_HOME: path.join(xdgRoot, "data"),
+    XDG_CACHE_HOME: path.join(xdgRoot, "cache"),
+    XDG_STATE_HOME: path.join(xdgRoot, "state"),
+  };
+}
+
 export function ensureOpenCodeRuntime({ home = os.homedir(), stdout = console.log } = {}) {
   const paths = pathsForHome(home);
   ensureDir(paths.opencodeDir);
@@ -64,21 +103,25 @@ export async function runCode({ session, home = os.homedir(), argv = [], stdout
   if (!session?.token) throw new Error("Not signed in. Run `4yi login`.");
   const modelConfig = await requestJson(session.baseUrl, "/api/cli/models", { token: session.token });
   if (!modelConfig.default_model) throw new Error("No chat models available for this organization.");
+  if (!(modelConfig.models || []).some(isClaudeModel)) throw new Error("No Claude models available for this organization.");
 
   const paths = pathsForHome(home);
   ensureDir(paths.opencodeConfigDir);
+  for (const dir of ["config", "data", "cache", "state"]) {
+    ensureDir(path.join(paths.opencodeConfigDir, "xdg", dir));
+  }
   fs.writeFileSync(paths.opencodeConfigFile, `${JSON.stringify(buildOpenCodeConfig({ modelConfig }), null, 2)}\n`, { mode: 0o600 });
 
   const bin = ensureOpenCodeRuntime({ home, stdout });
   stdout("Launching OpenCode with 4YI models...");
   const child = spawn(bin, argv, {
     stdio: "inherit",
-    env: {
-      ...process.env,
-      FOURYI_CLI_TOKEN: session.token,
-      FOURYI_ORG_ID: session.org?.id || "",
-      OPENCODE_CONFIG: paths.opencodeConfigFile,
-    },
+    env: opencodeEnv({
+      home,
+      token: session.token,
+      orgId: session.org?.id || "",
+      configFile: paths.opencodeConfigFile,
+    }),
   });
   return new Promise((resolve) => child.on("exit", (code) => resolve(code ?? 0)));
 }