replicas-engine 0.1.545 → 0.1.546

package/dist/src/index.js

@@ -702,7 +702,7 @@ var WORKSPACE_SIZES = ["small", "large"];
 var INVALID_WORKSPACE_SIZE_ERROR = `Invalid size: must be one of ${WORKSPACE_SIZES.join(", ")}`;
 
 // ../shared/src/e2b.ts
-var E2B_TEMPLATE_NAME = "replicas-sandbox-2026-08-02-v1";
+var E2B_TEMPLATE_NAME = "replicas-sandbox-2026-08-02-v2";
 
 // ../shared/src/runtime-env.ts
 function shellQuotePosix(value) {
@@ -2772,6 +2772,7 @@ var DEFAULT_USER_PREFERENCES = {
   default_agent: "codex",
   default_model: null,
   cursor_models: null,
+  openrouter_models: null,
   close_workspace_on_pr_close: true,
   close_workspace_on_pr_merge: true,
   default_environment_id: null,
@@ -5258,6 +5259,18 @@ var MonolithService = class {
       return null;
     }
   }
+  async getOpenRouterModels() {
+    try {
+      const response = await monolithRequest("/v1/engine/openrouter-models", { method: "GET" });
+      if (!response.ok) {
+        return null;
+      }
+      const body = await response.json();
+      return isRecord(body) && Array.isArray(body.models) && body.models.every((model) => typeof model === "string") ? body.models : null;
+    } catch {
+      return null;
+    }
+  }
   async sendEvent(event) {
     if (!ENGINE_ENV.WORKSPACE_ID) {
       return;
@@ -10263,7 +10276,7 @@ var DEFAULT_CODEX_ARGS = ["app-server", "--listen", "stdio://"];
 var MIN_CODEX_CLI_VERSION = "0.144.6";
 var CODEX_UPGRADE_TIMEOUT_MS = 12e4;
 var codexCliVersionEnsured = null;
-var ENGINE_PACKAGE_VERSION = "0.1.545";
+var ENGINE_PACKAGE_VERSION = "0.1.546";
 var INITIALIZE_METHOD = "initialize";
 var INITIALIZED_NOTIFICATION = "initialized";
 var ACCOUNT_LOGIN_START_METHOD = "account/login/start";
@@ -12887,6 +12900,25 @@ import {
   createOpencodeClient,
   createOpencodeServer
 } from "@opencode-ai/sdk/v2";
+
+// src/services/openrouter-models-service.ts
+var OPENROUTER_MODELS_CACHE_MS = 6e4;
+var OPENROUTER_MODELS_STALE_GRACE_MS = 10 * 6e4;
+var cache = null;
+async function getAllowedOpenRouterModels() {
+  const now = Date.now();
+  if (cache && now - cache.fetchedAt < OPENROUTER_MODELS_CACHE_MS) return cache.models;
+  const models = await monolithService.getOpenRouterModels();
+  if (models?.length) {
+    cache = { models, fetchedAt: now };
+    return models;
+  }
+  if (cache && now - cache.fetchedAt < OPENROUTER_MODELS_STALE_GRACE_MS) return cache.models;
+  cache = null;
+  throw new Error("Could not load the organization's allowed OpenRouter models; refusing to run an unverified model.");
+}
+
+// src/managers/opencode-manager.ts
 var OPENCODE_SHIM_DIR = dirname6(fileURLToPath(new URL("../../scripts/opencode", import.meta.url)));
 var OPENCODE_CONFIG_PATH = join20(ENGINE_ENV.HOME_DIR, ".config", "opencode", "opencode.json");
 var OPENCODE_FETCH_DISPATCHER = new Agent({ headersTimeout: 0, bodyTimeout: 0 });
@@ -12936,10 +12968,10 @@ async function getOpencodeProvider() {
   if (await hasOpencodeCredentials(ASTER_PROVIDER)) return ASTER_PROVIDER;
   return "openrouter";
 }
-function getDefaultOpencodeModel(provider) {
+async function getDefaultOpencodeModel(provider) {
   if (provider === OPENCODE_GO_PROVIDER) return DEFAULT_OPENCODE_GO_MODEL;
   if (provider === ASTER_PROVIDER) return DEFAULT_ASTER_MODEL;
-  return DEFAULT_OPENCODE_MODEL;
+  return (await getAllowedOpenRouterModels())[0] ?? DEFAULT_OPENCODE_MODEL;
 }
 function getOpenCodeGoModel(model) {
   return model.replace(/^[^/]+\//, "");
@@ -12986,7 +13018,7 @@ function opencodeConfig(provider, model) {
   return config;
 }
 function getConfiguredOpencodeModels(model, provider) {
-  return [.../* @__PURE__ */ new Set([model, ...provider === ASTER_PROVIDER ? ASTER_MODELS : OPENROUTER_MODELS])];
+  return [.../* @__PURE__ */ new Set([model, ...provider === ASTER_PROVIDER ? ASTER_MODELS : []])];
 }
 function opencodeCommandToSlashCommand(command) {
   return createProviderSlashCommand("opencode", command.name, command.description);
@@ -13177,7 +13209,7 @@ var OpencodeManager = class extends CodingAgentManager {
     this.slashCommandsRequest ??= (async () => {
       try {
         const provider = await getOpencodeProvider();
-        const client = await this.ensureClient(getDefaultOpencodeModel(provider), provider);
+        const client = await this.ensureClient(await getDefaultOpencodeModel(provider), provider);
         const directories = [this.workingDirectory, ...await getAgentAdditionalDirectories()];
         const perDirectory = await Promise.all(directories.map(async (directory) => {
           try {
@@ -13217,7 +13249,7 @@ var OpencodeManager = class extends CodingAgentManager {
     }
     this.providerId = providerId;
     const configuredModel = providerId === OPENCODE_GO_PROVIDER ? getOpenCodeGoModel(model) : model;
-    const providerModels = providerId === ASTER_PROVIDER ? ASTER_MODELS : providerId === "openrouter" ? OPENROUTER_MODELS : null;
+    const providerModels = providerId === ASTER_PROVIDER ? ASTER_MODELS : providerId === "openrouter" ? await getAllowedOpenRouterModels() : null;
     if (providerModels && !providerModels.includes(configuredModel)) {
       throw new Error(`Model ${configuredModel} is not available through ${providerId}.`);
     }
@@ -13317,7 +13349,6 @@ var OpencodeManager = class extends CodingAgentManager {
   }
   async processMessageInternal(request) {
     const provider = await getOpencodeProvider();
-    const model = request.model ?? getDefaultOpencodeModel(provider);
     const controller = new AbortController();
     const linearSessionId = ENGINE_ENV.LINEAR_SESSION_ID;
     const linearForwarder = new LinearEventForwarder(linearSessionId);
@@ -13325,6 +13356,7 @@ var OpencodeManager = class extends CodingAgentManager {
     this.activeLinearForwarder = linearForwarder;
     this.forwardedLinearPartKeys.clear();
     try {
+      const model = request.model ?? await getDefaultOpencodeModel(provider);
       const client = await this.ensureClient(model, provider);
       const providerModel = this.providerId === OPENCODE_GO_PROVIDER ? getOpenCodeGoModel(model) : model;
       const agent = request.planMode ? "plan" : "build";
@@ -13643,6 +13675,30 @@ import {
   SessionManager,
   DefaultResourceLoader
 } from "@earendil-works/pi-coding-agent";
+function registerOpenRouterModels(modelRegistry, apiKey, modelIds) {
+  const openRouterModels = modelRegistry.getAll().filter((model) => model.provider === "openrouter");
+  const modelTemplate = openRouterModels[0];
+  if (!modelTemplate) throw new Error("Pi OpenAI-compatible model template is unavailable.");
+  modelRegistry.registerProvider("openrouter", {
+    api: modelTemplate.api,
+    apiKey,
+    baseUrl: modelTemplate.baseUrl,
+    models: [
+      ...openRouterModels,
+      ...modelIds.filter((id) => !openRouterModels.some((model) => model.id === id)).map((id) => ({
+        id,
+        name: id,
+        api: modelTemplate.api,
+        baseUrl: modelTemplate.baseUrl,
+        reasoning: modelTemplate.reasoning,
+        input: modelTemplate.input,
+        cost: modelTemplate.cost,
+        contextWindow: modelTemplate.contextWindow,
+        maxTokens: modelTemplate.maxTokens
+      }))
+    ]
+  });
+}
 function eventPayload(event) {
   return isRecord4(event) ? { ...event } : { value: event };
 }
@@ -13674,6 +13730,7 @@ var PiManager = class extends CodingAgentManager {
   historyFilePath;
   historyFile;
   providerId = "openrouter";
+  providerApiKey = null;
   constructor(options) {
     super(options);
     this.historyFilePath = options.historyFilePath ?? join21(PI_HISTORY_DIR, `${Date.now()}.jsonl`);
@@ -13718,8 +13775,16 @@ var PiManager = class extends CodingAgentManager {
   }
   async ensureSession(request) {
     if (this.session) {
+      const turnModel = request.model ?? this.session.model?.id;
+      if (this.providerId === "openrouter" && turnModel && !(await getAllowedOpenRouterModels()).includes(turnModel)) {
+        throw new Error(`Pi model is not available through ${this.providerId}: ${turnModel}`);
+      }
       if (request.model && request.model !== this.session.model?.id) {
-        const model2 = this.session.modelRegistry.find(this.providerId, request.model);
+        let model2 = this.session.modelRegistry.find(this.providerId, request.model);
+        if (!model2 && this.providerId === "openrouter" && this.providerApiKey) {
+          registerOpenRouterModels(this.session.modelRegistry, this.providerApiKey, [request.model]);
+          model2 = this.session.modelRegistry.find(this.providerId, request.model);
+        }
         if (!model2) throw new Error(`Pi model is not available through ${this.providerId}: ${request.model}`);
         await this.session.setModel(model2);
       }
@@ -13732,10 +13797,8 @@ var PiManager = class extends CodingAgentManager {
     const credentials = this.getProviderCredentials(authStorage);
     if (!credentials) throw new Error("Aster or OpenRouter API key is not configured for Pi.");
     this.providerId = credentials.provider;
+    this.providerApiKey = credentials.apiKey;
     const modelRegistry = ModelRegistry.create(authStorage);
-    const openRouterModels = modelRegistry.getAll().filter((candidate) => candidate.provider === "openrouter");
-    const modelTemplate = openRouterModels[0];
-    if (!modelTemplate) throw new Error("Pi OpenAI-compatible model template is unavailable.");
     if (this.providerId === ASTER_PROVIDER) {
       modelRegistry.registerProvider(ASTER_PROVIDER, {
         name: "Aster",
@@ -13754,30 +13817,16 @@ var PiManager = class extends CodingAgentManager {
           maxTokens: 65536
         }))
       });
-    } else {
-      modelRegistry.registerProvider("openrouter", {
-        api: modelTemplate.api,
-        apiKey: credentials.apiKey,
-        baseUrl: modelTemplate.baseUrl,
-        models: [
-          ...openRouterModels,
-          ...OPENROUTER_MODELS.filter((id) => !openRouterModels.some((candidate) => candidate.id === id)).map((id) => ({
-            id,
-            name: id,
-            api: modelTemplate.api,
-            baseUrl: modelTemplate.baseUrl,
-            reasoning: modelTemplate.reasoning,
-            input: modelTemplate.input,
-            cost: modelTemplate.cost,
-            contextWindow: modelTemplate.contextWindow,
-            maxTokens: modelTemplate.maxTokens
-          }))
-        ]
-      });
     }
-    const defaultModel = this.providerId === ASTER_PROVIDER ? DEFAULT_ASTER_MODEL : DEFAULT_PI_MODEL;
+    let defaultModel = DEFAULT_ASTER_MODEL;
+    let allowedModels = null;
+    if (this.providerId !== ASTER_PROVIDER) {
+      allowedModels = await getAllowedOpenRouterModels();
+      registerOpenRouterModels(modelRegistry, credentials.apiKey, allowedModels);
+      defaultModel = allowedModels[0] ?? DEFAULT_PI_MODEL;
+    }
     const modelId = request.model ?? defaultModel;
-    const model = modelRegistry.find(this.providerId, modelId);
+    const model = allowedModels && !allowedModels.includes(modelId) ? null : modelRegistry.find(this.providerId, modelId);
     if (!model) throw new Error(`Pi model is not available through ${this.providerId}: ${modelId}`);
     const sessionManager = this.initialSessionId ? SessionManager.open(this.initialSessionId, PI_HISTORY_DIR, this.workingDirectory) : SessionManager.create(this.workingDirectory, PI_HISTORY_DIR);
     const resourceLoader = new DefaultResourceLoader({
@@ -13816,9 +13865,9 @@ var PiManager = class extends CodingAgentManager {
     this.recordHistoryEvent(`pi-${event.type}`, payload, this.historyFile);
   }
   async processMessageInternal(request) {
-    const session = await this.ensureSession(request);
-    this.recordHistoryEvent("event_msg", { type: "user_message", message: request.message }, this.historyFile);
     try {
+      const session = await this.ensureSession(request);
+      this.recordHistoryEvent("event_msg", { type: "user_message", message: request.message }, this.historyFile);
       const images = request.images && request.images.length > 0 ? (await normalizeImages(request.images)).map((image) => ({
         type: "image",
         data: image.source.data,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "replicas-engine",
-  "version": "0.1.545",
+  "version": "0.1.546",
   "description": "Lightweight API server for Replicas workspaces",
   "type": "module",
   "main": "dist/src/index.js",