@cortexkit/antigravity-auth-core 1.0.0

package/dist/model-registry.js

@@ -0,0 +1,214 @@
+const DEFAULT_MODALITIES = {
+    input: ["text", "image", "pdf"],
+    output: ["text"],
+};
+const MODEL_RELEASE_DATE = "";
+const DEFAULT_COST = { input: 0, output: 0 };
+const DEFAULT_OPTIONS = {};
+function defineModel(id, model) {
+    return {
+        id,
+        release_date: MODEL_RELEASE_DATE,
+        attachment: model.modalities.input.some((modality) => modality !== "text"),
+        temperature: true,
+        tool_call: true,
+        cost: { ...DEFAULT_COST },
+        options: { ...DEFAULT_OPTIONS },
+        ...model,
+    };
+}
+const ALL_MODEL_DEFINITIONS = {
+    "antigravity-gemini-3.1-pro": defineModel("antigravity-gemini-3.1-pro", {
+        name: "Gemini 3.1 Pro (Antigravity)",
+        reasoning: true,
+        limit: { context: 1048576, output: 65535 },
+        modalities: DEFAULT_MODALITIES,
+        variants: {
+            low: { thinkingLevel: "low" },
+            high: { thinkingLevel: "high" },
+        },
+    }),
+    "antigravity-gemini-3.5-flash": defineModel("antigravity-gemini-3.5-flash", {
+        name: "Gemini 3.5 Flash (Antigravity)",
+        reasoning: true,
+        limit: { context: 1048576, output: 65536 },
+        modalities: DEFAULT_MODALITIES,
+        variants: {
+            low: { thinkingLevel: "low" },
+            high: { thinkingLevel: "high" },
+        },
+    }),
+    "antigravity-claude-sonnet-4-6-thinking": defineModel("antigravity-claude-sonnet-4-6-thinking", {
+        name: "Claude Sonnet 4.6 Thinking (Antigravity)",
+        reasoning: true,
+        limit: { context: 250000, output: 64000 },
+        modalities: DEFAULT_MODALITIES,
+        variants: {
+            low: { disabled: true },
+            high: { disabled: true },
+        },
+    }),
+    "antigravity-claude-opus-4-6-thinking": defineModel("antigravity-claude-opus-4-6-thinking", {
+        name: "Claude Opus 4.6 Thinking (Antigravity)",
+        reasoning: true,
+        limit: { context: 250000, output: 64000 },
+        modalities: DEFAULT_MODALITIES,
+        variants: {
+            low: { disabled: true },
+            high: { disabled: true },
+        },
+    }),
+    "antigravity-gemini-3.1-flash-image": defineModel("antigravity-gemini-3.1-flash-image", {
+        name: "Gemini 3.1 Flash Image (Antigravity)",
+        reasoning: false,
+        limit: { context: 66000, output: 33000 },
+        modalities: {
+            input: ["text", "image"],
+            output: ["text", "image"],
+        },
+    }),
+    "antigravity-gpt-oss-120b": defineModel("antigravity-gpt-oss-120b", {
+        name: "GPT-OSS 120B (Antigravity)",
+        reasoning: false,
+        limit: { context: 128000, output: 16384 },
+        modalities: DEFAULT_MODALITIES,
+        variants: {
+            medium: {},
+        },
+    }),
+    "gemini-2.5-flash": defineModel("gemini-2.5-flash", {
+        name: "Gemini 2.5 Flash (Gemini CLI)",
+        reasoning: true,
+        limit: { context: 1048576, output: 65536 },
+        modalities: DEFAULT_MODALITIES,
+    }),
+    "gemini-2.5-pro": defineModel("gemini-2.5-pro", {
+        name: "Gemini 2.5 Pro (Gemini CLI)",
+        reasoning: true,
+        limit: { context: 1048576, output: 65536 },
+        modalities: DEFAULT_MODALITIES,
+    }),
+    "gemini-3-flash-preview": defineModel("gemini-3-flash-preview", {
+        name: "Gemini 3 Flash Preview (Gemini CLI)",
+        reasoning: true,
+        limit: { context: 1048576, output: 65536 },
+        modalities: DEFAULT_MODALITIES,
+    }),
+    "gemini-3.1-pro-preview": defineModel("gemini-3.1-pro-preview", {
+        name: "Gemini 3.1 Pro Preview (Gemini CLI)",
+        reasoning: true,
+        limit: { context: 1048576, output: 65535 },
+        modalities: DEFAULT_MODALITIES,
+    }),
+    "gemini-3.5-flash-preview": defineModel("gemini-3.5-flash-preview", {
+        name: "Gemini 3.5 Flash Preview (Gemini CLI)",
+        reasoning: true,
+        limit: { context: 1048576, output: 65536 },
+        modalities: DEFAULT_MODALITIES,
+    }),
+    "gemini-3.1-flash-image": defineModel("gemini-3.1-flash-image", {
+        name: "Gemini 3.1 Flash Image (Gemini CLI)",
+        reasoning: false,
+        limit: { context: 66000, output: 33000 },
+        modalities: {
+            input: ["text", "image"],
+            output: ["text", "image"],
+        },
+    }),
+    "gemini-3.1-flash-image-preview": defineModel("gemini-3.1-flash-image-preview", {
+        name: "Gemini 3.1 Flash Image Preview (Gemini CLI)",
+        reasoning: false,
+        limit: { context: 66000, output: 33000 },
+        modalities: {
+            input: ["text", "image"],
+            output: ["text", "image"],
+        },
+    }),
+    "gemini-3.1-pro-preview-customtools": defineModel("gemini-3.1-pro-preview-customtools", {
+        name: "Gemini 3.1 Pro Preview Custom Tools (Gemini CLI)",
+        reasoning: true,
+        limit: { context: 1048576, output: 65535 },
+        modalities: DEFAULT_MODALITIES,
+    }),
+};
+const RESOLVER_ALIASES = {
+    "gemini-3.1-pro-low": "gemini-3.1-pro",
+    "gemini-3.1-pro-high": "gemini-3.1-pro",
+    "gemini-3-flash-low": "gemini-3-flash",
+    "gemini-3-flash-medium": "gemini-3-flash",
+    "gemini-3-flash-high": "gemini-3-flash",
+    "gemini-3.5-flash-low": "gemini-3.5-flash",
+    "gemini-3.5-flash-medium": "gemini-3.5-flash",
+    "gemini-3.5-flash-high": "gemini-3.5-flash",
+    "gemini-claude-opus-4-6-thinking-low": "claude-opus-4-6-thinking",
+    "gemini-claude-opus-4-6-thinking-medium": "claude-opus-4-6-thinking",
+    "gemini-claude-opus-4-6-thinking-high": "claude-opus-4-6-thinking",
+    "gemini-claude-sonnet-4-6-thinking-low": "claude-sonnet-4-6",
+    "gemini-claude-sonnet-4-6-thinking-medium": "claude-sonnet-4-6",
+    "gemini-claude-sonnet-4-6-thinking-high": "claude-sonnet-4-6",
+    "gemini-claude-sonnet-4-6": "claude-sonnet-4-6",
+    "claude-sonnet-4-6-thinking": "claude-sonnet-4-6",
+    "claude-sonnet-4-6-thinking-low": "claude-sonnet-4-6",
+    "claude-sonnet-4-6-thinking-medium": "claude-sonnet-4-6",
+    "claude-sonnet-4-6-thinking-high": "claude-sonnet-4-6",
+};
+const GEMINI_35_FLASH_ROUTES = {
+    antigravity: {
+        defaultModel: "gemini-3-flash-agent",
+        byTier: {
+            low: "gemini-3.5-flash-extra-low",
+            medium: "gemini-3.5-flash-low",
+            high: "gemini-3-flash-agent",
+        },
+    },
+    geminiCliFallbackModel: "gemini-3-flash-preview",
+};
+const QUOTA_GROUP_BY_MODEL_ID = {
+    "claude-opus-4-6-thinking": "claude",
+    "claude-opus-4-6": "claude",
+    "claude-sonnet-4-6-thinking": "claude",
+    "claude-sonnet-4-6": "claude",
+    "gemini-pro-agent": "gemini-pro",
+    "gemini-3.1-pro": "gemini-pro",
+    "gemini-3.1-pro-low": "gemini-pro",
+    "gemini-3.1-pro-high": "gemini-pro",
+    "gemini-3-flash": "gemini-flash",
+    "gemini-3-flash-agent": "gemini-flash",
+    "gemini-3.5-flash-low": "gemini-flash",
+    "gemini-3.5-flash-extra-low": "gemini-flash",
+    "gemini-3.1-flash-image": "gemini-flash",
+    "gpt-oss-120b": "gpt-oss",
+    "gpt-oss-120b-medium": "gpt-oss",
+};
+const ANTIGRAVITY_OPENCODE_MODEL_IDS = [
+    "antigravity-gemini-3.5-flash",
+    "antigravity-gemini-3.1-pro",
+    "antigravity-claude-sonnet-4-6-thinking",
+    "antigravity-claude-opus-4-6-thinking",
+];
+function pickModelDefinitions(ids) {
+    return Object.fromEntries(ids.map((id) => [id, ALL_MODEL_DEFINITIONS[id]]));
+}
+export const OPENCODE_MODEL_DEFINITIONS = pickModelDefinitions(ANTIGRAVITY_OPENCODE_MODEL_IDS);
+export function getPublicModelDefinitions() {
+    return OPENCODE_MODEL_DEFINITIONS;
+}
+export function getAntigravityOpencodeModelIds() {
+    return [...ANTIGRAVITY_OPENCODE_MODEL_IDS];
+}
+export function getResolverAliasMap() {
+    return RESOLVER_ALIASES;
+}
+export function getGemini35FlashAntigravityModel(tier) {
+    if (!tier) {
+        return GEMINI_35_FLASH_ROUTES.antigravity.defaultModel;
+    }
+    return GEMINI_35_FLASH_ROUTES.antigravity.byTier[tier] ?? GEMINI_35_FLASH_ROUTES.antigravity.defaultModel;
+}
+export function getGemini35FlashGeminiCliFallbackModel() {
+    return GEMINI_35_FLASH_ROUTES.geminiCliFallbackModel;
+}
+export function getQuotaGroupForModel(modelId) {
+    return QUOTA_GROUP_BY_MODEL_ID[modelId.toLowerCase()];
+}
+//# sourceMappingURL=model-registry.js.map

