opencode-google-auth 0.0.3 → 0.0.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-google-auth",
-  "version": "0.0.3",
+  "version": "0.0.5",
   "description": "_description_",
   "keywords": [
     "opencode-google-auth"

package/src/antigravity-spoof.txt

@@ -1 +1,7 @@
-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**
+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**
+
+The USER will send you requests, which you must always prioritize addressing.

package/src/main.ts

@@ -69,6 +69,11 @@ const customFetch = Effect.fn(function* (
       )
     }
 
+    if (config.skipRequestTransform) {
+      yield* Effect.log("Skipping response transformation")
+      return response
+    }
+
     return result.streaming ?
         transformStreamingResponse(response)
       : yield* Effect.promise(() => transformNonStreamingResponse(response))
@@ -167,112 +172,112 @@ export const geminiCli: Plugin = async (context) => {
   }
 }
 
-export const antigravity: Plugin = async (context) => {
-  const runtime = makeRuntime({
-    openCodeCtx: context,
-    providerConfig: antigravityConfig(),
-  })
-
-  const config = await runtime.runPromise(
-    Effect.gen(function* () {
-      const providerConfig = yield* ProviderConfig
-      const modelsDev = yield* fetchModelsDev
-
-      return providerConfig.getConfig(modelsDev)
-    }),
-  )
-
-  return {
-    config: async (cfg) => {
-      cfg.provider ??= {}
-      cfg.provider[config.id as string] = config
-    },
-    auth: {
-      provider: config.id as string,
-      loader: async (getAuth) => {
-        const auth = await getAuth()
-        if (auth.type !== "oauth") return {}
-
-        const credentials: Credentials = {
-          access_token: auth.access,
-          refresh_token: auth.refresh,
-          expiry_date: auth.expires,
-        }
-
-        await runtime.runPromise(
-          Effect.gen(function* () {
-            const session = yield* Session
-            yield* session.setCredentials(credentials)
-          }),
-        )
-
-        return {
-          apiKey: "",
-          fetch: (async (input, init) => {
-            const response = await runtime.runPromise(customFetch(input, init))
-            return response
-          }) as typeof fetch,
-        } satisfies GoogleGenerativeAIProviderSettings
-      },
-      methods: [
-        {
-          type: "oauth",
-          label: "OAuth with Google",
-          authorize: async () => {
-            const result = await runtime.runPromise(
-              Effect.gen(function* () {
-                const oauth = yield* OAuth
-                return yield* oauth.authenticate
-              }),
-            )
-
-            return {
-              url: "",
-              method: "auto",
-              instructions: "You are now authenticated!",
-              callback: async () => {
-                const accessToken = result.access_token
-                const refreshToken = result.refresh_token
-                const expiryDate = result.expiry_date
-
-                if (!accessToken || !refreshToken || !expiryDate) {
-                  return { type: "failed" }
-                }
-
-                return {
-                  type: "success",
-                  provider: config.id as string,
-                  access: accessToken,
-                  refresh: refreshToken,
-                  expires: expiryDate,
-                }
-              },
-            }
-          },
-        },
-      ],
-    },
-    "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),
-      )
-
-      if (input.model.providerID === config.id) {
-        output.options = {
-          ...output.options,
-          labels: {
-            sessionId: input.sessionID,
-          },
-        } satisfies GoogleGenerativeAIProviderOptions
-      }
-
-      await runtime.runPromise(
-        Effect.log("chat.params event after:", input.model, output.options),
-      )
-    },
-  }
-}
+// export const antigravity: Plugin = async (context) => {
+//   const runtime = makeRuntime({
+//     openCodeCtx: context,
+//     providerConfig: antigravityConfig(),
+//   })
+
+//   const config = await runtime.runPromise(
+//     Effect.gen(function* () {
+//       const providerConfig = yield* ProviderConfig
+//       const modelsDev = yield* fetchModelsDev
+
+//       return providerConfig.getConfig(modelsDev)
+//     }),
+//   )
+
+//   return {
+//     config: async (cfg) => {
+//       cfg.provider ??= {}
+//       cfg.provider[config.id as string] = config
+//     },
+//     auth: {
+//       provider: config.id as string,
+//       loader: async (getAuth) => {
+//         const auth = await getAuth()
+//         if (auth.type !== "oauth") return {}
+
+//         const credentials: Credentials = {
+//           access_token: auth.access,
+//           refresh_token: auth.refresh,
+//           expiry_date: auth.expires,
+//         }
+
+//         await runtime.runPromise(
+//           Effect.gen(function* () {
+//             const session = yield* Session
+//             yield* session.setCredentials(credentials)
+//           }),
+//         )
+
+//         return {
+//           apiKey: "",
+//           fetch: (async (input, init) => {
+//             const response = await runtime.runPromise(customFetch(input, init))
+//             return response
+//           }) as typeof fetch,
+//         } satisfies GoogleGenerativeAIProviderSettings
+//       },
+//       methods: [
+//         {
+//           type: "oauth",
+//           label: "OAuth with Google",
+//           authorize: async () => {
+//             const result = await runtime.runPromise(
+//               Effect.gen(function* () {
+//                 const oauth = yield* OAuth
+//                 return yield* oauth.authenticate
+//               }),
+//             )
+
+//             return {
+//               url: "",
+//               method: "auto",
+//               instructions: "You are now authenticated!",
+//               callback: async () => {
+//                 const accessToken = result.access_token
+//                 const refreshToken = result.refresh_token
+//                 const expiryDate = result.expiry_date
+
+//                 if (!accessToken || !refreshToken || !expiryDate) {
+//                   return { type: "failed" }
+//                 }
+
+//                 return {
+//                   type: "success",
+//                   provider: config.id as string,
+//                   access: accessToken,
+//                   refresh: refreshToken,
+//                   expires: expiryDate,
+//                 }
+//               },
+//             }
+//           },
+//         },
+//       ],
+//     },
+//     "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),
+//       )
+
+//       if (input.model.providerID === config.id) {
+//         output.options = {
+//           ...output.options,
+//           labels: {
+//             sessionId: input.sessionID,
+//           },
+//         } satisfies GoogleGenerativeAIProviderOptions
+//       }
+
+//       await runtime.runPromise(
+//         Effect.log("chat.params event after:", input.model, output.options),
+//       )
+//     },
+//   }
+// }

