Merge pull request #9 from mnfst/feat/google-gemini

feat: support native Gemini requests

Author: guillaumegay13

CONTRIBUTING.md

@@ -15,6 +15,7 @@ node/src/core/heal-api.ts    the only code that talks to the heal service
 node/src/core/engine.ts      orchestration, and createAutofix(adapter)
 node/src/openai/             OpenAI's routes, /chat/completions + /responses
 node/src/anthropic/          Anthropic's routes, /messages
+node/src/google/             Google's native :generateContent routes
 ```
 
 `python/src/mnfst_autofix/core/` mirrors that, plus `attempt.py`, which holds

README.md

@@ -11,7 +11,7 @@ Heal your failed LLM requests on-the-fly to avoid any downtime. Malformed parame
 
 ### Node / TypeScript
 
-One package - compatible with every LLM SDK.
+One package for OpenAI, Anthropic, and native Google Gemini requests.
 
 ```bash
 npm install @mnfst/autofix
@@ -57,6 +57,18 @@ const message = await client.messages.create({
 });
 ```
 
+#### Vercel AI SDK - Google Gemini
+
+```ts
+import { createGoogleGenerativeAI } from '@ai-sdk/google';
+import { autofix } from '@mnfst/autofix';
+
+const google = createGoogleGenerativeAI({
+  apiKey: process.env.GOOGLE_GENERATIVE_AI_API_KEY,
+  fetch: autofix({ sendMessages: true }),
+});
+```
+
 Other providers like OpenRouter uses the OpenAI SDK with its own `baseURL`:
 
 ```ts