package/dist/model-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-registry.js","sourceRoot":"","sources":["../src/model-registry.ts"],"names":[],"mappings":"AAgEA,MAAM,kBAAkB,GAAoB;IAC1C,KAAK,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC;IAC/B,MAAM,EAAE,CAAC,MAAM,CAAC;CACjB,CAAA;AAED,MAAM,kBAAkB,GAAG,EAAE,CAAA;AAC7B,MAAM,YAAY,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,CAAA;AAC5C,MAAM,eAAe,GAA4B,EAAE,CAAA;AAEnD,SAAS,WAAW,CAClB,EAAU,EACV,KAAmC;IAEnC,OAAO;QACL,EAAE;QACF,YAAY,EAAE,kBAAkB;QAChC,UAAU,EAAE,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,KAAK,MAAM,CAAC;QAC1E,WAAW,EAAE,IAAI;QACjB,SAAS,EAAE,IAAI;QACf,IAAI,EAAE,EAAE,GAAG,YAAY,EAAE;QACzB,OAAO,EAAE,EAAE,GAAG,eAAe,EAAE;QAC/B,GAAG,KAAK;KACT,CAAA;AACH,CAAC;AAED,MAAM,qBAAqB,GAA6B;IACtD,4BAA4B,EAAE,WAAW,CAAC,4BAA4B,EAAE;QACtE,IAAI,EAAE,8BAA8B;QACpC,SAAS,EAAE,IAAI;QACf,KAAK,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE;QAC1C,UAAU,EAAE,kBAAkB;QAC9B,QAAQ,EAAE;YACR,GAAG,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE;YAC7B,IAAI,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE;SAChC;KACF,CAAC;IACF,8BAA8B,EAAE,WAAW,CAAC,8BAA8B,EAAE;QAC1E,IAAI,EAAE,gCAAgC;QACtC,SAAS,EAAE,IAAI;QACf,KAAK,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE;QAC1C,UAAU,EAAE,kBAAkB;QAC9B,QAAQ,EAAE;YACR,GAAG,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE;YAC7B,IAAI,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE;SAChC;KACF,CAAC;IACF,wCAAwC,EAAE,WAAW,CAAC,wCAAwC,EAAE;QAC9F,IAAI,EAAE,0CAA0C;QAChD,SAAS,EAAE,IAAI;QACf,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE;QACzC,UAAU,EAAE,kBAAkB;QAC9B,QAAQ,EAAE;YACR,GAAG,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE;YACvB,IAAI,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE;SACzB;KACF,CAAC;IACF,sCAAsC,EAAE,WAAW,CAAC,sCAAsC,EAAE;QAC1F,IAAI,EAAE,wCAAwC;QAC9C,SAAS,EAAE,IAAI;QACf,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE;QACzC,UAAU,EAAE,kBAAkB;QAC9B,QAAQ,EAAE;YACR,GAAG,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE;YACvB,IAAI,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE;SACzB;KACF,CAAC;IACF,oCAAoC,EAAE,WAAW,CAAC,oCAAoC,EAAE;QACtF,IAAI,EAAE,sCAAsC;QAC5C,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE;QACxC,UAAU,EAAE;YACV,KAAK,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;YACxB,MAAM,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;SAC1B;KACF,CAAC;IACF,0BAA0B,EAAE,WAAW,CAAC,0BAA0B,EAAE;QAClE,IAAI,EAAE,4BAA4B;QAClC,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE;QACzC,UAAU,EAAE,kBAAkB;QAC9B,QAAQ,EAAE;YACR,MAAM,EAAE,EAAE;SACX;KACF,CAAC;IACF,kBAAkB,EAAE,WAAW,CAAC,kBAAkB,EAAE;QAClD,IAAI,EAAE,+BAA+B;QACrC,SAAS,EAAE,IAAI;QACf,KAAK,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE;QAC1C,UAAU,EAAE,kBAAkB;KAC/B,CAAC;IACF,gBAAgB,EAAE,WAAW,CAAC,gBAAgB,EAAE;QAC9C,IAAI,EAAE,6BAA6B;QACnC,SAAS,EAAE,IAAI;QACf,KAAK,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE;QAC1C,UAAU,EAAE,kBAAkB;KAC/B,CAAC;IACF,wBAAwB,EAAE,WAAW,CAAC,wBAAwB,EAAE;QAC9D,IAAI,EAAE,qCAAqC;QAC3C,SAAS,EAAE,IAAI;QACf,KAAK,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE;QAC1C,UAAU,EAAE,kBAAkB;KAC/B,CAAC;IACF,wBAAwB,EAAE,WAAW,CAAC,wBAAwB,EAAE;QAC9D,IAAI,EAAE,qCAAqC;QAC3C,SAAS,EAAE,IAAI;QACf,KAAK,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE;QAC1C,UAAU,EAAE,kBAAkB;KAC/B,CAAC;IACF,0BAA0B,EAAE,WAAW,CAAC,0BAA0B,EAAE;QAClE,IAAI,EAAE,uCAAuC;QAC7C,SAAS,EAAE,IAAI;QACf,KAAK,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE;QAC1C,UAAU,EAAE,kBAAkB;KAC/B,CAAC;IACF,wBAAwB,EAAE,WAAW,CAAC,wBAAwB,EAAE;QAC9D,IAAI,EAAE,qCAAqC;QAC3C,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE;QACxC,UAAU,EAAE;YACV,KAAK,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;YACxB,MAAM,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;SAC1B;KACF,CAAC;IACF,gCAAgC,EAAE,WAAW,CAAC,gCAAgC,EAAE;QAC9E,IAAI,EAAE,6CAA6C;QACnD,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE;QACxC,UAAU,EAAE;YACV,KAAK,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;YACxB,MAAM,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;SAC1B;KACF,CAAC;IACF,oCAAoC,EAAE,WAAW,CAAC,oCAAoC,EAAE;QACtF,IAAI,EAAE,kDAAkD;QACxD,SAAS,EAAE,IAAI;QACf,KAAK,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE;QAC1C,UAAU,EAAE,kBAAkB;KAC/B,CAAC;CACH,CAAA;AAED,MAAM,gBAAgB,GAA2B;IAC/C,oBAAoB,EAAE,gBAAgB;IACtC,qBAAqB,EAAE,gBAAgB;IACvC,oBAAoB,EAAE,gBAAgB;IACtC,uBAAuB,EAAE,gBAAgB;IACzC,qBAAqB,EAAE,gBAAgB;IACvC,sBAAsB,EAAE,kBAAkB;IAC1C,yBAAyB,EAAE,kBAAkB;IAC7C,uBAAuB,EAAE,kBAAkB;IAC3C,qCAAqC,EAAE,0BAA0B;IACjE,wCAAwC,EAAE,0BAA0B;IACpE,sCAAsC,EAAE,0BAA0B;IAClE,uCAAuC,EAAE,mBAAmB;IAC5D,0CAA0C,EAAE,mBAAmB;IAC/D,wCAAwC,EAAE,mBAAmB;IAC7D,0BAA0B,EAAE,mBAAmB;IAC/C,4BAA4B,EAAE,mBAAmB;IACjD,gCAAgC,EAAE,mBAAmB;IACrD,mCAAmC,EAAE,mBAAmB;IACxD,iCAAiC,EAAE,mBAAmB;CACvD,CAAA;AAED,MAAM,sBAAsB,GAA+B;IACzD,WAAW,EAAE;QACX,YAAY,EAAE,sBAAsB;QACpC,MAAM,EAAE;YACN,GAAG,EAAE,4BAA4B;YACjC,MAAM,EAAE,sBAAsB;YAC9B,IAAI,EAAE,sBAAsB;SAC7B;KACF;IACD,sBAAsB,EAAE,wBAAwB;CACjD,CAAA;AAED,MAAM,uBAAuB,GAAoC;IAC/D,0BAA0B,EAAE,QAAQ;IACpC,iBAAiB,EAAE,QAAQ;IAC3B,4BAA4B,EAAE,QAAQ;IACtC,mBAAmB,EAAE,QAAQ;IAC7B,kBAAkB,EAAE,YAAY;IAChC,gBAAgB,EAAE,YAAY;IAC9B,oBAAoB,EAAE,YAAY;IAClC,qBAAqB,EAAE,YAAY;IACnC,gBAAgB,EAAE,cAAc;IAChC,sBAAsB,EAAE,cAAc;IACtC,sBAAsB,EAAE,cAAc;IACtC,4BAA4B,EAAE,cAAc;IAC5C,wBAAwB,EAAE,cAAc;IACxC,cAAc,EAAE,SAAS;IACzB,qBAAqB,EAAE,SAAS;CACjC,CAAA;AAED,MAAM,8BAA8B,GAAG;IACrC,8BAA8B;IAC9B,4BAA4B;IAC5B,wCAAwC;IACxC,sCAAsC;CAC9B,CAAA;AAEV,SAAS,oBAAoB,CAAC,GAAsB;IAClD,OAAO,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,qBAAqB,CAAC,EAAE,CAAE,CAAC,CAAC,CAAC,CAAA;AAC9E,CAAC;AAED,MAAM,CAAC,MAAM,0BAA0B,GAAG,oBAAoB,CAAC,8BAA8B,CAAC,CAAA;AAE9F,MAAM,UAAU,yBAAyB;IACvC,OAAO,0BAA0B,CAAA;AACnC,CAAC;AAED,MAAM,UAAU,8BAA8B;IAC5C,OAAO,CAAC,GAAG,8BAA8B,CAAC,CAAA;AAC5C,CAAC;AAED,MAAM,UAAU,mBAAmB;IACjC,OAAO,gBAAgB,CAAA;AACzB,CAAC;AAED,MAAM,UAAU,gCAAgC,CAAC,IAAmB;IAClE,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,sBAAsB,CAAC,WAAW,CAAC,YAAY,CAAA;IACxD,CAAC;IACD,OAAO,sBAAsB,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,sBAAsB,CAAC,WAAW,CAAC,YAAY,CAAA;AAC3G,CAAC;AAED,MAAM,UAAU,sCAAsC;IACpD,OAAO,sBAAsB,CAAC,sBAAsB,CAAA;AACtD,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,OAAe;IACnD,OAAO,uBAAuB,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAA;AACvD,CAAC"}

