@oh-my-pi/pi-ai 12.17.2 → 12.18.1

package/package.json

@@ -1,7 +1,7 @@
 {
 	"type": "module",
 	"name": "@oh-my-pi/pi-ai",
-	"version": "12.17.2",
+	"version": "12.18.1",
 	"description": "Unified LLM API with automatic model discovery and provider configuration",
 	"homepage": "https://github.com/can1357/oh-my-pi",
 	"author": "Can Bölük",
@@ -44,7 +44,7 @@
 		"@connectrpc/connect-node": "^2.1.1",
 		"@google/genai": "^1.41.0",
 		"@mistralai/mistralai": "^1.14.0",
-		"@oh-my-pi/pi-utils": "12.17.2",
+		"@oh-my-pi/pi-utils": "12.18.1",
 		"@sinclair/typebox": "^0.34.48",
 		"@smithy/node-http-handler": "^4.4.10",
 		"ajv": "^8.18.0",

package/src/providers/google-gemini-cli-usage.ts

@@ -8,20 +8,11 @@ import type {
 	UsageWindow,
 } from "../usage";
 import { refreshGoogleCloudToken } from "../utils/oauth/google-gemini-cli";
+import { getGeminiCliHeaders } from "./google-gemini-cli";
 
 const DEFAULT_ENDPOINT = "https://cloudcode-pa.googleapis.com";
 const CACHE_TTL_MS = 60_000;
 
-const GEMINI_CLI_HEADERS = {
-	"User-Agent": "google-cloud-sdk vscode_cloudshelleditor/0.1",
-	"X-Goog-Api-Client": "gl-node/22.17.0",
-	"Client-Metadata": JSON.stringify({
-		ideType: "IDE_UNSPECIFIED",
-		platform: "PLATFORM_UNSPECIFIED",
-		pluginType: "GEMINI",
-	}),
-};
-
 const GEMINI_TIER_MAP: Array<{ tier: string; models: string[] }> = [
 	{
 		tier: "3-Flash",
@@ -140,7 +131,7 @@ async function loadCodeAssist(
 		headers: {
 			Authorization: `Bearer ${accessToken}`,
 			"Content-Type": "application/json",
-			...GEMINI_CLI_HEADERS,
+			...getGeminiCliHeaders(),
 		},
 		body: JSON.stringify({
 			...(projectId ? { cloudaicompanionProject: projectId } : {}),
@@ -177,7 +168,7 @@ async function fetchQuota(
 		headers: {
 			Authorization: `Bearer ${accessToken}`,
 			"Content-Type": "application/json",
-			...GEMINI_CLI_HEADERS,
+			...getGeminiCliHeaders(),
 		},
 		body: JSON.stringify(projectId ? { project: projectId } : {}),
 		signal: params.signal,

package/src/providers/google-gemini-cli.ts

@@ -58,35 +58,55 @@ export interface GoogleGeminiCliOptions extends StreamOptions {
 const DEFAULT_ENDPOINT = "https://cloudcode-pa.googleapis.com";
 const ANTIGRAVITY_DAILY_ENDPOINT = "https://daily-cloudcode-pa.sandbox.googleapis.com";
 const ANTIGRAVITY_ENDPOINT_FALLBACKS = [ANTIGRAVITY_DAILY_ENDPOINT, DEFAULT_ENDPOINT] as const;
-// Headers for Gemini CLI (prod endpoint)
-const GEMINI_CLI_HEADERS = {
-	"User-Agent": "google-cloud-sdk vscode_cloudshelleditor/0.1",
+
+const GEMINI_CLI_USER_AGENT = process.env.PI_AI_GEMINI_CLI_USER_AGENT || "google-api-nodejs-client/9.15.1";
+
+const ANTIGRAVITY_USER_AGENT = (() => {
+	const DEFAULT_ANTIGRAVITY_VERSION = "1.104.0";
+	const version = process.env.PI_AI_ANTIGRAVITY_VERSION || DEFAULT_ANTIGRAVITY_VERSION;
+	return `antigravity/${version} darwin/arm64`;
+})();
+
+const GEMINI_CLI_HEADERS = Object.freeze({
+	"User-Agent": GEMINI_CLI_USER_AGENT,
 	"X-Goog-Api-Client": "gl-node/22.17.0",
+	"Client-Metadata": "ideType=IDE_UNSPECIFIED,platform=PLATFORM_UNSPECIFIED,pluginType=GEMINI",
+});
+
+// Antigravity auth headers (project discovery/onboarding) — matches CLIProxyAPI antigravity auth module.
+// Same User-Agent as Gemini CLI, but different X-Goog-Api-Client.
+const ANTIGRAVITY_AUTH_HEADERS = Object.freeze({
+	"User-Agent": GEMINI_CLI_USER_AGENT,
+	"X-Goog-Api-Client": "google-cloud-sdk vscode_cloudshelleditor/0.1",
 	"Client-Metadata": JSON.stringify({
 		ideType: "IDE_UNSPECIFIED",
 		platform: "PLATFORM_UNSPECIFIED",
 		pluginType: "GEMINI",
 	}),
-};
+});
 
-// Headers for Antigravity (sandbox endpoint) - requires specific User-Agent
-const DEFAULT_ANTIGRAVITY_VERSION = "1.18.3";
+// Antigravity executor headers (streaming/generation) — only User-Agent per CLIProxyAPI executor.
+const ANTIGRAVITY_STREAMING_HEADERS = Object.freeze({
+	"User-Agent": ANTIGRAVITY_USER_AGENT,
+});
 
-export function getAntigravityUserAgent() {
-	const version = process.env.PI_AI_ANTIGRAVITY_VERSION || DEFAULT_ANTIGRAVITY_VERSION;
-	return `antigravity/${version} darwin/arm64`;
+// Headers for Gemini CLI (prod endpoint)
+export function getGeminiCliHeaders() {
+	return GEMINI_CLI_HEADERS;
+}
+export function getGeminiCliUserAgent() {
+	return GEMINI_CLI_USER_AGENT;
 }
 
-function getAntigravityHeaders() {
-	return {
-		"User-Agent": getAntigravityUserAgent(),
-		"X-Goog-Api-Client": "google-cloud-sdk vscode_cloudshelleditor/0.1",
-		"Client-Metadata": JSON.stringify({
-			ideType: "IDE_UNSPECIFIED",
-			platform: "PLATFORM_UNSPECIFIED",
-			pluginType: "GEMINI",
-		}),
-	};
+// Headers for Antigravity (sandbox endpoint)
+export function getAntigravityAuthHeaders() {
+	return ANTIGRAVITY_AUTH_HEADERS;
+}
+export function getAntigravityHeaders() {
+	return ANTIGRAVITY_STREAMING_HEADERS;
+}
+export function getAntigravityUserAgent() {
+	return ANTIGRAVITY_USER_AGENT;
 }
 
 // Antigravity system instruction (compact version from CLIProxyAPI).

package/src/utils/oauth/google-antigravity.ts

@@ -2,6 +2,7 @@
  * Antigravity OAuth flow (Gemini 3, Claude, GPT-OSS via Google Cloud)
  * Uses different OAuth credentials than google-gemini-cli for access to additional models.
  */
+import { getAntigravityAuthHeaders } from "../../providers/google-gemini-cli";
 import { OAuthCallbackFlow } from "./callback-server";
 import { generatePKCE } from "./pkce";
 import type { OAuthController, OAuthCredentials } from "./types";
@@ -36,13 +37,7 @@ async function discoverProject(accessToken: string, onProgress?: (message: strin
 	const headers = {
 		Authorization: `Bearer ${accessToken}`,
 		"Content-Type": "application/json",
-		"User-Agent": "google-api-nodejs-client/9.15.1",
-		"X-Goog-Api-Client": "google-cloud-sdk vscode_cloudshelleditor/0.1",
-		"Client-Metadata": JSON.stringify({
-			ideType: "IDE_UNSPECIFIED",
-			platform: "PLATFORM_UNSPECIFIED",
-			pluginType: "GEMINI",
-		}),
+		...getAntigravityAuthHeaders(),
 	};
 
 	const endpoints = ["https://cloudcode-pa.googleapis.com", "https://daily-cloudcode-pa.sandbox.googleapis.com"];

package/src/utils/oauth/google-gemini-cli.ts

@@ -4,6 +4,7 @@
  */
 
 import { $env } from "@oh-my-pi/pi-utils";
+import { getGeminiCliHeaders } from "../../providers/google-gemini-cli";
 import { OAuthCallbackFlow } from "./callback-server";
 import { generatePKCE } from "./pkce";
 import type { OAuthController, OAuthCredentials } from "./types";
@@ -98,8 +99,7 @@ async function discoverProject(accessToken: string, onProgress?: (message: strin
 	const headers = {
 		Authorization: `Bearer ${accessToken}`,
 		"Content-Type": "application/json",
-		"User-Agent": "google-api-nodejs-client/9.15.1",
-		"X-Goog-Api-Client": "gl-node/22.17.0",
+		...getGeminiCliHeaders(),
 	};
 
 	onProgress?.("Checking for existing Cloud Code Assist project...");