@4yi/cli 0.1.2 → 0.1.3

package/package.json

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

package/src/opencode.mjs

@@ -6,18 +6,42 @@ 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,
     provider: {
       "4yi": {
         name: "4YI",
@@ -64,6 +88,7 @@ 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);