package/dist/model-types.d.ts

@@ -0,0 +1,8 @@
+export interface ProviderModel {
+    cost?: {
+        input: number;
+        output: number;
+    };
+    [key: string]: unknown;
+}
+//# sourceMappingURL=model-types.d.ts.map

package/dist/model-types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-types.d.ts","sourceRoot":"","sources":["../src/model-types.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,aAAa;IAC5B,IAAI,CAAC,EAAE;QACL,KAAK,EAAE,MAAM,CAAA;QACb,MAAM,EAAE,MAAM,CAAA;KACf,CAAA;IACD,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;CACvB"}

package/dist/model-types.js

@@ -0,0 +1,3 @@
+// Harness-agnostic model definition shape shared by core and harness packages.
+export {};
+//# sourceMappingURL=model-types.js.map

package/dist/model-types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-types.js","sourceRoot":"","sources":["../src/model-types.ts"],"names":[],"mappings":"AAAA,+EAA+E"}

package/dist/project.d.ts

@@ -0,0 +1,34 @@
+import type { OAuthAuthDetails, ProjectContextResult } from "./auth-types.ts";
+interface AntigravityUserTier {
+    id?: string;
+    isDefault?: boolean;
+    userDefinedCloudaicompanionProject?: boolean;
+}
+interface LoadCodeAssistPayload {
+    cloudaicompanionProject?: string | {
+        id?: string;
+    };
+    currentTier?: {
+        id?: string;
+    };
+    allowedTiers?: AntigravityUserTier[];
+}
+/**
+ * Clears cached project context results and pending promises, globally or for a refresh key.
+ */
+export declare function invalidateProjectContextCache(refresh?: string): void;
+export declare function clearProvisionFailedKeys(): void;
+/**
+ * Loads managed project information for the given access token and optional project.
+ */
+export declare function loadManagedProject(accessToken: string, _projectId?: string): Promise<LoadCodeAssistPayload | null>;
+/**
+ * Onboards a managed project for the user, optionally retrying until completion.
+ */
+export declare function onboardManagedProject(accessToken: string, tierId: string, projectId?: string, attempts?: number, delayMs?: number): Promise<string | undefined>;
+/**
+ * Resolves an effective project ID for the current auth state, caching results per refresh token.
+ */
+export declare function ensureProjectContext(auth: OAuthAuthDetails): Promise<ProjectContextResult>;
+export {};
+//# sourceMappingURL=project.d.ts.map

