@yina-npm/openrouterx 0.4.55 → 0.4.71

package/app/src/mitm/antigravityBootstrap.js

@@ -1,53 +0,0 @@
-const LOCAL_ANTIGRAVITY_PROJECT = "local-openrouterx";
-
-function createLocalAntigravityTier(options = {}) {
-  const tier = {
-    id: "standard-tier",
-    name: "Gemini Code Assist",
-    description: "Unlimited coding assistant with the most powerful Gemini models",
-    userDefinedCloudaicompanionProject: true,
-    privacyNotice: {},
-    usesGcpTos: true,
-  };
-
-  if (options.isDefault) tier.isDefault = true;
-  return tier;
-}
-
-function createLocalAntigravityLoadCodeAssistPayload() {
-  const currentTier = createLocalAntigravityTier();
-
-  return {
-    cloudaicompanionProject: LOCAL_ANTIGRAVITY_PROJECT,
-    currentTier,
-    userTier: currentTier,
-    allowedTiers: [createLocalAntigravityTier({ isDefault: true })],
-    gcpManaged: false,
-    tosAccepted: true,
-    done: true,
-    releaseChannel: {
-      type: "EXPERIMENTAL",
-      name: "Preview Channel",
-    },
-    paidTier: currentTier,
-  };
-}
-
-function createLocalAntigravityOnboardUserPayload() {
-  const response = createLocalAntigravityLoadCodeAssistPayload();
-
-  return {
-    done: true,
-    cloudaicompanionProject: LOCAL_ANTIGRAVITY_PROJECT,
-    currentTier: response.currentTier,
-    allowedTiers: response.allowedTiers,
-    gcpManaged: false,
-    response,
-  };
-}
-
-module.exports = {
-  LOCAL_ANTIGRAVITY_PROJECT,
-  createLocalAntigravityLoadCodeAssistPayload,
-  createLocalAntigravityOnboardUserPayload,
-};

package/app/src/mitm/antigravityModels.js