@@ -72,7 +84,7 @@ const client = new OpenAI({
 
 ### Python
 
-One package - compatible with every LLM SDK.
+One package for OpenAI, Anthropic, and native Google Gemini requests.
 
 ```bash
 pip install mnfst-autofix
@@ -161,10 +173,9 @@ schema bodies, identity fields, and credential fields stay local. The provider e
 endpoint origin and path, derived workspace id, and SDK source are also sent.
 
 `autofix({ sendMessages: true })` in Node / `autofix(send_messages=True)` in
-Python (default off) opts the top-level `messages` array in, verbatim — for
-teams pointing at their own heal service who want the conversation visible next
-to the failure it caused. It is observability only: the heal API still cannot
-author or rewrite a message on the way back.
+Python (default off) opts the top-level `messages` or Gemini `contents` array in,
+verbatim. It is observability only: the heal API still cannot author or rewrite
+the conversation on the way back.
 
 ## See heals
 
@@ -182,8 +193,8 @@ const client = new OpenAI({
 
 | Package | Language | SDKs |
 |---|---|---|
-| [`@mnfst/autofix`](node) | Node / TypeScript | OpenAI, Anthropic |
-| [`mnfst-autofix`](python) | Python | OpenAI, Anthropic |
+| [`@mnfst/autofix`](node) | Node / TypeScript | OpenAI, Anthropic, Google Gemini |
+| [`mnfst-autofix`](python) | Python | OpenAI, Anthropic, Google Gemini |
 
 ## License
 

SECURITY.md

@@ -12,10 +12,10 @@ endpoint, on whatever host your SDK is pointed at, and sends the failed request
 to a heal API. Worth knowing exactly what that means:
 
 **Removed before anything is sent, and restored from your own copy on the
-replay:** `messages`, `input`, `instructions`, `system`, `tools`,
-`response_format`, `text.format`. The strip is a union across dialects, so a
-content field is removed on every API, not just the one that normally carries
-it. See `node/src/core/anonymize.ts` (about 100 lines) or
+replay:** `messages`, Gemini `contents`, `input`, `instructions`, system prompts,
+tools, and response schemas. The strip is a union across dialects, so a content
+field is removed on every API, not just the one that normally carries it. See
+`node/src/core/anonymize.ts` or
 `python/src/mnfst_autofix/core/anonymize.py`.
 
 **Sent as-is:** the provider's error message. Healing depends on it, and

node/README.md

@@ -1,6 +1,6 @@
 # Autofix Node
 
-Self-healing `fetch()` for LLM SDKs — the [OpenAI SDK](https://github.com/openai/openai-node) and the [Anthropic SDK](https://github.com/anthropics/anthropic-sdk-typescript). When a request fails with a fixable request-shape error — a renamed parameter, a restricted value, a retired model — autofix repairs it at runtime and retries. Your code doesn't change; your users never see the failure.
+Self-healing `fetch()` for OpenAI, Anthropic, and native Google Gemini requests. When a request fails with a fixable request-shape error — a renamed parameter, a restricted value, a retired model — autofix repairs it at runtime and retries. Your code doesn't change; your users never see the failure.
 
 ## Installation
 
@@ -49,6 +49,16 @@ const message = await client.messages.create({
 });
 ```
 
+```ts
+import { createGoogleGenerativeAI } from '@ai-sdk/google';
+import { autofix } from '@mnfst/autofix';
+
+const google = createGoogleGenerativeAI({
+  apiKey: process.env.GOOGLE_GENERATIVE_AI_API_KEY,
+  fetch: autofix({ sendMessages: true }),
+});
+```
+
 OpenRouter uses the OpenAI SDK with its own `baseURL`:
 
 ```ts
@@ -95,9 +105,8 @@ schema bodies, identity fields, and credential fields stay local. The provider e
 endpoint origin and path, derived workspace id, and SDK source are also sent.
 
 `autofix({ sendMessages: true })` (default `false`) opts the top-level `messages`
-array in, verbatim — for teams pointing at their own heal service who want the
-conversation visible next to the failure it caused. It is observability only:
-the heal API still cannot author or rewrite a message on the way back.
+or Gemini `contents` array in, verbatim. It is observability only: the heal API
+still cannot author or rewrite the conversation on the way back.
 
 ## See heals
 

node/examples/ai-demo.ts

@@ -1,5 +1,5 @@
 // E2E demo: Vercel AI SDK + autofix + the hosted heal API.
-// One autofix() fetch shared by BOTH providers — the root export routes by
+// One autofix() fetch shared by both providers — the root export routes by
 // path. Each scenario would normally throw; with autofix it heals to a 200.
 // Needs OPENAI_API_KEY and ANTHROPIC_API_KEY (via .env or the environment).
 //
@@ -9,8 +9,8 @@
 // the Responses API), so a bad *param* never reaches the provider through this
 // SDK and there is nothing for autofix to heal. The model id is the one field
 // the SDK forwards verbatim. The three scenarios below therefore cover the
-// three distinct paths in the root route table — /chat/completions, /responses
-// and /messages — which is the claim this demo exists to prove.
+// three distinct paths exercised here — /chat/completions, /responses and
+// /messages — which is the claim this demo exists to prove.
 
 import { generateText } from 'ai';
 import { createOpenAI } from '@ai-sdk/openai';
@@ -26,7 +26,7 @@ const healingFetch = autofix({
   },
 });
 
-// ONE fetch, two providers: routing is by path, not by import.
+// One fetch, two providers: routing is by path, not by import.
 const openai = createOpenAI({ fetch: healingFetch });
 const anthropic = createAnthropic({ fetch: healingFetch });
 

node/package-lock.json

@@ -1,7 +1,7 @@
 {
   "name": "@mnfst/autofix",
-  "version": "0.3.2",
-  "description": "Self-healing fetch() for LLM SDKs — OpenAI SDK, Anthropic SDK, and the Vercel AI SDK. Fixes request-side failures at runtime (renamed params, restricted values, retired models) and retries. Anonymized: your prompts never travel in the request.",
+  "version": "0.4.0",
+  "description": "Self-healing fetch() for OpenAI, Anthropic, and native Google Gemini requests. Fixes request-side failures at runtime and retries once.",
   "license": "MIT",
   "type": "module",
   "repository": {
@@ -13,6 +13,8 @@
   "keywords": [
     "openai",
     "anthropic",
+    "google",
+    "gemini",
     "vercel",
     "ai-sdk",
     "llm",
@@ -40,7 +42,7 @@
     "demo:anthropic": "tsx --env-file-if-exists=.env examples/anthropic-demo.ts",
     "lint": "eslint .",
     "prepublishOnly": "npm test && npm run build",
-    "test": "tsx --test tests/*.test.ts",
+    "test": "node scripts/run-tests.mjs",
     "typecheck": "tsc --noEmit",
     "test:package": "npm run build && node scripts/package-smoke.mjs",
     "test:coverage": "tsx --test --experimental-test-coverage --test-coverage-exclude='tests/**' --test-coverage-lines=100 --test-coverage-functions=100 --test-coverage-branches=95 tests/*.test.ts"
@@ -50,6 +52,7 @@
   },
   "devDependencies": {
     "@ai-sdk/anthropic": "^4.0.33",
+    "@ai-sdk/google": "^4.0.45",
     "@ai-sdk/openai": "^4.0.32",
     "@anthropic-ai/sdk": "^0.120.0",
     "@eslint/js": "^10.0.1",

node/package.json

@@ -0,0 +1,23 @@
+import { spawnSync } from 'node:child_process';
+import { readdirSync } from 'node:fs';
+import { createRequire } from 'node:module';
+import { fileURLToPath } from 'node:url';
+
+const require = createRequire(import.meta.url);
+const tsxCli = require.resolve('tsx/cli');
+const testsDirectory = new URL('../tests/', import.meta.url);
+const testFiles = readdirSync(testsDirectory)
+  .filter((file) => file.endsWith('.test.ts'))
+  .sort();
+
+for (const testFile of testFiles) {
+  const result = spawnSync(
+    process.execPath,
+    [tsxCli, fileURLToPath(new URL(testFile, testsDirectory))],
+    { stdio: 'inherit' },
+  );
+
+  if (result.status !== 0) {
+    process.exit(result.status ?? 1);
+  }
+}

node/scripts/run-tests.mjs

@@ -27,9 +27,11 @@
  * `user_location` is where the end user physically is.
  */
 const NEVER_TRAVELS: ReadonlySet<string> = new Set([
-  'content', 'input', 'instructions', 'system', 'prompt', 'stop', 'stop_sequences',
-  'schema', 'json_schema',
-  'user', 'metadata', 'safety_identifier', 'prompt_cache_key', 'user_location',
+  'content', 'contents', 'input', 'instructions', 'system', 'systemInstruction',
+  'prompt', 'stop', 'stop_sequences', 'stopSequences',
+  'schema', 'json_schema', 'responseSchema',
+  'user', 'metadata', 'labels', 'safety_identifier', 'prompt_cache_key', 'user_location',
+  'cachedContent',
   'prediction', 'authorization_token',
 ]);
 
@@ -88,9 +90,9 @@ function sentObject(source: Record<string, unknown>): Record<string, unknown> {
 /** Caller-chosen relaxations of the boundary. Nothing here is on by default. */
 export interface StripOptions {
   /**
-   * Send the top-level `messages` array to the heal API verbatim, content and
-   * all. Off by default. For callers who run their own heal service and want
-   * the conversation visible next to the failure it caused. Observability
+   * Send the top-level `messages` or Gemini `contents` array to the heal API
+   * verbatim, content and all. Off by default. For callers who want the
+   * conversation visible next to the failure it caused. Observability
    * only: the merge below never consults this, so the heal API still cannot
    * author or rewrite a message even after being shown them.
    */
@@ -104,6 +106,9 @@ export function stripRequest(request: Record<string, unknown>,
   if (options.sendMessages && Array.isArray(request.messages)) {
     kept.messages = structuredClone(request.messages);
   }
+  if (options.sendMessages && Array.isArray(request.contents)) {
+    kept.contents = structuredClone(request.contents);
+  }
   return kept;
 }