package/dist/project.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"project.d.ts","sourceRoot":"","sources":["../src/project.ts"],"names":[],"mappings":"AAaA,OAAO,KAAK,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAe9E,UAAU,mBAAmB;IAC3B,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,kCAAkC,CAAC,EAAE,OAAO,CAAC;CAC9C;AAED,UAAU,qBAAqB;IAC7B,uBAAuB,CAAC,EAAE,MAAM,GAAG;QAAE,EAAE,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IACnD,WAAW,CAAC,EAAE;QACZ,EAAE,CAAC,EAAE,MAAM,CAAC;KACb,CAAC;IACF,YAAY,CAAC,EAAE,mBAAmB,EAAE,CAAC;CACtC;AAkED;;GAEG;AACH,wBAAgB,6BAA6B,CAAC,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAUpE;AAED,wBAAgB,wBAAwB,IAAI,IAAI,CAE/C;AAED;;GAEG;AACH,wBAAsB,kBAAkB,CACtC,WAAW,EAAE,MAAM,EACnB,UAAU,CAAC,EAAE,MAAM,GAClB,OAAO,CAAC,qBAAqB,GAAG,IAAI,CAAC,CA+BvC;AAGD;;GAEG;AACH,wBAAsB,qBAAqB,CACzC,WAAW,EAAE,MAAM,EACnB,MAAM,EAAE,MAAM,EACd,SAAS,CAAC,EAAE,MAAM,EAClB,QAAQ,SAAK,EACb,OAAO,SAAO,GACb,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CA6C7B;AAED;;GAEG;AACH,wBAAsB,oBAAoB,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,oBAAoB,CAAC,CA4GhG"}

package/dist/project.js