@@ -1,265 +0,0 @@
-const REAL_AVAILABLE_MODELS_FIXTURE = require("./antigravityAvailableModels.fixture.json");
-
-const BASE_MODEL_EXPERIMENTS = {
-  experiments: {
-    CASCADE_USE_EXPERIMENT_CHECKPOINTER: {
-      stringValue: JSON.stringify({
-        strategy: "CHECKPOINT_STRATEGY_SINGLE_PROMPT",
-        max_token_limit: "128000",
-        token_threshold: "50000",
-        max_overhead_ratio: "0.15",
-        moving_window_size: "1",
-        enabled: true,
-        max_output_tokens: "16384",
-        checkpoint_model: "MODEL_PLACEHOLDER_M50",
-        retry_config: {
-          max_retries: 0,
-          initial_sleep_duration_ms: 1000,
-          exponential_multiplier: 2,
-          include_error_feedback: false,
-        },
-      }, null, 4),
-    },
-  },
-};
-
-const COMMON_CODE_MIME_TYPES = {
-  "application/json": true,
-  "application/pdf": true,
-  "application/rtf": true,
-  "application/x-ipynb+json": true,
-  "application/x-javascript": true,
-  "application/x-python-code": true,
-  "application/x-typescript": true,
-  "image/heic": true,
-  "image/heif": true,
-  "image/jpeg": true,
-  "image/png": true,
-  "image/webp": true,
-  "text/css": true,
-  "text/csv": true,
-  "text/html": true,
-  "text/javascript": true,
-  "text/markdown": true,
-  "text/plain": true,
-  "text/rtf": true,
-  "text/x-python": true,
-  "text/x-python-script": true,
-  "text/x-typescript": true,
-  "text/xml": true,
-  "video/mp4": true,
-  "video/webm": true,
-};
-
-function quota(resetDays = 7) {
-  const resetTime = new Date(Date.now() + resetDays * 24 * 60 * 60 * 1000).toISOString();
-  return { remainingFraction: 1, resetTime };
-}
-
-function geminiModel(displayName, model, options = {}) {
-  return {
-    displayName,
-    supportsImages: true,
-    supportsThinking: true,
-    thinkingBudget: options.thinkingBudget ?? 1024,
-    minThinkingBudget: options.minThinkingBudget ?? 128,
-    recommended: true,
-    maxTokens: 1048576,
-    maxOutputTokens: 65535,
-    tokenizerType: "LLAMA_WITH_SPECIAL",
-    quotaInfo: quota(options.resetDays),
-    model,
-    apiProvider: "API_PROVIDER_GOOGLE_GEMINI",
-    modelProvider: "MODEL_PROVIDER_GOOGLE",
-    supportsVideo: true,
-    supportedMimeTypes: COMMON_CODE_MIME_TYPES,
-    requiresImageOutputOutsideFunctionResponses: options.requiresImageOutputOutsideFunctionResponses,
-    modelExperiments: BASE_MODEL_EXPERIMENTS,
-  };
-}
-
-function anthropicModel(displayName, model) {
-  return {
-    displayName,
-    supportsImages: true,
-    supportsThinking: true,
-    thinkingBudget: 1024,
-    recommended: true,
-    maxTokens: 250000,
-    maxOutputTokens: 64000,
-    tokenizerType: "LLAMA_WITH_SPECIAL",
-    quotaInfo: quota(),
-    model,
-    apiProvider: "API_PROVIDER_ANTHROPIC_VERTEX",
-    modelProvider: "MODEL_PROVIDER_ANTHROPIC",
-    supportedMimeTypes: {
-      "image/heic": true,
-      "image/heif": true,
-      "image/jpeg": true,
-      "image/png": true,
-      "image/webp": true,
-    },
-    modelExperiments: BASE_MODEL_EXPERIMENTS,
-  };
-}
-
-function openAiModel(displayName, model) {
-  return {
-    displayName,
-    supportsThinking: true,
-    thinkingBudget: 8192,
-    recommended: true,
-    maxTokens: 131072,
-    maxOutputTokens: 32768,
-    tokenizerType: "LLAMA_WITH_SPECIAL",
-    quotaInfo: quota(),
-    model,
-    apiProvider: "API_PROVIDER_OPENAI_VERTEX",
-    modelProvider: "MODEL_PROVIDER_OPENAI",
-    modelExperiments: BASE_MODEL_EXPERIMENTS,
-  };
-}
-
-function tabModel(model, maxTokens = 16384) {
-  return {
-    maxTokens,
-    maxOutputTokens: 4096,
-    tokenizerType: "QWEN2",
-    quotaInfo: { remainingFraction: 1 },
-    model,
-    apiProvider: "API_PROVIDER_INTERNAL",
-    supportsCumulativeContext: true,
-    supportsEstimateTokenCounter: true,
-    isInternal: true,
-    promptTemplaterType: "PROMPT_TEMPLATER_TYPE_CHATML",
-    toolFormatterType: "TOOL_FORMATTER_TYPE_XML",
-    requiresLeadInGeneration: true,
-  };
-}
-
-function googleTabModel(model, maxTokens = 16384) {
-  return {
-    maxTokens,
-    maxOutputTokens: 4096,
-    tokenizerType: "LLAMA_WITH_SPECIAL",
-    quotaInfo: { remainingFraction: 1 },
-    model,
-    apiProvider: "API_PROVIDER_GOOGLE_GEMINI",
-    modelProvider: "MODEL_PROVIDER_GOOGLE",
-    supportsCumulativeContext: true,
-    supportsEstimateTokenCounter: true,
-    isInternal: true,
-    promptTemplaterType: "PROMPT_TEMPLATER_TYPE_CHATML",
-    toolFormatterType: "TOOL_FORMATTER_TYPE_XML",
-    requiresLeadInGeneration: true,
-  };
-}
-
-function imageModel(displayName, model) {
-  return {
-    displayName,
-    supportsImages: true,
-    maxTokens: 32768,
-    maxOutputTokens: 8192,
-    tokenizerType: "LLAMA_WITH_SPECIAL",
-    quotaInfo: quota(),
-    model,
-    apiProvider: "API_PROVIDER_GOOGLE_GEMINI",
-    modelProvider: "MODEL_PROVIDER_GOOGLE",
-    supportedMimeTypes: COMMON_CODE_MIME_TYPES,
-  };
-}
-
-function freshResetTime(days = 7) {
-  return new Date(Date.now() + days * 24 * 60 * 60 * 1000).toISOString();
-}
-
-function cloneFixtureWithFreshQuota() {
-  const payload = JSON.parse(JSON.stringify(REAL_AVAILABLE_MODELS_FIXTURE));
-  const agentIds = new Set(
-    (payload.agentModelSorts || [])
-      .flatMap((sort) => sort.groups || [])
-      .flatMap((group) => group.modelIds || []),
-  );
-
-  for (const [id, model] of Object.entries(payload.models || {})) {
-    if (!model.quotaInfo) model.quotaInfo = { remainingFraction: 1 };
-    model.quotaInfo.remainingFraction = typeof model.quotaInfo.remainingFraction === "number"
-      ? model.quotaInfo.remainingFraction
-      : 1;
-    if (model.quotaInfo.resetTime) {
-      model.quotaInfo.resetTime = freshResetTime(agentIds.has(id) ? 7 : 1);
-    }
-  }
-
-  return payload;
-}
-
-function buildAntigravityAvailableModelsResponse() {
-  if (REAL_AVAILABLE_MODELS_FIXTURE?.models) return cloneFixtureWithFreshQuota();
-
-  const models = {
-    "gemini-2.5-flash-thinking": geminiModel("Gemini 3.1 Flash Lite", "MODEL_GOOGLE_GEMINI_2_5_FLASH_THINKING"),
-    "gemini-2.5-pro": geminiModel("Gemini 2.5 Pro", "MODEL_GOOGLE_GEMINI_2_5_PRO", {
-      requiresImageOutputOutsideFunctionResponses: true,
-    }),
-    "gemini-2.5-flash-lite": geminiModel("Gemini 3.1 Flash Lite", "MODEL_GOOGLE_GEMINI_2_5_FLASH_LITE"),
-    "gemini-2.5-flash": geminiModel("Gemini 3.1 Flash Lite", "MODEL_GOOGLE_GEMINI_2_5_FLASH"),
-    "gemini-3.1-flash-lite": geminiModel("Gemini 3.1 Flash Lite", "MODEL_PLACEHOLDER_M50"),
-    "gemini-3.1-pro-high": geminiModel("Gemini 3.1 Pro (High)", "MODEL_PLACEHOLDER_M37", {
-      thinkingBudget: 10001,
-    }),
-    "gemini-3.1-pro-low": geminiModel("Gemini 3.1 Pro (Low)", "MODEL_PLACEHOLDER_M36", {
-      thinkingBudget: 1001,
-    }),
-    "gemini-3-flash-agent": geminiModel("Gemini 3 Flash", "MODEL_PLACEHOLDER_M84", {
-      thinkingBudget: -1,
-      minThinkingBudget: 32,
-    }),
-    "gemini-3-flash": geminiModel("Gemini 3 Flash", "MODEL_PLACEHOLDER_M18", {
-      thinkingBudget: -1,
-      minThinkingBudget: 32,
-    }),
-    "gemini-3.1-flash-image": imageModel("Gemini 3.1 Flash Image", "MODEL_PLACEHOLDER_M21"),
-    "claude-sonnet-4-6": anthropicModel("Claude Sonnet 4.6 (Thinking)", "MODEL_PLACEHOLDER_M35"),
-    "claude-opus-4-6-thinking": anthropicModel("Claude Opus 4.6 (Thinking)", "MODEL_PLACEHOLDER_M26"),
-    "gpt-oss-120b-medium": openAiModel("GPT-OSS 120B (Medium)", "MODEL_OPENAI_GPT_OSS_120B_MEDIUM"),
-    "tab_flash_lite_preview": googleTabModel("MODEL_PLACEHOLDER_M19"),
-    "tab_jump_flash_lite_preview": googleTabModel("MODEL_PLACEHOLDER_M28"),
-    "chat_20706": tabModel("MODEL_CHAT_20706"),
-    "chat_23310": tabModel("MODEL_CHAT_23310", 32768),
-  };
-
-  return {
-    models,
-    defaultAgentModelId: "gemini-3.1-pro-high",
-    agentModelSorts: [
-      {
-        displayName: "Recommended",
-        groups: [
-          {
-            modelIds: [
-              "gemini-3.1-pro-high",
-              "gemini-3.1-pro-low",
-              "gemini-3-flash-agent",
-              "claude-sonnet-4-6",
-              "claude-opus-4-6-thinking",
-              "gpt-oss-120b-medium",
-            ],
-          },
-        ],
-      },
-    ],
-    commandModelIds: ["gemini-3-flash"],
-    tabModelIds: ["chat_20706", "chat_23310"],
-    imageGenerationModelIds: [],
-    mqueryModelIds: ["gemini-3.1-flash-lite"],
-    webSearchModelIds: ["gemini-3.1-flash-lite"],
-    commitMessageModelIds: ["gemini-3-flash"],
-    audioTranscriptionModelIds: [],
-    experimentIds: [],
-    tieredModelIds: {},
-  };
-}
-
-module.exports = { buildAntigravityAvailableModelsResponse };

