opencode-google-auth 0.0.2 → 0.0.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-google-auth",
-  "version": "0.0.2",
+  "version": "0.0.4",
   "description": "_description_",
   "keywords": [
     "opencode-google-auth"
@@ -26,9 +26,7 @@
     "src"
   ],
   "scripts": {
-    "prebuild": "bun ./scripts/fetch-models.ts",
-    "predeploy": "bun run build",
-    "deploy": "bun ./scripts/deploy.ts",
+    "deploy": "bun build ./src/main.ts --target bun --outfile .opencode/plugin/opencode-google-auth.js",
     "format": "prettier --write .",
     "lint": "oxlint",
     "release": "bumpp && npm publish",
@@ -38,19 +36,21 @@
     "@effect/experimental": "^0.58.0",
     "@effect/platform": "^0.94.1",
     "@effect/platform-bun": "^0.87.0",
-    "@opencode-ai/plugin": "^1.1.15",
+    "@opencode-ai/plugin": "^1.1.18",
     "arkregex": "^0.0.5",
     "effect": "^3.19.14",
-    "google-auth-library": "^10.5.0"
+    "google-auth-library": "^10.5.0",
+    "open": "^11.0.0",
+    "xdg-basedir": "^5.1.0"
   },
   "devDependencies": {
     "@ai-sdk/google": "^3.0.7",
-    "@effect/language-service": "^0.64.1",
+    "@effect/language-service": "^0.65.0",
     "@types/bun": "^1.3.5",
     "@types/yargs": "^17.0.35",
-    "@typescript/native-preview": "^7.0.0-dev.20260112.1",
-    "ai": "^6.0.30",
-    "bumpp": "^10.3.2",
+    "@typescript/native-preview": "^7.0.0-dev.20260113.1",
+    "ai": "^6.0.33",
+    "bumpp": "^10.4.0",
     "oxlint": "^1.39.0",
     "oxlint-tsgolint": "^0.11.0",
     "prettier": "^3.7.4",

package/src/antigravity-spoof.txt

@@ -0,0 +1 @@
+You are Antigravity, a powerful agentic AI coding assistant designed by the Google Deepmind team working on Advanced Agentic Coding.You are pair programming with a USER to solve their coding task. The task may require creating a new codebase, modifying or debugging an existing codebase, or simply answering a question.**Absolute paths only****Proactiveness**

package/src/lib/logger.ts

@@ -2,8 +2,9 @@ import { PlatformLogger } from "@effect/platform"
 import { Effect, Inspectable, Logger, LogLevel, pipe } from "effect"
 import path from "node:path"
 import type { OpenCodeLogLevel } from "../types"
-import { ProviderConfig } from "../services/config"
+import { PLUGIN_NAME, ProviderConfig } from "../services/config"
 import { OpenCodeContext } from "../services/opencode"
+import { xdgData } from "xdg-basedir"
 
 const makeOpenCodeLogger = Effect.gen(function* () {
   const openCode = yield* OpenCodeContext
@@ -32,11 +33,13 @@ const makeOpenCodeLogger = Effect.gen(function* () {
   })
 })
 
+const LOG_DIR = xdgData ? path.join(xdgData, "opencode") : import.meta.dir
+
 export const combinedLogger = Effect.gen(function* () {
   const openCodeLogger = yield* makeOpenCodeLogger
   const fileLogger = yield* pipe(
     Logger.jsonLogger,
-    PlatformLogger.toFile(path.join(import.meta.dir, "plugin.log")),
+    PlatformLogger.toFile(path.join(LOG_DIR, `${PLUGIN_NAME}.log`)),
   )
 
   return Logger.zip(openCodeLogger, fileLogger)

package/src/main.ts

@@ -17,6 +17,7 @@ import { transformRequest } from "./transform/request"
 import { transformNonStreamingResponse } from "./transform/response"
 import { transformStreamingResponse } from "./transform/stream"
 import type { Credentials, ModelsDev } from "./types"
+import antigravitySpoof from "./antigravity-spoof.txt"
 
 const fetchModelsDev = Effect.gen(function* () {
   const client = yield* HttpClient.HttpClient
@@ -251,6 +252,10 @@ export const antigravity: Plugin = async (context) => {
         },
       ],
     },
+    "experimental.chat.system.transform": async (_input, output) => {
+      // THIS IS REQUIRED OTHERWISE YOU'LL GET 429 OR 403 FOR SOME GODDAMN REASON
+      output.system.unshift(antigravitySpoof)
+    },
     "chat.params": async (input, output) => {
       await runtime.runPromise(
         Effect.log("chat.params event before:", input.model, output.options),

package/src/services/config.ts

@@ -40,8 +40,8 @@ export class ProviderConfig extends Context.Tag("ProviderConfig")<
   ProviderConfigShape
 >() {}
 
+export const PLUGIN_NAME = "opencode-google-auth"
 export const CODE_ASSIST_VERSION = "v1internal"
-
 export const CLIENT_METADATA = {
   ideType: "IDE_UNSPECIFIED",
   platform: "PLATFORM_UNSPECIFIED",
@@ -49,8 +49,8 @@ export const CLIENT_METADATA = {
 } as const
 
 export const GEMINI_CLI_MODELS = [
-  "gemini-2.5-pro",
-  "gemini-2.5-flash",
+  // "gemini-2.5-pro",
+  // "gemini-2.5-flash",
   "gemini-2.5-flash-lite",
   "gemini-3-pro-preview",
   "gemini-3-flash-preview",
@@ -148,7 +148,12 @@ export const antigravityConfig = (): ProviderConfigShape => ({
       "claude-opus-4-5@20251101"
     ] as OpenCodeModel
 
-    const models: Record<string, OpenCodeModel> = {
+    const models: Record<
+      string,
+      OpenCodeModel & {
+        variants?: Record<string, Record<string, unknown>>
+      }
+    > = {
       "gemini-3-flash": {
         ...geminiFlash,
         id: "gemini-3-flash",
@@ -157,7 +162,6 @@ export const antigravityConfig = (): ProviderConfigShape => ({
         ...geminiPro,
         id: "gemini-3-pro-low",
         name: "Gemini 3 Pro (Low)",
-        temperature: false,
         options: {
           thinkingConfig: {
             thinkingLevel: "low",
@@ -188,12 +192,12 @@ export const antigravityConfig = (): ProviderConfigShape => ({
       "claude-sonnet-4-5-thinking": {
         ...claudeSonnet,
         id: "claude-sonnet-4-5-thinking",
-        name: "Claude Sonnet 4.5 (Reasoning)",
+        name: "Claude Sonnet 4.5 (Thinking)",
       },
       "claude-opus-4-5-thinking": {
         ...claudeOpus,
         id: "claude-opus-4-5-thinking",
-        name: "Claude Opus 4.5 (Reasoning)",
+        name: "Claude Opus 4.5 (Thinking)",
       },
     }
 
@@ -201,7 +205,7 @@ export const antigravityConfig = (): ProviderConfigShape => ({
       ...googleProvider,
       id: antigravityConfig().SERVICE_NAME,
       name: antigravityConfig().DISPLAY_NAME,
-      api: antigravityConfig().ENDPOINTS.at(2) as string,
+      api: antigravityConfig().ENDPOINTS.at(0) as string,
       models,
     }
   },
@@ -238,6 +242,24 @@ export const antigravityConfig = (): ProviderConfigShape => ({
         | Record<string, unknown>
         | undefined
 
+      const tools = request.tools as
+        | Array<{ functionDeclarations?: Array<{ parameters?: unknown }> }>
+        | undefined
+      if (tools && Array.isArray(tools)) {
+        for (const tool of tools) {
+          if (
+            tool.functionDeclarations
+            && Array.isArray(tool.functionDeclarations)
+          ) {
+            for (const func of tool.functionDeclarations) {
+              if (!func.parameters) {
+                func.parameters = { type: "object", properties: {} }
+              }
+            }
+          }
+        }
+      }
+
       innerRequest.toolConfig = {
         functionCallingConfig: {
           mode: "VALIDATED",
@@ -278,101 +300,18 @@ export const antigravityConfig = (): ProviderConfigShape => ({
     }
 
     if (sessionId) {
-      const hashedSession = yield* Effect.promise(() => hash(sessionId))
-
-      const finalSessionId = [
-        `-${crypto.randomUUID()}`,
-        body.model,
-        body.project,
-        `seed-${hashedSession}`,
-      ].join(":")
-
-      innerRequest.sessionId = finalSessionId
-    }
-
-    if (innerRequest.tools && Array.isArray(innerRequest.tools)) {
-      const tools = innerRequest.tools as Array<Record<string, unknown>>
-      for (const tool of tools) {
-        if (
-          tool.functionDeclarations
-          && Array.isArray(tool.functionDeclarations)
-        ) {
-          const functionDeclarations = tool.functionDeclarations as Array<
-            Record<string, unknown>
-          >
-          for (let i = 0; i < functionDeclarations.length; i++) {
-            const declaration = functionDeclarations[i]
-            if (declaration && declaration.name === "todoread") {
-              functionDeclarations[i] = {
-                ...functionDeclarations[i],
-                parameters: {
-                  type: "object",
-                  properties: {
-                    _placeholder: {
-                      type: "boolean",
-                      description: "Placeholder. Always pass true.",
-                    },
-                  },
-                  required: ["_placeholder"],
-                  additionalProperties: false,
-                },
-              }
-            }
-          }
-        }
-      }
-    }
-
-    // if (
-    //   innerRequest.systemInstruction
-    //   && typeof innerRequest.systemInstruction === "object"
-    // ) {
-    //   const systemInstruction = innerRequest.systemInstruction as Record<
-    //     string,
-    //     unknown
-    //   >
-
-    //   if (systemInstruction.parts && Array.isArray(systemInstruction.parts)) {
-    //     let parts = systemInstruction.parts as Array<{ text: string }>
-
-    //     parts.unshift({
-    //       text: "You are Antigravity, a powerful agentic AI coding assistant designed by the Google DeepMind team working on Advanced Agentic Coding.",
-    //     })
-    //   }
-    // }
-
-    if (
-      innerRequest.systemInstruction
-      && typeof innerRequest.systemInstruction === "object"
-    ) {
-      const systemInstruction = innerRequest.systemInstruction as Record<
-        string,
-        unknown
-      >
-
-      systemInstruction.role = "user"
+      innerRequest.sessionId = sessionId
     }
 
     return {
       headers,
       url,
       body: {
-        ...body,
         requestType: "agent",
         userAgent: "antigravity",
         requestId: `agent-${crypto.randomUUID()}`,
+        ...body,
       },
     }
   }),
 })
-
-async function hash(str: string) {
-  const encoder = new TextEncoder()
-  const data = encoder.encode(str)
-
-  const hashBuffer = await crypto.subtle.digest("SHA-256", data)
-  const hashArray = Array.from(new Uint8Array(hashBuffer))
-  const hashHex = hashArray.map((b) => b.toString(16).padStart(2, "0")).join("")
-
-  return hashHex.slice(0, 16)
-}

package/src/services/oauth.ts

@@ -16,6 +16,7 @@ import { Data, Deferred, Effect, Fiber, Schema } from "effect"
 import { OAuth2Client } from "google-auth-library"
 import type { BunServeOptions } from "../types"
 import { ProviderConfig } from "./config"
+import open from "open"
 
 export class OAuthError extends Data.TaggedError("OAuthError")<{
   readonly reason: "browser" | "callback" | "state_mismatch" | "token_exchange"
@@ -69,7 +70,17 @@ class OAuth extends Effect.Service<OAuth>()("OAuth", {
         scope: config.SCOPES as unknown as string[],
         prompt: "consent",
       })
+
       yield* Effect.log(`OAuth2 authorization URL: ${authUrl}`)
+      yield* Effect.tryPromise({
+        try: () => open(authUrl),
+        catch: (cause) =>
+          new OAuthError({
+            reason: "browser",
+            message: "Failed to open browser",
+            cause,
+          }),
+      })
 
       const serverFiber = yield* HttpRouter.empty.pipe(
         HttpRouter.get(

package/src/transform/request.ts

@@ -49,8 +49,7 @@ export const transformRequest = Effect.fn("transformRequest")(function* (
   // Transform headers
   const headers = new Headers(init?.headers)
   headers.delete("x-api-key")
-  // headers.delete("x-goog-api-key")
-  headers.set("x-opencode-tools-debug", "1")
+  headers.delete("x-goog-api-key")
   headers.set("Authorization", `Bearer ${accessToken}`)
 
   for (const [key, value] of Object.entries(config.HEADERS)) {
@@ -69,8 +68,8 @@ export const transformRequest = Effect.fn("transformRequest")(function* (
   )
 
   const wrappedBody = {
-    project: projectId,
     model,
+    project: projectId,
     request: parsedBody ?? {},
   }
 

package/src/models.json

@@ -1,155 +0,0 @@
-{
-  "id": "google",
-  "env": ["GOOGLE_GENERATIVE_AI_API_KEY", "GEMINI_API_KEY"],
-  "npm": "@ai-sdk/google",
-  "name": "Google",
-  "doc": "https://ai.google.dev/gemini-api/docs/pricing",
-  "models": {
-    "gemini-3-flash-preview": {
-      "id": "gemini-3-flash-preview",
-      "name": "Gemini 3 Flash Preview",
-      "family": "gemini-flash",
-      "attachment": true,
-      "reasoning": true,
-      "tool_call": true,
-      "structured_output": true,
-      "temperature": true,
-      "knowledge": "2025-01",
-      "release_date": "2025-12-17",
-      "last_updated": "2025-12-17",
-      "modalities": {
-        "input": ["text", "image", "video", "audio", "pdf"],
-        "output": ["text"]
-      },
-      "open_weights": false,
-      "cost": {
-        "input": 0.5,
-        "output": 3,
-        "cache_read": 0.05,
-        "context_over_200k": {
-          "input": 0.5,
-          "output": 3,
-          "cache_read": 0.05
-        }
-      },
-      "limit": {
-        "context": 1048576,
-        "output": 65536
-      }
-    },
-    "gemini-3-pro-preview": {
-      "id": "gemini-3-pro-preview",
-      "name": "Gemini 3 Pro Preview",
-      "family": "gemini-pro",
-      "attachment": true,
-      "reasoning": true,
-      "tool_call": true,
-      "structured_output": true,
-      "temperature": true,
-      "knowledge": "2025-01",
-      "release_date": "2025-11-18",
-      "last_updated": "2025-11-18",
-      "modalities": {
-        "input": ["text", "image", "video", "audio", "pdf"],
-        "output": ["text"]
-      },
-      "open_weights": false,
-      "cost": {
-        "input": 2,
-        "output": 12,
-        "cache_read": 0.2,
-        "context_over_200k": {
-          "input": 4,
-          "output": 18,
-          "cache_read": 0.4
-        }
-      },
-      "limit": {
-        "context": 1000000,
-        "output": 64000
-      }
-    },
-    "gemini-2.5-flash": {
-      "id": "gemini-2.5-flash",
-      "name": "Gemini 2.5 Flash",
-      "family": "gemini-flash",
-      "attachment": true,
-      "reasoning": true,
-      "tool_call": true,
-      "structured_output": true,
-      "temperature": true,
-      "knowledge": "2025-01",
-      "release_date": "2025-03-20",
-      "last_updated": "2025-06-05",
-      "modalities": {
-        "input": ["text", "image", "audio", "video", "pdf"],
-        "output": ["text"]
-      },
-      "open_weights": false,
-      "cost": {
-        "input": 0.3,
-        "output": 2.5,
-        "cache_read": 0.075,
-        "input_audio": 1
-      },
-      "limit": {
-        "context": 1048576,
-        "output": 65536
-      }
-    },
-    "gemini-2.5-flash-lite": {
-      "id": "gemini-2.5-flash-lite",
-      "name": "Gemini 2.5 Flash Lite",
-      "family": "gemini-flash-lite",
-      "attachment": true,
-      "reasoning": true,
-      "tool_call": true,
-      "structured_output": true,
-      "temperature": true,
-      "knowledge": "2025-01",
-      "release_date": "2025-06-17",
-      "last_updated": "2025-06-17",
-      "modalities": {
-        "input": ["text", "image", "audio", "video", "pdf"],
-        "output": ["text"]
-      },
-      "open_weights": false,
-      "cost": {
-        "input": 0.1,
-        "output": 0.4,
-        "cache_read": 0.025
-      },
-      "limit": {
-        "context": 1048576,
-        "output": 65536
-      }
-    },
-    "gemini-2.5-pro": {
-      "id": "gemini-2.5-pro",
-      "name": "Gemini 2.5 Pro",
-      "family": "gemini-pro",
-      "attachment": true,
-      "reasoning": true,
-      "tool_call": true,
-      "structured_output": true,
-      "temperature": true,
-      "knowledge": "2025-01",
-      "release_date": "2025-03-20",
-      "last_updated": "2025-06-05",
-      "modalities": {
-        "input": ["text", "image", "audio", "video", "pdf"],
-        "output": ["text"]
-      },
-      "open_weights": false,
-      "cost": {
-        "input": 1.25,
-        "output": 10,
-        "cache_read": 0.31
-      },
-      "limit": {
-        "context": 1048576,
-        "output": 65536
-      }
-    }
-  }
-}

package/src/services/config.test.ts

@@ -1,194 +0,0 @@
-import { describe, expect, it } from "bun:test"
-import { transformRequest } from "../transform/request"
-import { antigravityConfig, ProviderConfig } from "./config"
-import { Session } from "./session"
-import { Effect, Layer, pipe } from "effect"
-import { generateText } from "ai"
-import { createGoogleGenerativeAI } from "@ai-sdk/google"
-
-describe("Antigravity transformRequest", () => {
-  const baseParams = {
-    accessToken: "test-token",
-    projectId: "test-project-123",
-  }
-
-  const config = antigravityConfig()
-
-  const MockSession = Layer.succeed(
-    Session,
-    Session.of({
-      getAccessToken: Effect.succeed(baseParams.accessToken),
-      ensureProject: Effect.succeed({
-        cloudaicompanionProject: baseParams.projectId,
-        currentTier: {
-          id: "free",
-          name: "Free",
-          description: "",
-          userDefinedCloudaicompanionProject: false,
-          userDefinedProjectId: "test-project-123",
-        },
-        allowedTiers: [],
-        gcpManaged: false,
-        manageSubscriptionUri: "",
-      }),
-      setCredentials: () => Effect.void,
-    } as unknown as Session),
-  ).pipe(Layer.provideMerge(Layer.succeed(ProviderConfig, config)))
-
-  const setupTest = () => {
-    let capturedBody: unknown = null
-
-    const mockFetch = async (input: string | Request, init?: RequestInit) => {
-      const result = await pipe(
-        transformRequest(
-          input,
-          init as unknown as Parameters<typeof fetch>[1],
-          "https://example.com",
-        ),
-        Effect.provide(MockSession),
-        Effect.runPromise,
-      )
-      capturedBody = JSON.parse(result.init.body as string)
-
-      return new Response(
-        JSON.stringify({
-          candidates: [
-            {
-              content: {
-                parts: [{ text: "Hello" }],
-                role: "model",
-              },
-              finishReason: "STOP",
-            },
-          ],
-          usageMetadata: {
-            promptTokenCount: 1,
-            candidatesTokenCount: 1,
-            totalTokenCount: 2,
-          },
-        }),
-        {
-          status: 200,
-          headers: { "Content-Type": "application/json" },
-        },
-      )
-    }
-
-    const google = createGoogleGenerativeAI({
-      apiKey: "test-key",
-      fetch: mockFetch as typeof fetch,
-    })
-
-    return {
-      google,
-      getCapturedBody: () => capturedBody as Record<string, unknown>,
-    }
-  }
-
-  it("propagates sessionId from labels and cleans up labels", async () => {
-    const { google, getCapturedBody } = setupTest()
-
-    await generateText({
-      model: google("gemini-1.5-flash"),
-      prompt: "Hello",
-      providerOptions: {
-        google: {
-          labels: {
-            sessionId: "test-session-id",
-            otherLabel: "keep-me",
-          },
-        },
-      },
-    })
-
-    const body = getCapturedBody()
-    const innerRequest = body.request as Record<string, unknown>
-    const labels = innerRequest.labels as Record<string, unknown>
-
-    // sessionId should be moved to innerRequest and transformed
-    // The actual format is `-${uuid}:${model}:${project}:seed-${hashedSession}`
-    expect(innerRequest.sessionId).not.toBe("test-session-id")
-    expect(innerRequest.sessionId).toBeString()
-    expect(
-      (innerRequest.sessionId as string).includes("gemini-1.5-flash"),
-    ).toBe(true)
-    expect(
-      (innerRequest.sessionId as string).includes("test-project-123"),
-    ).toBe(true)
-
-    // labels should be cleaned up (sessionId removed)
-    expect(labels.sessionId).toBeUndefined()
-    expect(labels.otherLabel).toBe("keep-me")
-
-    // Metadata should be present
-    expect(body.requestType).toBe("agent")
-    expect(body.userAgent).toBe("antigravity")
-    expect(body.requestId).toBeDefined()
-  })
-
-  it("removes labels object if it becomes empty after sessionId extraction", async () => {
-    const { google, getCapturedBody } = setupTest()
-
-    await generateText({
-      model: google("gemini-1.5-flash"),
-      prompt: "Hello",
-      providerOptions: {
-        google: {
-          labels: {
-            sessionId: "test-session-id",
-          },
-        },
-      },
-    })
-
-    const body = getCapturedBody()
-    const innerRequest = body.request as Record<string, unknown>
-
-    expect(innerRequest.sessionId).not.toBe("test-session-id")
-    expect(innerRequest.sessionId).toBeString()
-    expect(innerRequest.labels).toBeUndefined()
-  })
-
-  it("replaces todoread tool definition with placeholder schema", async () => {
-    const { google, getCapturedBody } = setupTest()
-
-    await generateText({
-      model: google("gemini-1.5-flash"),
-      prompt: "Hello",
-      tools: {
-        todoread: {
-          description: "Use this tool to read your todo list",
-          parameters: { type: "object", properties: {} },
-        },
-      },
-    })
-
-    const body = getCapturedBody()
-    const innerRequest = body.request as Record<string, unknown>
-    const tools = innerRequest.tools as Array<Record<string, unknown>>
-    const firstTool = tools[0]
-    expect(firstTool).toBeDefined()
-    if (!firstTool) throw new Error("firstTool is undefined")
-    const functionDeclarations = firstTool.functionDeclarations as Array<
-      Record<string, unknown>
-    >
-    const todoread = functionDeclarations.find((f) => f.name === "todoread")
-
-    expect(todoread).toBeDefined()
-    expect(todoread).toEqual({
-      name: "todoread",
-      description: "Use this tool to read your todo list",
-      parameters: {
-        type: "object",
-        properties: {
-          _placeholder: {
-            type: "boolean",
-            description: "Placeholder. Always pass true.",
-          },
-        },
-        required: ["_placeholder"],
-        additionalProperties: false,
-      },
-    })
-  })
-})