@@ -0,0 +1,236 @@
+import { ANTIGRAVITY_ENDPOINT_FALLBACKS, ANTIGRAVITY_LOAD_ENDPOINTS, ANTIGRAVITY_ENDPOINT_PROD, ANTIGRAVITY_DEFAULT_PROJECT_ID, } from "./constants.js";
+import { fetchWithAgyCliTransport } from "./agy-transport.js";
+import { formatRefreshParts, parseRefreshParts } from "./auth.js";
+import { buildAntigravityHarnessBootstrapHeaders, buildAntigravityLoadCodeAssistMetadata, } from "./fingerprint.js";
+import { createLogger } from "./logger.js";
+const log = createLogger("project");
+/** TTL for project context cache entries (30 minutes). */
+const PROJECT_CONTEXT_CACHE_TTL_MS = 30 * 60 * 1000;
+const projectContextResultCache = new Map();
+const projectContextPendingCache = new Map();
+const provisionFailedKeys = new Set();
+function buildBootstrapRequestBody(extra = {}) {
+    return {
+        ...extra,
+        metadata: buildAntigravityLoadCodeAssistMetadata(),
+    };
+}
+/**
+ * Selects the default tier ID from the allowed tiers list.
+ */
+function getDefaultTierId(allowedTiers) {
+    if (!allowedTiers || allowedTiers.length === 0) {
+        return undefined;
+    }
+    for (const tier of allowedTiers) {
+        if (tier?.isDefault) {
+            return tier.id;
+        }
+    }
+    return allowedTiers[0]?.id;
+}
+/**
+ * Promise-based delay utility.
+ */
+function wait(ms) {
+    return new Promise(function (resolve) {
+        setTimeout(resolve, ms);
+    });
+}
+/**
+ * Extracts the cloudaicompanion project id from loadCodeAssist responses.
+ */
+function extractManagedProjectId(payload) {
+    if (!payload) {
+        return undefined;
+    }
+    if (typeof payload.cloudaicompanionProject === "string") {
+        return payload.cloudaicompanionProject;
+    }
+    if (payload.cloudaicompanionProject && typeof payload.cloudaicompanionProject.id === "string") {
+        return payload.cloudaicompanionProject.id;
+    }
+    return undefined;
+}
+/**
+ * Generates a cache key for project context based on refresh token.
+ */
+function getCacheKey(auth) {
+    const refresh = auth.refresh?.trim();
+    return refresh ? refresh : undefined;
+}
+/**
+ * Clears cached project context results and pending promises, globally or for a refresh key.
+ */
+export function invalidateProjectContextCache(refresh) {
+    if (!refresh) {
+        projectContextPendingCache.clear();
+        projectContextResultCache.clear();
+        provisionFailedKeys.clear();
+        return;
+    }
+    projectContextPendingCache.delete(refresh);
+    projectContextResultCache.delete(refresh);
+    provisionFailedKeys.delete(refresh);
+}
+export function clearProvisionFailedKeys() {
+    provisionFailedKeys.clear();
+}
+/**
+ * Loads managed project information for the given access token and optional project.
+ */
+export async function loadManagedProject(accessToken, _projectId) {
+    const requestBody = buildBootstrapRequestBody();
+    const loadHeaders = buildAntigravityHarnessBootstrapHeaders(accessToken);
+    const loadEndpoints = Array.from(new Set([...ANTIGRAVITY_LOAD_ENDPOINTS, ...ANTIGRAVITY_ENDPOINT_FALLBACKS]));
+    for (const baseEndpoint of loadEndpoints) {
+        try {
+            const response = await fetchWithAgyCliTransport(`${baseEndpoint}/v1internal:loadCodeAssist`, {
+                method: "POST",
+                headers: loadHeaders,
+                body: JSON.stringify(requestBody),
+            });
+            if (!response.ok) {
+                continue;
+            }
+            return (await response.json());
+        }
+        catch (error) {
+            log.debug("Failed to load managed project", { endpoint: baseEndpoint, error: String(error) });
+            continue;
+        }
+    }
+    return null;
+}
+/**
+ * Onboards a managed project for the user, optionally retrying until completion.
+ */
+export async function onboardManagedProject(accessToken, tierId, projectId, attempts = 10, delayMs = 5000) {
+    const requestBody = { tierId };
+    const onboardEndpoints = Array.from(new Set([ANTIGRAVITY_ENDPOINT_PROD, ...ANTIGRAVITY_LOAD_ENDPOINTS, ...ANTIGRAVITY_ENDPOINT_FALLBACKS]));
+    for (const baseEndpoint of onboardEndpoints) {
+        for (let attempt = 0; attempt < attempts; attempt += 1) {
+            try {
+                const response = await fetchWithAgyCliTransport(`${baseEndpoint}/v1internal:onboardUser`, {
+                    method: "POST",
+                    headers: buildAntigravityHarnessBootstrapHeaders(accessToken),
+                    body: JSON.stringify(requestBody),
+                });
+                if (!response.ok) {
+                    log.debug("Onboard request failed", {
+                        endpoint: baseEndpoint,
+                        status: response.status,
+                        statusText: response.statusText,
+                    });
+                    break;
+                }
+                const payload = (await response.json());
+                const managedProjectId = payload.response?.cloudaicompanionProject?.id;
+                if (payload.done && managedProjectId) {
+                    return managedProjectId;
+                }
+                if (payload.done && projectId) {
+                    return projectId;
+                }
+            }
+            catch (error) {
+                log.debug("Failed to onboard managed project", { endpoint: baseEndpoint, error: String(error) });
+                break;
+            }
+            await wait(delayMs);
+        }
+    }
+    return undefined;
+}
+/**
+ * Resolves an effective project ID for the current auth state, caching results per refresh token.
+ */
+export async function ensureProjectContext(auth) {
+    const accessToken = auth.access;
+    if (!accessToken) {
+        return { auth, effectiveProjectId: "" };
+    }
+    const cacheKey = getCacheKey(auth);
+    if (cacheKey) {
+        const cached = projectContextResultCache.get(cacheKey);
+        if (cached && (Date.now() - cached.cachedAt) < PROJECT_CONTEXT_CACHE_TTL_MS) {
+            return cached.result;
+        }
+        if (cached) {
+            // Expired — evict stale entry
+            projectContextResultCache.delete(cacheKey);
+        }
+        const pending = projectContextPendingCache.get(cacheKey);
+        if (pending) {
+            return pending;
+        }
+    }
+    const resolveContext = async () => {
+        const parts = parseRefreshParts(auth.refresh);
+        if (parts.managedProjectId) {
+            return { auth, effectiveProjectId: parts.managedProjectId };
+        }
+        const fallbackProjectId = ANTIGRAVITY_DEFAULT_PROJECT_ID;
+        if (cacheKey && provisionFailedKeys.has(cacheKey)) {
+            const effectiveProjectId = parts.projectId || fallbackProjectId;
+            return { auth, effectiveProjectId };
+        }
+        const persistManagedProject = async (managedProjectId) => {
+            const updatedAuth = {
+                ...auth,
+                refresh: formatRefreshParts({
+                    refreshToken: parts.refreshToken,
+                    projectId: parts.projectId,
+                    managedProjectId,
+                }),
+            };
+            return { auth: updatedAuth, effectiveProjectId: managedProjectId };
+        };
+        // Try to resolve a managed project from Antigravity if possible.
+        const loadPayload = await loadManagedProject(accessToken, parts.projectId ?? fallbackProjectId);
+        const resolvedManagedProjectId = extractManagedProjectId(loadPayload);
+        if (resolvedManagedProjectId) {
+            return persistManagedProject(resolvedManagedProjectId);
+        }
+        // No managed project found - try to auto-provision one via onboarding.
+        // This handles accounts that were added before managed project provisioning was required.
+        const tierId = getDefaultTierId(loadPayload?.allowedTiers) ?? "free-tier";
+        log.debug("Auto-provisioning managed project", { tierId, projectId: parts.projectId });
+        const provisionedProjectId = await onboardManagedProject(accessToken, tierId, parts.projectId);
+        if (provisionedProjectId) {
+            log.debug("Successfully provisioned managed project", { provisionedProjectId });
+            return persistManagedProject(provisionedProjectId);
+        }
+        log.warn("Failed to provision managed project - account may not work correctly", {
+            hasProjectId: !!parts.projectId,
+        });
+        if (cacheKey) {
+            provisionFailedKeys.add(cacheKey);
+        }
+        if (parts.projectId) {
+            return { auth, effectiveProjectId: parts.projectId };
+        }
+        // No project id present in auth; fall back to the hardcoded id for requests.
+        return { auth, effectiveProjectId: fallbackProjectId };
+    };
+    if (!cacheKey) {
+        return resolveContext();
+    }
+    const promise = resolveContext()
+        .then((result) => {
+        const nextKey = getCacheKey(result.auth) ?? cacheKey;
+        projectContextPendingCache.delete(cacheKey);
+        projectContextResultCache.set(nextKey, { result, cachedAt: Date.now() });
+        if (nextKey !== cacheKey) {
+            projectContextResultCache.delete(cacheKey);
+        }
+        return result;
+    })
+        .catch((error) => {
+        projectContextPendingCache.delete(cacheKey);
+        throw error;
+    });
+    projectContextPendingCache.set(cacheKey, promise);
+    return promise;
+}
+//# sourceMappingURL=project.js.map