package/app/src/mitm/handlers/openrouter.js

@@ -1,201 +0,0 @@
-const fs = require("node:fs");
-const path = require("node:path");
-const { err } = require("../logger");
-const { fetchRouter, pipeSSE } = require("./base");
-
-const URL_MAP = {
-  "/api/v1/chat/completions": "/v1/chat/completions",
-  "/v1/chat/completions": "/v1/chat/completions",
-  "/chat/completions": "/v1/chat/completions",
-  "/v1/messages": "/v1/messages",
-  "/api/v1/responses": "/v1/responses",
-  "/v1/responses": "/v1/responses",
-  "/responses": "/v1/responses",
-  "/api/v1/models": "/v1/models",
-  "/v1/models": "/v1/models",
-};
-
-function resolveRouterPath(reqUrl) {
-  const [pathname, query = ""] = String(reqUrl || "").split("?");
-  for (const [pattern, routerPath] of Object.entries(URL_MAP)) {
-    if (pathname === pattern || pathname.endsWith(pattern)) {
-      return query ? `${routerPath}?${query}` : routerPath;
-    }
-  }
-  return "/v1/chat/completions";
-}
-
-function isModelsRequest(reqUrl) {
-  const pathname = String(reqUrl || "").split("?")[0];
-  return pathname === "/models" || pathname === "/v1/models" || pathname === "/api/v1/models" || pathname.endsWith("/models");
-}
-
-function isKeyInfoRequest(reqUrl) {
-  const pathname = String(reqUrl || "").split("?")[0];
-  return pathname === "/api/v1/key" || pathname === "/api/v1/auth/key";
-}
-
-function getLocalModelsPayload() {
-  const modelsPath = path.join(__dirname, "..", "models.json");
-  return fs.readFileSync(modelsPath, "utf8");
-}
-
-function sendJsonPayload(res, payload) {
-  if (res.writableEnded) return;
-  res.writeHead(200, {
-    "Content-Type": "application/json; charset=utf-8",
-    "Cache-Control": "public, max-age=300",
-    "Content-Length": Buffer.byteLength(payload),
-  });
-  res.end(payload);
-}
-
-function sendOptions(res) {
-  if (res.writableEnded) return;
-  res.writeHead(204, {
-    "Access-Control-Allow-Origin": "*",
-    "Access-Control-Allow-Methods": "GET, POST, OPTIONS",
-    "Access-Control-Allow-Headers": "*",
-  });
-  res.end();
-}
-
-function buildKeyInfoPayload() {
-  return {
-    data: {
-      label: "openrouterX MITM API Key",
-      limit: null,
-      usage: 0,
-      usage_daily: 0,
-      usage_weekly: 0,
-      usage_monthly: 0,
-      byok_usage: 0,
-      byok_usage_daily: 0,
-      byok_usage_weekly: 0,
-      byok_usage_monthly: 0,
-      is_free_tier: false,
-      limit_remaining: null,
-      limit_reset: null,
-      include_byok_in_limit: false,
-      is_provisioning_key: false,
-      is_management_key: false,
-      rate_limit: { interval: "1h", requests: 100000 },
-    },
-  };
-}
-
-function sendKeyInfo(res) {
-  if (res.writableEnded) return;
-  const payload = JSON.stringify(buildKeyInfoPayload());
-  res.writeHead(200, {
-    "Content-Type": "application/json; charset=utf-8",
-    "Access-Control-Allow-Origin": "*",
-    "Cache-Control": "no-store",
-    "Content-Length": Buffer.byteLength(payload),
-  });
-  res.end(payload);
-}
-
-function buildInjectedModel(publicModelId) {
-  const [provider = "openrouter"] = String(publicModelId || "").split("/");
-  return {
-    id: publicModelId,
-    canonical_slug: publicModelId,
-    name: publicModelId,
-    created: 0,
-    description: "Injected by OpenrouterX MITM mapping",
-    architecture: {
-      modality: "text->text",
-      input_modalities: ["text"],
-      output_modalities: ["text"],
-      tokenizer: "unknown",
-      instruct_type: null,
-    },
-    top_provider: {
-      is_moderated: false,
-    },
-    pricing: {
-      prompt: "0",
-      completion: "0",
-      image: "0",
-      request: "0",
-      web_search: "0",
-      internal_reasoning: "0",
-    },
-    per_request_limits: null,
-    context_length: 128000,
-    hugging_face_id: null,
-    supported_parameters: ["max_tokens", "temperature", "top_p", "stream", "stop"],
-    provider,
-  };
-}
-
-function injectPublicModels(payload, aliasMappings = {}) {
-  const normalized = payload && typeof payload === "object" ? payload : {};
-  const currentData = Array.isArray(normalized.data) ? normalized.data : [];
-  const existingIds = new Set(currentData.map((item) => item?.id).filter(Boolean));
-
-  const injected = Object.keys(aliasMappings)
-    .filter(Boolean)
-    .filter((publicId) => !existingIds.has(publicId))
-    .map(buildInjectedModel);
-
-  return {
-    ...normalized,
-    data: [...currentData, ...injected],
-  };
-}
-
-async function handleModelsRequest(req, res, bodyBuffer, passthrough, aliasMappings) {
-  try {
-    const payload = getLocalModelsPayload();
-    sendJsonPayload(res, payload);
-  } catch (error) {
-    err(`[openrouter] local models response failed: ${error.message}`);
-    if (!res.headersSent) res.writeHead(500, { "Content-Type": "application/json" });
-    if (!res.writableEnded) res.end(JSON.stringify({ error: { message: error.message, type: "mitm_error" } }));
-  }
-}
-
-/**
- * Intercept OpenRouter request — replace public model alias and forward to matching OpenrouterX endpoint.
- */
-async function intercept(req, res, bodyBuffer, mappedModel, passthrough, aliasMappings = {}) {
-  try {
-    if (req.method === "OPTIONS") {
-      sendOptions(res);
-      return;
-    }
-
-    if (req.method === "GET" && isKeyInfoRequest(req.url)) {
-      sendKeyInfo(res);
-      return;
-    }
-
-    if (isModelsRequest(req.url)) {
-      return handleModelsRequest(req, res, bodyBuffer, passthrough, aliasMappings);
-    }
-
-    const body = JSON.parse(bodyBuffer.toString());
-    const resolvedModel = mappedModel || aliasMappings[body.model] || body.model;
-    body.model = resolvedModel;
-    const routerPath = resolveRouterPath(req.url);
-    const routerRes = await fetchRouter(body, routerPath, req.headers);
-    await pipeSSE(routerRes, res);
-  } catch (error) {
-    err(`[openrouter] ${error.message}`);
-    if (!res.headersSent) res.writeHead(500, { "Content-Type": "application/json" });
-    res.end(JSON.stringify({ error: { message: error.message, type: "mitm_error" } }));
-  }
-}
-
-module.exports = {
-  intercept,
-  resolveRouterPath,
-  isModelsRequest,
-  isKeyInfoRequest,
-  sendKeyInfo,
-  buildKeyInfoPayload,
-  injectPublicModels,
-  buildInjectedModel,
-};