package/src/services/config.ts

@@ -33,6 +33,7 @@ export interface ProviderConfigShape {
     headers: Headers
     url: URL
   }>
+  readonly skipRequestTransform?: boolean
 }
 
 export class ProviderConfig extends Context.Tag("ProviderConfig")<
@@ -103,6 +104,7 @@ export const geminiCliConfig = (): ProviderConfigShape => ({
       models: filteredModels as Record<string, OpenCodeModel>,
     }
   },
+  skipRequestTransform: false,
 })
 
 export const antigravityConfig = (): ProviderConfigShape => ({
@@ -193,11 +195,51 @@ export const antigravityConfig = (): ProviderConfigShape => ({
         ...claudeSonnet,
         id: "claude-sonnet-4-5-thinking",
         name: "Claude Sonnet 4.5 (Thinking)",
+        options: {
+          thinkingConfig: {
+            includeThoughts: true,
+            thinkingBudget: 31999,
+          },
+        } satisfies GoogleGenerativeAIProviderOptions,
+        variants: {
+          high: {
+            thinkingConfig: {
+              includeThoughts: true,
+              thinkingBudget: 16000,
+            },
+          } satisfies GoogleGenerativeAIProviderOptions,
+          max: {
+            thinkingConfig: {
+              includeThoughts: true,
+              thinkingBudget: 31999,
+            },
+          } satisfies GoogleGenerativeAIProviderOptions,
+        },
       },
       "claude-opus-4-5-thinking": {
         ...claudeOpus,
         id: "claude-opus-4-5-thinking",
         name: "Claude Opus 4.5 (Thinking)",
+        options: {
+          thinkingConfig: {
+            includeThoughts: true,
+            thinkingBudget: 31999,
+          },
+        } satisfies GoogleGenerativeAIProviderOptions,
+        variants: {
+          high: {
+            thinkingConfig: {
+              includeThoughts: true,
+              thinkingBudget: 16000,
+            },
+          } satisfies GoogleGenerativeAIProviderOptions,
+          max: {
+            thinkingConfig: {
+              includeThoughts: true,
+              thinkingBudget: 31999,
+            },
+          } satisfies GoogleGenerativeAIProviderOptions,
+        },
       },
     }
 
@@ -205,7 +247,8 @@ 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,
+      npm: "google-antigravity-ai-provider",
       models,
     }
   },
@@ -237,63 +280,12 @@ export const antigravityConfig = (): ProviderConfigShape => ({
     const isThinking = body.model.toLowerCase().includes("thinking")
 
     if (isClaude && body.request && typeof body.request === "object") {
-      const request = body.request as Record<string, unknown>
-      const generationConfig = request.generationConfig as
-        | 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",
         },
       }
 
-      // For non-thinking Claude, remove thinkingConfig entirely
-      if (!isThinking && generationConfig?.thinkingConfig) {
-        delete generationConfig.thinkingConfig
-      }
-
-      // For thinking Claude, convert camelCase to snake_case and add default budget
-      if (isThinking && generationConfig?.thinkingConfig) {
-        const thinkingConfig = generationConfig.thinkingConfig as Record<
-          string,
-          unknown
-        >
-
-        if (thinkingConfig.includeThoughts !== undefined) {
-          thinkingConfig.include_thoughts = thinkingConfig.includeThoughts
-          delete thinkingConfig.includeThoughts
-        }
-
-        if (thinkingConfig.thinkingBudget !== undefined) {
-          thinkingConfig.thinking_budget = thinkingConfig.thinkingBudget
-          delete thinkingConfig.thinkingBudget
-        }
-
-        // Add default thinking_budget if not present (required for Claude thinking)
-        if (thinkingConfig.thinking_budget === undefined) {
-          thinkingConfig.thinking_budget = 32768 // Default to high tier
-        }
-      }
-
       if (isThinking) {
         headers.set("anthropic-beta", "interleaved-thinking-2025-05-14")
       }
@@ -314,4 +306,5 @@ export const antigravityConfig = (): ProviderConfigShape => ({
       },
     }
   }),
+  skipRequestTransform: true,
 })