package/dist/project.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"project.js","sourceRoot":"","sources":["../src/project.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,8BAA8B,EAC9B,0BAA0B,EAC1B,yBAAyB,EACzB,8BAA8B,GAC/B,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,wBAAwB,EAAE,MAAM,oBAAoB,CAAC;AAC9D,OAAO,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAClE,OAAO,EACL,uCAAuC,EACvC,sCAAsC,GACvC,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAG3C,MAAM,GAAG,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC;AAEpC,0DAA0D;AAC1D,MAAM,4BAA4B,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;AAOpD,MAAM,yBAAyB,GAAG,IAAI,GAAG,EAAgC,CAAC;AAC1E,MAAM,0BAA0B,GAAG,IAAI,GAAG,EAAyC,CAAC;AACpF,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAU,CAAC;AAwB9C,SAAS,yBAAyB,CAAC,QAAiC,EAAE;IACpE,OAAO;QACL,GAAG,KAAK;QACR,QAAQ,EAAE,sCAAsC,EAAE;KACnD,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,gBAAgB,CAAC,YAAoC;IAC5D,IAAI,CAAC,YAAY,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/C,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;QAChC,IAAI,IAAI,EAAE,SAAS,EAAE,CAAC;YACpB,OAAO,IAAI,CAAC,EAAE,CAAC;QACjB,CAAC;IACH,CAAC;IACD,OAAO,YAAY,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;AAC7B,CAAC;AAED;;GAEG;AACH,SAAS,IAAI,CAAC,EAAU;IACtB,OAAO,IAAI,OAAO,CAAC,UAAU,OAAO;QAClC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IAC1B,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,SAAS,uBAAuB,CAAC,OAAqC;IACpE,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,IAAI,OAAO,OAAO,CAAC,uBAAuB,KAAK,QAAQ,EAAE,CAAC;QACxD,OAAO,OAAO,CAAC,uBAAuB,CAAC;IACzC,CAAC;IACD,IAAI,OAAO,CAAC,uBAAuB,IAAI,OAAO,OAAO,CAAC,uBAAuB,CAAC,EAAE,KAAK,QAAQ,EAAE,CAAC;QAC9F,OAAO,OAAO,CAAC,uBAAuB,CAAC,EAAE,CAAC;IAC5C,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,IAAsB;IACzC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC;IACrC,OAAO,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;AACvC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,6BAA6B,CAAC,OAAgB;IAC5D,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,0BAA0B,CAAC,KAAK,EAAE,CAAC;QACnC,yBAAyB,CAAC,KAAK,EAAE,CAAC;QAClC,mBAAmB,CAAC,KAAK,EAAE,CAAC;QAC5B,OAAO;IACT,CAAC;IACD,0BAA0B,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC3C,yBAAyB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC1C,mBAAmB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;AACtC,CAAC;AAED,MAAM,UAAU,wBAAwB;IACtC,mBAAmB,CAAC,KAAK,EAAE,CAAC;AAC9B,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,WAAmB,EACnB,UAAmB;IAEnB,MAAM,WAAW,GAAG,yBAAyB,EAAE,CAAC;IAChD,MAAM,WAAW,GAAG,uCAAuC,CAAC,WAAW,CAAC,CAAC;IAEzE,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAC9B,IAAI,GAAG,CAAS,CAAC,GAAG,0BAA0B,EAAE,GAAG,8BAA8B,CAAC,CAAC,CACpF,CAAC;IAEF,KAAK,MAAM,YAAY,IAAI,aAAa,EAAE,CAAC;QACzC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,wBAAwB,CAC7C,GAAG,YAAY,4BAA4B,EAC3C;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,WAAW;gBACpB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;aAClC,CACF,CAAC;YAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,SAAS;YACX,CAAC;YAED,OAAO,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA0B,CAAC;QAC1D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,KAAK,CAAC,gCAAgC,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAC9F,SAAS;QACX,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAGD;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,WAAmB,EACnB,MAAc,EACd,SAAkB,EAClB,QAAQ,GAAG,EAAE,EACb,OAAO,GAAG,IAAI;IAEd,MAAM,WAAW,GAA4B,EAAE,MAAM,EAAE,CAAC;IACxD,MAAM,gBAAgB,GAAG,KAAK,CAAC,IAAI,CACjC,IAAI,GAAG,CAAS,CAAC,yBAAyB,EAAE,GAAG,0BAA0B,EAAE,GAAG,8BAA8B,CAAC,CAAC,CAC/G,CAAC;IAEF,KAAK,MAAM,YAAY,IAAI,gBAAgB,EAAE,CAAC;QAC5C,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,QAAQ,EAAE,OAAO,IAAI,CAAC,EAAE,CAAC;YACvD,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,wBAAwB,CAC7C,GAAG,YAAY,yBAAyB,EACxC;oBACE,MAAM,EAAE,MAAM;oBACd,OAAO,EAAE,uCAAuC,CAAC,WAAW,CAAC;oBAC7D,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;iBAClC,CACF,CAAC;gBAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACjB,GAAG,CAAC,KAAK,CAAC,wBAAwB,EAAE;wBAClC,QAAQ,EAAE,YAAY;wBACtB,MAAM,EAAE,QAAQ,CAAC,MAAM;wBACvB,UAAU,EAAE,QAAQ,CAAC,UAAU;qBAChC,CAAC,CAAC;oBACH,MAAM;gBACR,CAAC;gBAED,MAAM,OAAO,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAuB,CAAC;gBAC9D,MAAM,gBAAgB,GAAG,OAAO,CAAC,QAAQ,EAAE,uBAAuB,EAAE,EAAE,CAAC;gBACvE,IAAI,OAAO,CAAC,IAAI,IAAI,gBAAgB,EAAE,CAAC;oBACrC,OAAO,gBAAgB,CAAC;gBAC1B,CAAC;gBACD,IAAI,OAAO,CAAC,IAAI,IAAI,SAAS,EAAE,CAAC;oBAC9B,OAAO,SAAS,CAAC;gBACnB,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,KAAK,CAAC,mCAAmC,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBACjG,MAAM;YACR,CAAC;YAED,MAAM,IAAI,CAAC,OAAO,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,IAAsB;IAC/D,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC;IAChC,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,EAAE,EAAE,CAAC;IAC1C,CAAC;IAED,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IACnC,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,MAAM,GAAG,yBAAyB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACvD,IAAI,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC,GAAG,4BAA4B,EAAE,CAAC;YAC5E,OAAO,MAAM,CAAC,MAAM,CAAC;QACvB,CAAC;QACD,IAAI,MAAM,EAAE,CAAC;YACX,8BAA8B;YAC9B,yBAAyB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC7C,CAAC;QAAI,MAAM,OAAO,GAAG,0BAA0B,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC9D,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,OAAO,CAAC;QACjB,CAAC;IACH,CAAC;IAED,MAAM,cAAc,GAAG,KAAK,IAAmC,EAAE;QAC/D,MAAM,KAAK,GAAG,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9C,IAAI,KAAK,CAAC,gBAAgB,EAAE,CAAC;YAC3B,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,KAAK,CAAC,gBAAgB,EAAE,CAAC;QAC9D,CAAC;QAED,MAAM,iBAAiB,GAAG,8BAA8B,CAAC;QAEzD,IAAI,QAAQ,IAAI,mBAAmB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YAClD,MAAM,kBAAkB,GAAG,KAAK,CAAC,SAAS,IAAI,iBAAiB,CAAC;YAChE,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC;QACtC,CAAC;QAED,MAAM,qBAAqB,GAAG,KAAK,EAAE,gBAAwB,EAAiC,EAAE;YAC9F,MAAM,WAAW,GAAqB;gBACpC,GAAG,IAAI;gBACP,OAAO,EAAE,kBAAkB,CAAC;oBAC1B,YAAY,EAAE,KAAK,CAAC,YAAY;oBAChC,SAAS,EAAE,KAAK,CAAC,SAAS;oBAC1B,gBAAgB;iBACjB,CAAC;aACH,CAAC;YAEF,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,CAAC;QACrE,CAAC,CAAC;QAEF,iEAAiE;QACjE,MAAM,WAAW,GAAG,MAAM,kBAAkB,CAAC,WAAW,EAAE,KAAK,CAAC,SAAS,IAAI,iBAAiB,CAAC,CAAC;QAChG,MAAM,wBAAwB,GAAG,uBAAuB,CAAC,WAAW,CAAC,CAAC;QAEtE,IAAI,wBAAwB,EAAE,CAAC;YAC7B,OAAO,qBAAqB,CAAC,wBAAwB,CAAC,CAAC;QACzD,CAAC;QAED,uEAAuE;QACvE,0FAA0F;QAC1F,MAAM,MAAM,GAAG,gBAAgB,CAAC,WAAW,EAAE,YAAY,CAAC,IAAI,WAAW,CAAC;QAC1E,GAAG,CAAC,KAAK,CAAC,mCAAmC,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC;QAEvF,MAAM,oBAAoB,GAAG,MAAM,qBAAqB,CACtD,WAAW,EACX,MAAM,EACN,KAAK,CAAC,SAAS,CAChB,CAAC;QAEF,IAAI,oBAAoB,EAAE,CAAC;YACzB,GAAG,CAAC,KAAK,CAAC,0CAA0C,EAAE,EAAE,oBAAoB,EAAE,CAAC,CAAC;YAChF,OAAO,qBAAqB,CAAC,oBAAoB,CAAC,CAAC;QACrD,CAAC;QAED,GAAG,CAAC,IAAI,CAAC,sEAAsE,EAAE;YAC/E,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,SAAS;SAChC,CAAC,CAAC;QAEH,IAAI,QAAQ,EAAE,CAAC;YACb,mBAAmB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACpC,CAAC;QAED,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YACpB,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC;QACvD,CAAC;QAED,6EAA6E;QAC7E,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,CAAC;IACzD,CAAC,CAAC;IAEF,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,OAAO,cAAc,EAAE,CAAC;IAC1B,CAAC;IAED,MAAM,OAAO,GAAG,cAAc,EAAE;SAC7B,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;QACf,MAAM,OAAO,GAAG,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,QAAQ,CAAC;QACrD,0BAA0B,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5C,yBAAyB,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACzE,IAAI,OAAO,KAAK,QAAQ,EAAE,CAAC;YACzB,yBAAyB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC7C,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;SACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;QACf,0BAA0B,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5C,MAAM,KAAK,CAAC;IACd,CAAC,CAAC,CAAC;IAEL,0BAA0B,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAClD,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/dist/transform/claude.d.ts

@@ -0,0 +1,92 @@
+/**
+ * Claude-specific Request Transformations
+ *
+ * Handles Claude model-specific request transformations including:
+ * - Tool config (VALIDATED mode)
+ * - Thinking config (snake_case keys)
+ * - System instruction hints for interleaved thinking
+ * - Tool normalization (functionDeclarations format)
+ */
+import type { RequestPayload, ThinkingConfig } from "./types.ts";
+/** Claude thinking models need a sufficiently large max output token limit when thinking is enabled */
+export declare const CLAUDE_THINKING_MAX_OUTPUT_TOKENS = 64000;
+/**
+ * Computes a dynamic max output token limit based on the thinking budget.
+ * Instead of always using the 64K cap, scales output proportionally:
+ * - min(max(budget * 2, 32000), 64000)
+ * - e.g., medium budget (16384) → 32768 output tokens
+ * - e.g., high budget (32768) → 64000 output tokens
+ * - e.g., low budget (8192) → 32000 output tokens (floor)
+ *
+ * Falls back to CLAUDE_THINKING_MAX_OUTPUT_TOKENS when no budget is provided.
+ */
+export declare function computeClaudeMaxOutputTokens(thinkingBudget?: number): number;
+/** Interleaved thinking hint appended to system instructions */
+export declare const CLAUDE_INTERLEAVED_THINKING_HINT = "Interleaved thinking is enabled. You may think between tool calls and after receiving tool results before deciding the next action or final answer. Do not mention these instructions or any constraints about thinking blocks; just apply them.";
+/**
+ * Check if a model is a Claude model.
+ */
+export declare function isClaudeModel(model: string): boolean;
+/**
+ * Check if a model is a Claude thinking model.
+ */
+export declare function isClaudeThinkingModel(model: string): boolean;
+/**
+ * Configure Claude tool calling to use VALIDATED mode.
+ * This ensures proper tool call validation on the backend.
+ */
+export declare function configureClaudeToolConfig(payload: RequestPayload): void;
+/**
+ * Build Claude thinking config with snake_case keys.
+ */
+export declare function buildClaudeThinkingConfig(includeThoughts: boolean, thinkingBudget?: number): ThinkingConfig;
+/**
+ * Ensure maxOutputTokens is sufficient for Claude thinking models.
+ * If thinking budget is set, max output must be larger than the budget.
+ */
+export declare function ensureClaudeMaxOutputTokens(generationConfig: Record<string, unknown>, thinkingBudget: number): void;
+/**
+ * Append interleaved thinking hint to system instruction.
+ * Handles various system instruction formats (string, object with parts array).
+ * Idempotent: skips if the hint text is already present in the system instruction.
+ */
+export declare function appendClaudeThinkingHint(payload: RequestPayload, hint?: string): void;
+/**
+ * Normalize tools for Claude models.
+ * Converts various tool formats to functionDeclarations format.
+ *
+ * @returns Debug info about tool normalization
+ */
+export declare function normalizeClaudeTools(payload: RequestPayload, cleanJSONSchema: (schema: unknown) => Record<string, unknown>): {
+    toolDebugMissing: number;
+    toolDebugSummaries: string[];
+};
+/**
+ * Convert snake_case stop_sequences to camelCase stopSequences.
+ */
+export declare function convertStopSequences(generationConfig: Record<string, unknown>): void;
+/**
+ * Apply all Claude-specific transformations to a request payload.
+ */
+export interface ClaudeTransformOptions {
+    /** The effective model name (resolved) */
+    model: string;
+    /** Tier-based thinking budget (from model suffix) */
+    tierThinkingBudget?: number;
+    /** Normalized thinking config from user settings */
+    normalizedThinking?: {
+        includeThoughts?: boolean;
+        thinkingBudget?: number;
+    };
+    /** Function to clean JSON schema for Antigravity */
+    cleanJSONSchema: (schema: unknown) => Record<string, unknown>;
+}
+export interface ClaudeTransformResult {
+    toolDebugMissing: number;
+    toolDebugSummaries: string[];
+}
+/**
+ * Apply all Claude-specific transformations.
+ */
+export declare function applyClaudeTransforms(payload: RequestPayload, options: ClaudeTransformOptions): ClaudeTransformResult;
+//# sourceMappingURL=claude.d.ts.map

package/dist/transform/claude.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"claude.d.ts","sourceRoot":"","sources":["../../src/transform/claude.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAMjE,uGAAuG;AACvG,eAAO,MAAM,iCAAiC,QAAS,CAAC;AAExD;;;;;;;;;GASG;AACH,wBAAgB,4BAA4B,CAAC,cAAc,CAAC,EAAE,MAAM,GAAG,MAAM,CAK5E;AACD,gEAAgE;AAChE,eAAO,MAAM,gCAAgC,qPACuM,CAAC;AAErP;;GAEG;AACH,wBAAgB,aAAa,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAEpD;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAG5D;AAED;;;GAGG;AACH,wBAAgB,yBAAyB,CAAC,OAAO,EAAE,cAAc,GAAG,IAAI,CAcvE;AAED;;GAEG;AACH,wBAAgB,yBAAyB,CACvC,eAAe,EAAE,OAAO,EACxB,cAAc,CAAC,EAAE,MAAM,GACtB,cAAc,CAOhB;AAED;;;GAGG;AACH,wBAAgB,2BAA2B,CACzC,gBAAgB,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACzC,cAAc,EAAE,MAAM,GACrB,IAAI,CASN;AACD;;;;GAIG;AACH,wBAAgB,wBAAwB,CACtC,OAAO,EAAE,cAAc,EACvB,IAAI,GAAE,MAAyC,GAC9C,IAAI,CAqCN;AAED;;;;;GAKG;AACH,wBAAgB,oBAAoB,CAClC,OAAO,EAAE,cAAc,EACvB,eAAe,EAAE,CAAC,MAAM,EAAE,OAAO,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC5D;IAAE,gBAAgB,EAAE,MAAM,CAAC;IAAC,kBAAkB,EAAE,MAAM,EAAE,CAAA;CAAE,CA2I5D;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAClC,gBAAgB,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GACxC,IAAI,CAKN;AAED;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACrC,0CAA0C;IAC1C,KAAK,EAAE,MAAM,CAAC;IACd,qDAAqD;IACrD,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,oDAAoD;IACpD,kBAAkB,CAAC,EAAE;QAAE,eAAe,CAAC,EAAE,OAAO,CAAC;QAAC,cAAc,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IAC5E,oDAAoD;IACpD,eAAe,EAAE,CAAC,MAAM,EAAE,OAAO,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC/D;AAED,MAAM,WAAW,qBAAqB;IACpC,gBAAgB,EAAE,MAAM,CAAC;IACzB,kBAAkB,EAAE,MAAM,EAAE,CAAC;CAC9B;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CACnC,OAAO,EAAE,cAAc,EACvB,OAAO,EAAE,sBAAsB,GAC9B,qBAAqB,CAuCvB"}