@byline/ai 1.9.0 → 1.9.1

package/dist/models/google/generate.test.node.js

@@ -7,39 +7,32 @@
  */
 import { beforeEach, describe, expect, it } from 'vitest';
 import { getAiServerConfig as getServerConfig } from '../../config/ai-config';
-import { generateDoc as generateDocNative, generateDocStreaming as generateDocStreamingNative, } from './generate-native';
-import { generateDocStreaming as generateDocStreamingVercel, generateDoc as generateDocVercel, } from './generate-vercel';
+import { generateDoc, generateDocStreaming } from './generate';
 const MODEL = 'gemini-2.5-flash';
 describe('google generate', () => {
     beforeEach(async () => { });
     const runReal = true; // = process.env.AI_RUN_REAL_TESTS === 'true'
     if (runReal) {
-        /***
-         * Generates a document from Google using structured outputs.
-         */
-        it('makes a real Google request (manual run) from native provider', async () => {
+        it('makes a real Google request (manual run)', async () => {
             const config = getServerConfig();
             if (!config.ai.google.apiKey) {
                 throw new Error('GOOGLE_API_KEY is required for real Google tests.');
             }
-            const result = await generateDocNative({
+            const result = await generateDoc({
                 apiKey: config.ai.google.apiKey,
                 model: MODEL,
                 prompt: 'Write a short description of the solar system..',
             });
-            console.log('RESULT (Google Native):', result);
+            console.log('RESULT (Google):', result);
             expect(result).toBeTruthy();
             expect(typeof result).toBe('object');
         }, 30000);
-        /***
-         * Generates a document from Google using structured outputs via streaming.
-         */
-        it.skip('streams a real Google response (manual run) from native provider', async () => {
+        it.skip('streams a real Google response (manual run)', async () => {
             const config = getServerConfig();
             if (!config.ai.google.apiKey) {
                 throw new Error('GOOGLE_API_KEY is required for real Google tests.');
             }
-            const streamResult = generateDocStreamingNative({
+            const streamResult = generateDocStreaming({
                 apiKey: config.ai.google.apiKey,
                 model: MODEL,
                 prompt: 'Write a short description of the solar system..',
@@ -47,51 +40,10 @@ describe('google generate', () => {
             let streamedText = '';
             for await (const chunk of streamResult.text) {
                 streamedText += chunk;
-                console.log('STREAM CHUNK (Google Native):', chunk);
+                console.log('STREAM CHUNK (Google):', chunk);
             }
             const final = await streamResult.final;
-            console.log('FINAL RESULT (Google Native):', final);
-            expect(streamedText.length).toBeGreaterThanOrEqual(0);
-            expect(final).toBeTruthy();
-            expect(typeof final).toBe('object');
-        }, 30000);
-        /***
-         * Generates a document from Google using structured outputs.
-         */
-        it('makes a real Google request (manual run) from vercel provider', async () => {
-            const config = getServerConfig();
-            if (!config.ai.google.apiKey) {
-                throw new Error('GOOGLE_API_KEY is required for real Google tests.');
-            }
-            const result = await generateDocVercel({
-                apiKey: config.ai.google.apiKey,
-                model: MODEL,
-                prompt: 'Write a short description of the solar system..',
-            });
-            console.log('RESULT (Google Vercel):', result);
-            expect(result).toBeTruthy();
-            expect(typeof result).toBe('object');
-        }, 30000);
-        /***
-         * Generates a document from Google using structured outputs via streaming.
-         */
-        it.skip('streams a real Google response (manual run) from vercel provider', async () => {
-            const config = getServerConfig();
-            if (!config.ai.google.apiKey) {
-                throw new Error('GOOGLE_API_KEY is required for real Google tests.');
-            }
-            const streamResult = generateDocStreamingVercel({
-                apiKey: config.ai.google.apiKey,
-                model: MODEL,
-                prompt: 'Write a short description of the solar system..',
-            });
-            let streamedText = '';
-            for await (const chunk of streamResult.text) {
-                streamedText += chunk;
-                console.log('STREAM CHUNK (Google Vercel):', chunk);
-            }
-            const final = await streamResult.final;
-            console.log('FINAL RESULT (Google Vercel):', final);
+            console.log('FINAL RESULT (Google):', final);
             expect(streamedText.length).toBeGreaterThanOrEqual(0);
             expect(final).toBeTruthy();
             expect(typeof final).toBe('object');

package/dist/models/google/generate.test.node.js.map

@@ -1 +1 @@
-{"version":3,"file":"generate.test.node.js","sourceRoot":"","sources":["../../../src/models/google/generate.test.node.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAEzD,OAAO,EAAE,iBAAiB,IAAI,eAAe,EAAE,MAAM,wBAAwB,CAAA;AAC7E,OAAO,EACL,WAAW,IAAI,iBAAiB,EAChC,oBAAoB,IAAI,0BAA0B,GACnD,MAAM,mBAAmB,CAAA;AAC1B,OAAO,EACL,oBAAoB,IAAI,0BAA0B,EAClD,WAAW,IAAI,iBAAiB,GACjC,MAAM,mBAAmB,CAAA;AAE1B,MAAM,KAAK,GAAG,kBAAkB,CAAA;AAEhC,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,UAAU,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC,CAAA;IAE1B,MAAM,OAAO,GAAG,IAAI,CAAA,CAAC,6CAA6C;IAElE,IAAI,OAAO,EAAE,CAAC;QACZ;;WAEG;QACH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,MAAM,GAAG,eAAe,EAAE,CAAA;YAChC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAA;YACtE,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC;gBACrC,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM;gBAC/B,KAAK,EAAE,KAAK;gBACZ,MAAM,EAAE,iDAAiD;aAC1D,CAAC,CAAA;YAEF,OAAO,CAAC,GAAG,CAAC,yBAAyB,EAAE,MAAM,CAAC,CAAA;YAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,EAAE,CAAA;YAC3B,MAAM,CAAC,OAAO,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACtC,CAAC,EAAE,KAAK,CAAC,CAAA;QAET;;WAEG;QACH,EAAE,CAAC,IAAI,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YACrF,MAAM,MAAM,GAAG,eAAe,EAAE,CAAA;YAChC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAA;YACtE,CAAC;YAED,MAAM,YAAY,GAAG,0BAA0B,CAAC;gBAC9C,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM;gBAC/B,KAAK,EAAE,KAAK;gBACZ,MAAM,EAAE,iDAAiD;aAC1D,CAAC,CAAA;YAEF,IAAI,YAAY,GAAG,EAAE,CAAA;YACrB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,YAAY,CAAC,IAAI,EAAE,CAAC;gBAC5C,YAAY,IAAI,KAAK,CAAA;gBACrB,OAAO,CAAC,GAAG,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAA;YACrD,CAAC;YAED,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,KAAK,CAAA;YACtC,OAAO,CAAC,GAAG,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAA;YAEnD,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAA;YACrD,MAAM,CAAC,KAAK,CAAC,CAAC,UAAU,EAAE,CAAA;YAC1B,MAAM,CAAC,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACrC,CAAC,EAAE,KAAK,CAAC,CAAA;QAET;;WAEG;QACH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,MAAM,GAAG,eAAe,EAAE,CAAA;YAChC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAA;YACtE,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC;gBACrC,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM;gBAC/B,KAAK,EAAE,KAAK;gBACZ,MAAM,EAAE,iDAAiD;aAC1D,CAAC,CAAA;YAEF,OAAO,CAAC,GAAG,CAAC,yBAAyB,EAAE,MAAM,CAAC,CAAA;YAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,EAAE,CAAA;YAC3B,MAAM,CAAC,OAAO,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACtC,CAAC,EAAE,KAAK,CAAC,CAAA;QAET;;WAEG;QACH,EAAE,CAAC,IAAI,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YACrF,MAAM,MAAM,GAAG,eAAe,EAAE,CAAA;YAChC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAA;YACtE,CAAC;YAED,MAAM,YAAY,GAAG,0BAA0B,CAAC;gBAC9C,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM;gBAC/B,KAAK,EAAE,KAAK;gBACZ,MAAM,EAAE,iDAAiD;aAC1D,CAAC,CAAA;YAEF,IAAI,YAAY,GAAG,EAAE,CAAA;YACrB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,YAAY,CAAC,IAAI,EAAE,CAAC;gBAC5C,YAAY,IAAI,KAAK,CAAA;gBACrB,OAAO,CAAC,GAAG,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAA;YACrD,CAAC;YAED,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,KAAK,CAAA;YACtC,OAAO,CAAC,GAAG,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAA;YACnD,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAA;YACrD,MAAM,CAAC,KAAK,CAAC,CAAC,UAAU,EAAE,CAAA;YAC1B,MAAM,CAAC,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACrC,CAAC,EAAE,KAAK,CAAC,CAAA;IACX,CAAC;SAAM,CAAC;QACN,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE;YACtC,OAAO,CAAC,GAAG,CAAC,sDAAsD,CAAC,CAAA;QACrE,CAAC,CAAC,CAAA;IACJ,CAAC;AACH,CAAC,CAAC,CAAA"}
+{"version":3,"file":"generate.test.node.js","sourceRoot":"","sources":["../../../src/models/google/generate.test.node.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAEzD,OAAO,EAAE,iBAAiB,IAAI,eAAe,EAAE,MAAM,wBAAwB,CAAA;AAC7E,OAAO,EAAE,WAAW,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAA;AAE9D,MAAM,KAAK,GAAG,kBAAkB,CAAA;AAEhC,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,UAAU,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC,CAAA;IAE1B,MAAM,OAAO,GAAG,IAAI,CAAA,CAAC,6CAA6C;IAElE,IAAI,OAAO,EAAE,CAAC;QACZ,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,MAAM,GAAG,eAAe,EAAE,CAAA;YAChC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAA;YACtE,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC;gBAC/B,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM;gBAC/B,KAAK,EAAE,KAAK;gBACZ,MAAM,EAAE,iDAAiD;aAC1D,CAAC,CAAA;YAEF,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,MAAM,CAAC,CAAA;YACvC,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,EAAE,CAAA;YAC3B,MAAM,CAAC,OAAO,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACtC,CAAC,EAAE,KAAK,CAAC,CAAA;QAET,EAAE,CAAC,IAAI,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,MAAM,GAAG,eAAe,EAAE,CAAA;YAChC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAA;YACtE,CAAC;YAED,MAAM,YAAY,GAAG,oBAAoB,CAAC;gBACxC,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM;gBAC/B,KAAK,EAAE,KAAK;gBACZ,MAAM,EAAE,iDAAiD;aAC1D,CAAC,CAAA;YAEF,IAAI,YAAY,GAAG,EAAE,CAAA;YACrB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,YAAY,CAAC,IAAI,EAAE,CAAC;gBAC5C,YAAY,IAAI,KAAK,CAAA;gBACrB,OAAO,CAAC,GAAG,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAA;YAC9C,CAAC;YAED,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,KAAK,CAAA;YACtC,OAAO,CAAC,GAAG,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAA;YAE5C,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAA;YACrD,MAAM,CAAC,KAAK,CAAC,CAAC,UAAU,EAAE,CAAA;YAC1B,MAAM,CAAC,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACrC,CAAC,EAAE,KAAK,CAAC,CAAA;IACX,CAAC;SAAM,CAAC;QACN,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE;YACtC,OAAO,CAAC,GAAG,CAAC,sDAAsD,CAAC,CAAA;QACrE,CAAC,CAAC,CAAA;IACJ,CAAC;AACH,CAAC,CAAC,CAAA"}

package/dist/models/google/patch.d.ts

@@ -5,8 +5,29 @@
  *
  * Copyright (c) Infonomic Company Limited
  */
-import { patchDoc as patchVercelDoc, patchDocStreaming as patchVercelDocStreaming } from './patch-vercel';
-import type { Sdk } from '../../@types';
-export declare const getPatchDoc: (sdk: Sdk) => typeof patchVercelDoc;
-export declare const getPatchDocStreaming: (sdk: Sdk) => typeof patchVercelDocStreaming;
+import type { LexicalTextEditsResponse } from '../../utils/lexical-text-edits';
+export declare function patchDoc(options: {
+    apiKey: string;
+    model: string;
+    prompt: string;
+    textNodes: Array<{
+        id: number;
+        text: string;
+    }>;
+    signal?: AbortSignal;
+}): Promise<LexicalTextEditsResponse>;
+export type PatchDocStreamingResult = {
+    text: AsyncIterable<string>;
+    final: Promise<LexicalTextEditsResponse>;
+};
+export declare function patchDocStreaming(options: {
+    apiKey: string;
+    model: string;
+    prompt: string;
+    textNodes: Array<{
+        id: number;
+        text: string;
+    }>;
+    signal?: AbortSignal;
+}): PatchDocStreamingResult;
 //# sourceMappingURL=patch.d.ts.map

package/dist/models/google/patch.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"patch.d.ts","sourceRoot":"","sources":["../../../src/models/google/patch.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAMH,OAAO,EACL,QAAQ,IAAI,cAAc,EAC1B,iBAAiB,IAAI,uBAAuB,EAC7C,MAAM,gBAAgB,CAAA;AACvB,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,UAAU,CAAA;AAEnC,eAAO,MAAM,WAAW,GAAI,KAAK,GAAG,0BAEnC,CAAA;AAED,eAAO,MAAM,oBAAoB,GAAI,KAAK,GAAG,mCAE5C,CAAA"}
+{"version":3,"file":"patch.d.ts","sourceRoot":"","sources":["../../../src/models/google/patch.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAKH,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,4BAA4B,CAAA;AAqG1E,wBAAsB,QAAQ,CAAC,OAAO,EAAE;IACtC,MAAM,EAAE,MAAM,CAAA;IACd,KAAK,EAAE,MAAM,CAAA;IACb,MAAM,EAAE,MAAM,CAAA;IACd,SAAS,EAAE,KAAK,CAAC;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAA;IAC9C,MAAM,CAAC,EAAE,WAAW,CAAA;CACrB,GAAG,OAAO,CAAC,wBAAwB,CAAC,CA2BpC;AAED,MAAM,MAAM,uBAAuB,GAAG;IACpC,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,CAAA;IAC3B,KAAK,EAAE,OAAO,CAAC,wBAAwB,CAAC,CAAA;CACzC,CAAA;AAED,wBAAgB,iBAAiB,CAAC,OAAO,EAAE;IACzC,MAAM,EAAE,MAAM,CAAA;IACd,KAAK,EAAE,MAAM,CAAA;IACb,MAAM,EAAE,MAAM,CAAA;IACd,SAAS,EAAE,KAAK,CAAC;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAA;IAC9C,MAAM,CAAC,EAAE,WAAW,CAAA;CACrB,GAAG,uBAAuB,CAuD1B"}

package/dist/models/google/patch.js

@@ -5,12 +5,160 @@
  *
  * Copyright (c) Infonomic Company Limited
  */
-import { patchDoc as patchNativeDoc, patchDocStreaming as patchNativeDocStreaming, } from './patch-native';
-import { patchDoc as patchVercelDoc, patchDocStreaming as patchVercelDocStreaming, } from './patch-vercel';
-export const getPatchDoc = (sdk) => {
-    return sdk === 'vercel' ? patchVercelDoc : patchNativeDoc;
+import { GoogleGenAI } from '@google/genai';
+import { buildPatchSystemPrompt } from '../../prompts';
+const buildPatchUserPrompt = (instruction, textNodes) => {
+    return [
+        `INSTRUCTION: ${instruction}`,
+        '',
+        'INPUT_TEXT_NODES_JSON:',
+        JSON.stringify(textNodes),
+    ].join('\n');
 };
-export const getPatchDocStreaming = (sdk) => {
-    return sdk === 'vercel' ? patchVercelDocStreaming : patchNativeDocStreaming;
+const tryParseJson = (text) => {
+    const trimmed = text.trim();
+    if (trimmed.length === 0)
+        return undefined;
+    const fenced = trimmed.match(/^```(?:json)?\s*([\s\S]*?)\s*```$/i);
+    const candidate = fenced?.[1]?.trim() ?? trimmed;
+    return JSON.parse(candidate);
 };
+const normalizeJsonSchemaForGemini = (schema) => {
+    if (schema == null)
+        return schema;
+    if (Array.isArray(schema))
+        return schema.map(normalizeJsonSchemaForGemini);
+    if (typeof schema !== 'object')
+        return schema;
+    const obj = schema;
+    const next = {};
+    for (const [key, value] of Object.entries(obj)) {
+        if (key === 'const') {
+            next.enum = [value];
+            continue;
+        }
+        if (key === 'type' && Array.isArray(value)) {
+            next.anyOf = value.map((t) => ({ type: t }));
+            continue;
+        }
+        next[key] = normalizeJsonSchemaForGemini(value);
+    }
+    return next;
+};
+const normalizePatchResponse = (value) => {
+    const maybeObj = value;
+    // Allow the model to return the edits array directly.
+    const editsCandidate = Array.isArray(maybeObj)
+        ? maybeObj
+        : Array.isArray(maybeObj?.edits)
+            ? maybeObj.edits
+            : Array.isArray(maybeObj?.output)
+                ? maybeObj.output
+                : undefined;
+    if (!Array.isArray(editsCandidate)) {
+        throw new Error('Google patch response is missing an edits array.');
+    }
+    const edits = editsCandidate
+        .map((e) => {
+        const idRaw = e?.id;
+        const textRaw = e?.text;
+        const id = typeof idRaw === 'number' ? idRaw : typeof idRaw === 'string' ? Number(idRaw) : NaN;
+        const text = typeof textRaw === 'string' ? textRaw : textRaw == null ? '' : String(textRaw);
+        if (!Number.isFinite(id))
+            return undefined;
+        return { id, text };
+    })
+        .filter(Boolean);
+    return { edits };
+};
+const googlePatchSchema = {
+    type: 'object',
+    additionalProperties: false,
+    properties: {
+        edits: {
+            type: 'array',
+            items: {
+                type: 'object',
+                additionalProperties: false,
+                properties: {
+                    id: { type: 'number' },
+                    text: { type: 'string' },
+                },
+                required: ['id', 'text'],
+            },
+        },
+    },
+    required: ['edits'],
+};
+export async function patchDoc(options) {
+    const ai = new GoogleGenAI({ apiKey: options.apiKey });
+    const normalizedResponseJsonSchema = normalizeJsonSchemaForGemini(googlePatchSchema);
+    const response = await ai.models.generateContent({
+        model: options.model,
+        contents: buildPatchUserPrompt(options.prompt, options.textNodes),
+        config: {
+            systemInstruction: buildPatchSystemPrompt(),
+            responseMimeType: 'application/json',
+            responseJsonSchema: normalizedResponseJsonSchema,
+            abortSignal: options.signal,
+        },
+    });
+    const text = response.text?.trim() ?? '';
+    if (text.length === 0) {
+        throw new Error('Google model returned an empty response.');
+    }
+    const parsed = tryParseJson(text);
+    if (parsed && typeof parsed === 'object') {
+        return normalizePatchResponse(parsed);
+    }
+    throw new Error('Google structured output did not return a JSON object.');
+}
+export function patchDocStreaming(options) {
+    const ai = new GoogleGenAI({ apiKey: options.apiKey });
+    const normalizedResponseJsonSchema = normalizeJsonSchemaForGemini(googlePatchSchema);
+    const streamPromise = ai.models.generateContentStream({
+        model: options.model,
+        contents: buildPatchUserPrompt(options.prompt, options.textNodes),
+        config: {
+            systemInstruction: buildPatchSystemPrompt(),
+            responseMimeType: 'application/json',
+            responseJsonSchema: normalizedResponseJsonSchema,
+            abortSignal: options.signal,
+        },
+    });
+    let resolveFinal;
+    let rejectFinal;
+    const final = new Promise((resolve, reject) => {
+        resolveFinal = resolve;
+        rejectFinal = reject;
+    });
+    const text = (async function* () {
+        let buffered = '';
+        try {
+            const stream = await streamPromise;
+            for await (const chunk of stream) {
+                const chunkText = typeof chunk?.text === 'function' ? chunk.text() : chunk?.text;
+                if (typeof chunkText === 'string' && chunkText.length > 0) {
+                    buffered += chunkText;
+                    yield chunkText;
+                }
+            }
+            const trimmed = buffered.trim();
+            if (trimmed.length === 0) {
+                throw new Error('Google model returned an empty response.');
+            }
+            const parsed = tryParseJson(trimmed);
+            if (parsed && typeof parsed === 'object') {
+                resolveFinal(normalizePatchResponse(parsed));
+                return;
+            }
+            throw new Error('Google structured output did not return a JSON object.');
+        }
+        catch (error) {
+            rejectFinal(error);
+            throw error;
+        }
+    })();
+    return { text, final };
+}
 //# sourceMappingURL=patch.js.map

package/dist/models/google/patch.js.map

@@ -1 +1 @@
-{"version":3,"file":"patch.js","sourceRoot":"","sources":["../../../src/models/google/patch.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EACL,QAAQ,IAAI,cAAc,EAC1B,iBAAiB,IAAI,uBAAuB,GAC7C,MAAM,gBAAgB,CAAA;AACvB,OAAO,EACL,QAAQ,IAAI,cAAc,EAC1B,iBAAiB,IAAI,uBAAuB,GAC7C,MAAM,gBAAgB,CAAA;AAGvB,MAAM,CAAC,MAAM,WAAW,GAAG,CAAC,GAAQ,EAAE,EAAE;IACtC,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,cAAc,CAAA;AAC3D,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,GAAQ,EAAE,EAAE;IAC/C,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,uBAAuB,CAAC,CAAC,CAAC,uBAAuB,CAAA;AAC7E,CAAC,CAAA"}
+{"version":3,"file":"patch.js","sourceRoot":"","sources":["../../../src/models/google/patch.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAA;AAE3C,OAAO,EAAE,sBAAsB,EAAE,MAAM,WAAW,CAAA;AAGlD,MAAM,oBAAoB,GAAG,CAC3B,WAAmB,EACnB,SAA8C,EAC9C,EAAE;IACF,OAAO;QACL,gBAAgB,WAAW,EAAE;QAC7B,EAAE;QACF,wBAAwB;QACxB,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;KAC1B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;AACd,CAAC,CAAA;AAED,MAAM,YAAY,GAAG,CAAC,IAAY,EAAW,EAAE;IAC7C,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAA;IAC3B,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,SAAS,CAAA;IAE1C,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,oCAAoC,CAAC,CAAA;IAClE,MAAM,SAAS,GAAG,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,OAAO,CAAA;IAEhD,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAA;AAC9B,CAAC,CAAA;AAED,MAAM,4BAA4B,GAAG,CAAC,MAAe,EAAW,EAAE;IAChE,IAAI,MAAM,IAAI,IAAI;QAAE,OAAO,MAAM,CAAA;IACjC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;QAAE,OAAO,MAAM,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAA;IAC1E,IAAI,OAAO,MAAM,KAAK,QAAQ;QAAE,OAAO,MAAM,CAAA;IAE7C,MAAM,GAAG,GAAG,MAA6B,CAAA;IACzC,MAAM,IAAI,GAAwB,EAAE,CAAA;IAEpC,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QAC/C,IAAI,GAAG,KAAK,OAAO,EAAE,CAAC;YACpB,IAAI,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,CAAA;YACnB,SAAQ;QACV,CAAC;QAED,IAAI,GAAG,KAAK,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YAC3C,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAA;YAC5C,SAAQ;QACV,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,GAAG,4BAA4B,CAAC,KAAK,CAAC,CAAA;IACjD,CAAC;IAED,OAAO,IAAI,CAAA;AACb,CAAC,CAAA;AAED,MAAM,sBAAsB,GAAG,CAAC,KAAc,EAA4B,EAAE;IAC1E,MAAM,QAAQ,GAAG,KAAY,CAAA;IAE7B,sDAAsD;IACtD,MAAM,cAAc,GAAG,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC;QAC5C,CAAC,CAAC,QAAQ;QACV,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,KAAK,CAAC;YAC9B,CAAC,CAAC,QAAQ,CAAC,KAAK;YAChB,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,MAAM,CAAC;gBAC/B,CAAC,CAAC,QAAQ,CAAC,MAAM;gBACjB,CAAC,CAAC,SAAS,CAAA;IAEjB,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAA;IACrE,CAAC;IAED,MAAM,KAAK,GAAG,cAAc;SACzB,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE;QACd,MAAM,KAAK,GAAG,CAAC,EAAE,EAAE,CAAA;QACnB,MAAM,OAAO,GAAG,CAAC,EAAE,IAAI,CAAA;QAEvB,MAAM,EAAE,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAA;QAC9F,MAAM,IAAI,GAAG,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;QAE3F,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;YAAE,OAAO,SAAS,CAAA;QAC1C,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAA;IACrB,CAAC,CAAC;SACD,MAAM,CAAC,OAAO,CAAwC,CAAA;IAEzD,OAAO,EAAE,KAAK,EAAE,CAAA;AAClB,CAAC,CAAA;AAED,MAAM,iBAAiB,GAAG;IACxB,IAAI,EAAE,QAAQ;IACd,oBAAoB,EAAE,KAAK;IAC3B,UAAU,EAAE;QACV,KAAK,EAAE;YACL,IAAI,EAAE,OAAO;YACb,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,oBAAoB,EAAE,KAAK;gBAC3B,UAAU,EAAE;oBACV,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;oBACtB,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;iBACzB;gBACD,QAAQ,EAAE,CAAC,IAAI,EAAE,MAAM,CAAC;aACzB;SACF;KACF;IACD,QAAQ,EAAE,CAAC,OAAO,CAAC;CACX,CAAA;AAEV,MAAM,CAAC,KAAK,UAAU,QAAQ,CAAC,OAM9B;IACC,MAAM,EAAE,GAAG,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAA;IAEtD,MAAM,4BAA4B,GAAG,4BAA4B,CAAC,iBAAiB,CAAC,CAAA;IAEpF,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC;QAC/C,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,QAAQ,EAAE,oBAAoB,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,SAAS,CAAC;QACjE,MAAM,EAAE;YACN,iBAAiB,EAAE,sBAAsB,EAAE;YAC3C,gBAAgB,EAAE,kBAAkB;YACpC,kBAAkB,EAAE,4BAA4B;YAChD,WAAW,EAAE,OAAO,CAAC,MAAM;SAC5B;KACF,CAAC,CAAA;IAEF,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,CAAA;IACxC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACtB,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAA;IAC7D,CAAC;IAED,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,CAAA;IACjC,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QACzC,OAAO,sBAAsB,CAAC,MAAM,CAAC,CAAA;IACvC,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAA;AAC3E,CAAC;AAOD,MAAM,UAAU,iBAAiB,CAAC,OAMjC;IACC,MAAM,EAAE,GAAG,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAA;IAEtD,MAAM,4BAA4B,GAAG,4BAA4B,CAAC,iBAAiB,CAAC,CAAA;IAEpF,MAAM,aAAa,GAAG,EAAE,CAAC,MAAM,CAAC,qBAAqB,CAAC;QACpD,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,QAAQ,EAAE,oBAAoB,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,SAAS,CAAC;QACjE,MAAM,EAAE;YACN,iBAAiB,EAAE,sBAAsB,EAAE;YAC3C,gBAAgB,EAAE,kBAAkB;YACpC,kBAAkB,EAAE,4BAA4B;YAChD,WAAW,EAAE,OAAO,CAAC,MAAM;SAC5B;KACF,CAAC,CAAA;IAEF,IAAI,YAAwD,CAAA;IAC5D,IAAI,WAAwC,CAAA;IAC5C,MAAM,KAAK,GAAG,IAAI,OAAO,CAA2B,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACtE,YAAY,GAAG,OAAO,CAAA;QACtB,WAAW,GAAG,MAAM,CAAA;IACtB,CAAC,CAAC,CAAA;IAEF,MAAM,IAAI,GAAG,CAAC,KAAK,SAAS,CAAC;QAC3B,IAAI,QAAQ,GAAG,EAAE,CAAA;QACjB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,aAAa,CAAA;YAClC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBACjC,MAAM,SAAS,GACb,OAAQ,KAAa,EAAE,IAAI,KAAK,UAAU,CAAC,CAAC,CAAE,KAAa,CAAC,IAAI,EAAE,CAAC,CAAC,CAAE,KAAa,EAAE,IAAI,CAAA;gBAC3F,IAAI,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC1D,QAAQ,IAAI,SAAS,CAAA;oBACrB,MAAM,SAAS,CAAA;gBACjB,CAAC;YACH,CAAC;YAED,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAA;YAC/B,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAA;YAC7D,CAAC;YAED,MAAM,MAAM,GAAG,YAAY,CAAC,OAAO,CAAC,CAAA;YACpC,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;gBACzC,YAAY,CAAC,sBAAsB,CAAC,MAAM,CAAC,CAAC,CAAA;gBAC5C,OAAM;YACR,CAAC;YAED,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAA;QAC3E,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,WAAW,CAAC,KAAK,CAAC,CAAA;YAClB,MAAM,KAAK,CAAA;QACb,CAAC;IACH,CAAC,CAAC,EAAE,CAAA;IAEJ,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAA;AACxB,CAAC"}

package/dist/models/google/patch.test.node.js

@@ -7,22 +7,18 @@
  */
 import { beforeEach, describe, expect, it } from 'vitest';
 import { getAiServerConfig as getServerConfig } from '../../config/ai-config';
-import { patchDoc as patchDocNative, patchDocStreaming as patchDocStreamingNative, } from './patch-native';
-import { patchDocStreaming as patchDocStreamingVercel, patchDoc as patchDocVercel, } from './patch-vercel';
+import { patchDoc, patchDocStreaming } from './patch';
 const MODEL = 'gemini-2.5-flash';
 describe('google patch', () => {
     beforeEach(async () => { });
     const runReal = process.env.AI_RUN_REAL_TESTS === 'true';
     if (runReal) {
-        /***
-         * Generates a document from Google using structured outputs.
-         */
-        it.skip('makes a real Google request (manual run) from native provider', async () => {
+        it.skip('makes a real Google request (manual run)', async () => {
             const config = getServerConfig();
             if (!config.ai.google.apiKey) {
                 throw new Error('GOOGLE_API_KEY is required for real Google tests.');
             }
-            const result = await patchDocNative({
+            const result = await patchDoc({
                 apiKey: config.ai.google.apiKey,
                 model: MODEL,
                 prompt: 'Translate into French.',
@@ -32,15 +28,12 @@ describe('google patch', () => {
             expect(result).toBeTruthy();
             expect(typeof result).toBe('object');
         }, 30000);
-        /***
-         * Generates a document from Google using structured outputs via streaming.
-         */
-        it.skip('streams a real Google response (manual run) from native provider', async () => {
+        it.skip('streams a real Google response (manual run)', async () => {
             const config = getServerConfig();
             if (!config.ai.google.apiKey) {
                 throw new Error('GOOGLE_API_KEY is required for real Google tests.');
             }
-            const streamResult = patchDocStreamingNative({
+            const streamResult = patchDocStreaming({
                 apiKey: config.ai.google.apiKey,
                 model: MODEL,
                 prompt: 'Translate into French.',
@@ -49,53 +42,10 @@ describe('google patch', () => {
             let streamedText = '';
             for await (const chunk of streamResult.text) {
                 streamedText += chunk;
-                console.log('STREAM CHUNK (Google Native):', chunk);
+                console.log('STREAM CHUNK (Google):', chunk);
             }
             const final = await streamResult.final;
-            console.log('FINAL RESULT (Google Native):', final);
-            expect(streamedText.length).toBeGreaterThanOrEqual(0);
-            expect(final).toBeTruthy();
-            expect(typeof final).toBe('object');
-        }, 30000);
-        /***
-         * Generates a document from Google using structured outputs.
-         */
-        it.skip('makes a real Google request (manual run) from vercel provider', async () => {
-            const config = getServerConfig();
-            if (!config.ai.google.apiKey) {
-                throw new Error('GOOGLE_API_KEY is required for real Google tests.');
-            }
-            const result = await patchDocVercel({
-                apiKey: config.ai.google.apiKey,
-                model: MODEL,
-                prompt: 'Translate into French.',
-                textNodes: [{ id: 0, text: 'The sun is shining' }],
-            });
-            console.log(result);
-            expect(result).toBeTruthy();
-            expect(typeof result).toBe('object');
-        }, 30000);
-        /***
-         * Generates a document from Google using structured outputs via streaming.
-         */
-        it.skip('streams a real Google response (manual run) from vercel provider', async () => {
-            const config = getServerConfig();
-            if (!config.ai.google.apiKey) {
-                throw new Error('GOOGLE_API_KEY is required for real Google tests.');
-            }
-            const streamResult = patchDocStreamingVercel({
-                apiKey: config.ai.google.apiKey,
-                model: MODEL,
-                prompt: 'Translate into French.',
-                textNodes: [{ id: 0, text: 'The sun is shining' }],
-            });
-            let streamedText = '';
-            for await (const chunk of streamResult.text) {
-                streamedText += chunk;
-                console.log('STREAM CHUNK (Google Vercel):', chunk);
-            }
-            const final = await streamResult.final;
-            console.log('FINAL RESULT (Google Vercel):', final);
+            console.log('FINAL RESULT (Google):', final);
             expect(streamedText.length).toBeGreaterThanOrEqual(0);
             expect(final).toBeTruthy();
             expect(typeof final).toBe('object');

package/dist/models/google/patch.test.node.js.map

@@ -1 +1 @@
-{"version":3,"file":"patch.test.node.js","sourceRoot":"","sources":["../../../src/models/google/patch.test.node.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAEzD,OAAO,EAAE,iBAAiB,IAAI,eAAe,EAAE,MAAM,wBAAwB,CAAA;AAC7E,OAAO,EACL,QAAQ,IAAI,cAAc,EAC1B,iBAAiB,IAAI,uBAAuB,GAC7C,MAAM,gBAAgB,CAAA;AACvB,OAAO,EACL,iBAAiB,IAAI,uBAAuB,EAC5C,QAAQ,IAAI,cAAc,GAC3B,MAAM,gBAAgB,CAAA;AAEvB,MAAM,KAAK,GAAG,kBAAkB,CAAA;AAEhC,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,UAAU,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC,CAAA;IAE1B,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,iBAAiB,KAAK,MAAM,CAAA;IAExD,IAAI,OAAO,EAAE,CAAC;QACZ;;WAEG;QACH,EAAE,CAAC,IAAI,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAClF,MAAM,MAAM,GAAG,eAAe,EAAE,CAAA;YAChC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAA;YACtE,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC;gBAClC,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM;gBAC/B,KAAK,EAAE,KAAK;gBACZ,MAAM,EAAE,wBAAwB;gBAChC,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC;aACnD,CAAC,CAAA;YAEF,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;YACnB,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,EAAE,CAAA;YAC3B,MAAM,CAAC,OAAO,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACtC,CAAC,EAAE,KAAK,CAAC,CAAA;QAET;;WAEG;QACH,EAAE,CAAC,IAAI,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YACrF,MAAM,MAAM,GAAG,eAAe,EAAE,CAAA;YAChC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAA;YACtE,CAAC;YAED,MAAM,YAAY,GAAG,uBAAuB,CAAC;gBAC3C,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM;gBAC/B,KAAK,EAAE,KAAK;gBACZ,MAAM,EAAE,wBAAwB;gBAChC,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC;aACnD,CAAC,CAAA;YAEF,IAAI,YAAY,GAAG,EAAE,CAAA;YACrB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,YAAY,CAAC,IAAI,EAAE,CAAC;gBAC5C,YAAY,IAAI,KAAK,CAAA;gBACrB,OAAO,CAAC,GAAG,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAA;YACrD,CAAC;YAED,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,KAAK,CAAA;YACtC,OAAO,CAAC,GAAG,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAA;YACnD,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAA;YACrD,MAAM,CAAC,KAAK,CAAC,CAAC,UAAU,EAAE,CAAA;YAC1B,MAAM,CAAC,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACrC,CAAC,EAAE,KAAK,CAAC,CAAA;QAET;;WAEG;QACH,EAAE,CAAC,IAAI,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAClF,MAAM,MAAM,GAAG,eAAe,EAAE,CAAA;YAChC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAA;YACtE,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC;gBAClC,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM;gBAC/B,KAAK,EAAE,KAAK;gBACZ,MAAM,EAAE,wBAAwB;gBAChC,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC;aACnD,CAAC,CAAA;YAEF,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;YACnB,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,EAAE,CAAA;YAC3B,MAAM,CAAC,OAAO,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACtC,CAAC,EAAE,KAAK,CAAC,CAAA;QAET;;WAEG;QACH,EAAE,CAAC,IAAI,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YACrF,MAAM,MAAM,GAAG,eAAe,EAAE,CAAA;YAChC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAA;YACtE,CAAC;YAED,MAAM,YAAY,GAAG,uBAAuB,CAAC;gBAC3C,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM;gBAC/B,KAAK,EAAE,KAAK;gBACZ,MAAM,EAAE,wBAAwB;gBAChC,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC;aACnD,CAAC,CAAA;YAEF,IAAI,YAAY,GAAG,EAAE,CAAA;YACrB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,YAAY,CAAC,IAAI,EAAE,CAAC;gBAC5C,YAAY,IAAI,KAAK,CAAA;gBACrB,OAAO,CAAC,GAAG,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAA;YACrD,CAAC;YAED,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,KAAK,CAAA;YACtC,OAAO,CAAC,GAAG,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAA;YAEnD,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAA;YACrD,MAAM,CAAC,KAAK,CAAC,CAAC,UAAU,EAAE,CAAA;YAC1B,MAAM,CAAC,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACrC,CAAC,EAAE,KAAK,CAAC,CAAA;IACX,CAAC;SAAM,CAAC;QACN,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE;YACtC,OAAO,CAAC,GAAG,CAAC,sDAAsD,CAAC,CAAA;QACrE,CAAC,CAAC,CAAA;IACJ,CAAC;AACH,CAAC,CAAC,CAAA"}
+{"version":3,"file":"patch.test.node.js","sourceRoot":"","sources":["../../../src/models/google/patch.test.node.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAEzD,OAAO,EAAE,iBAAiB,IAAI,eAAe,EAAE,MAAM,wBAAwB,CAAA;AAC7E,OAAO,EAAE,QAAQ,EAAE,iBAAiB,EAAE,MAAM,SAAS,CAAA;AAErD,MAAM,KAAK,GAAG,kBAAkB,CAAA;AAEhC,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,UAAU,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC,CAAA;IAE1B,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,iBAAiB,KAAK,MAAM,CAAA;IAExD,IAAI,OAAO,EAAE,CAAC;QACZ,EAAE,CAAC,IAAI,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,MAAM,GAAG,eAAe,EAAE,CAAA;YAChC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAA;YACtE,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC;gBAC5B,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM;gBAC/B,KAAK,EAAE,KAAK;gBACZ,MAAM,EAAE,wBAAwB;gBAChC,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC;aACnD,CAAC,CAAA;YAEF,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;YACnB,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,EAAE,CAAA;YAC3B,MAAM,CAAC,OAAO,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACtC,CAAC,EAAE,KAAK,CAAC,CAAA;QAET,EAAE,CAAC,IAAI,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,MAAM,GAAG,eAAe,EAAE,CAAA;YAChC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAA;YACtE,CAAC;YAED,MAAM,YAAY,GAAG,iBAAiB,CAAC;gBACrC,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM;gBAC/B,KAAK,EAAE,KAAK;gBACZ,MAAM,EAAE,wBAAwB;gBAChC,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC;aACnD,CAAC,CAAA;YAEF,IAAI,YAAY,GAAG,EAAE,CAAA;YACrB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,YAAY,CAAC,IAAI,EAAE,CAAC;gBAC5C,YAAY,IAAI,KAAK,CAAA;gBACrB,OAAO,CAAC,GAAG,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAA;YAC9C,CAAC;YAED,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,KAAK,CAAA;YACtC,OAAO,CAAC,GAAG,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAA;YAC5C,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAA;YACrD,MAAM,CAAC,KAAK,CAAC,CAAC,UAAU,EAAE,CAAA;YAC1B,MAAM,CAAC,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACrC,CAAC,EAAE,KAAK,CAAC,CAAA;IACX,CAAC;SAAM,CAAC;QACN,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE;YACtC,OAAO,CAAC,GAAG,CAAC,sDAAsD,CAAC,CAAA;QACrE,CAAC,CAAC,CAAA;IACJ,CAAC;AACH,CAAC,CAAC,CAAA"}

package/dist/models/openai/generate.d.ts

@@ -5,12 +5,61 @@
  *
  * Copyright (c) Infonomic Company Limited
  */
-import { generateDoc as generateVercelDoc, generateDocStreaming as generateVercelDocStreaming, generateHtml as generateVercelHtml, generateHtmlStreaming as generateVercelHtmlStreaming, generateText as generateVercelText, generateTextStreaming as generateVercelTextStreaming } from './generate-vercel';
-import type { Sdk } from '../../@types';
-export declare const getGenerateDoc: (sdk: Sdk) => typeof generateVercelDoc;
-export declare const getGenerateDocStreaming: (sdk: Sdk) => typeof generateVercelDocStreaming;
-export declare const getGenerateHtml: (sdk: Sdk) => typeof generateVercelHtml;
-export declare const getGenerateHtmlStreaming: (sdk: Sdk) => typeof generateVercelHtmlStreaming;
-export declare const getGenerateText: (sdk: Sdk) => typeof generateVercelText;
-export declare const getGenerateTextStreaming: (sdk: Sdk) => typeof generateVercelTextStreaming;
+import type { GeneratedDoc } from '../../utils/convert-to-lexical';
+export type GenerateDocStreamingResult = {
+    text: AsyncIterable<string>;
+    final: Promise<GeneratedDoc>;
+};
+export type GenerateTextStreamingResult = {
+    text: AsyncIterable<string>;
+    final: Promise<string>;
+};
+export type GenerateHtmlStreamingResult = {
+    text: AsyncIterable<string>;
+    final: Promise<string>;
+};
+export declare function generateHtml(options: {
+    apiKey: string;
+    model: string;
+    prompt: string;
+    signal?: AbortSignal;
+}): Promise<string>;
+export declare function generateHtmlStreaming(options: {
+    apiKey: string;
+    model: string;
+    prompt: string;
+    signal?: AbortSignal;
+}): GenerateHtmlStreamingResult;
+export declare function generateText(options: {
+    apiKey: string;
+    model: string;
+    prompt: string;
+    maxLength?: number;
+    signal?: AbortSignal;
+}): Promise<string>;
+export declare function generateTextStreaming(options: {
+    apiKey: string;
+    model: string;
+    prompt: string;
+    maxLength?: number;
+    signal?: AbortSignal;
+}): GenerateTextStreamingResult;
+/***
+ * Generates a document from OpenAI using structured outputs.
+ */
+export declare function generateDoc(options: {
+    apiKey: string;
+    model: string;
+    prompt: string;
+    signal?: AbortSignal;
+}): Promise<GeneratedDoc>;
+/***
+ * Streams a document generation from OpenAI using structured outputs.
+ */
+export declare function generateDocStreaming(options: {
+    apiKey: string;
+    model: string;
+    prompt: string;
+    signal?: AbortSignal;
+}): GenerateDocStreamingResult;
 //# sourceMappingURL=generate.d.ts.map

package/dist/models/openai/generate.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"generate.d.ts","sourceRoot":"","sources":["../../../src/models/openai/generate.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAUH,OAAO,EACL,WAAW,IAAI,iBAAiB,EAChC,oBAAoB,IAAI,0BAA0B,EAClD,YAAY,IAAI,kBAAkB,EAClC,qBAAqB,IAAI,2BAA2B,EACpD,YAAY,IAAI,kBAAkB,EAClC,qBAAqB,IAAI,2BAA2B,EACrD,MAAM,mBAAmB,CAAA;AAC1B,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,UAAU,CAAA;AAEnC,eAAO,MAAM,cAAc,GAAI,KAAK,GAAG,6BAEtC,CAAA;AAED,eAAO,MAAM,uBAAuB,GAAI,KAAK,GAAG,sCAE/C,CAAA;AAED,eAAO,MAAM,eAAe,GAAI,KAAK,GAAG,8BAEvC,CAAA;AAED,eAAO,MAAM,wBAAwB,GAAI,KAAK,GAAG,uCAEhD,CAAA;AAED,eAAO,MAAM,eAAe,GAAI,KAAK,GAAG,8BAEvC,CAAA;AAED,eAAO,MAAM,wBAAwB,GAAI,KAAK,GAAG,uCAEhD,CAAA"}
+{"version":3,"file":"generate.d.ts","sourceRoot":"","sources":["../../../src/models/openai/generate.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAYH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAA;AAE9D,MAAM,MAAM,0BAA0B,GAAG;IACvC,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,CAAA;IAC3B,KAAK,EAAE,OAAO,CAAC,YAAY,CAAC,CAAA;CAC7B,CAAA;AAED,MAAM,MAAM,2BAA2B,GAAG;IACxC,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,CAAA;IAC3B,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;CACvB,CAAA;AAED,MAAM,MAAM,2BAA2B,GAAG;IACxC,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,CAAA;IAC3B,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;CACvB,CAAA;AAED,wBAAsB,YAAY,CAAC,OAAO,EAAE;IAC1C,MAAM,EAAE,MAAM,CAAA;IACd,KAAK,EAAE,MAAM,CAAA;IACb,MAAM,EAAE,MAAM,CAAA;IACd,MAAM,CAAC,EAAE,WAAW,CAAA;CACrB,GAAG,OAAO,CAAC,MAAM,CAAC,CA8BlB;AAED,wBAAgB,qBAAqB,CAAC,OAAO,EAAE;IAC7C,MAAM,EAAE,MAAM,CAAA;IACd,KAAK,EAAE,MAAM,CAAA;IACb,MAAM,EAAE,MAAM,CAAA;IACd,MAAM,CAAC,EAAE,WAAW,CAAA;CACrB,GAAG,2BAA2B,CA6C9B;AAED,wBAAsB,YAAY,CAAC,OAAO,EAAE;IAC1C,MAAM,EAAE,MAAM,CAAA;IACd,KAAK,EAAE,MAAM,CAAA;IACb,MAAM,EAAE,MAAM,CAAA;IACd,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,MAAM,CAAC,EAAE,WAAW,CAAA;CACrB,GAAG,OAAO,CAAC,MAAM,CAAC,CA6ClB;AAED,wBAAgB,qBAAqB,CAAC,OAAO,EAAE;IAC7C,MAAM,EAAE,MAAM,CAAA;IACd,KAAK,EAAE,MAAM,CAAA;IACb,MAAM,EAAE,MAAM,CAAA;IACd,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,MAAM,CAAC,EAAE,WAAW,CAAA;CACrB,GAAG,2BAA2B,CA4E9B;AAgDD;;GAEG;AACH,wBAAsB,WAAW,CAAC,OAAO,EAAE;IACzC,MAAM,EAAE,MAAM,CAAA;IACd,KAAK,EAAE,MAAM,CAAA;IACb,MAAM,EAAE,MAAM,CAAA;IACd,MAAM,CAAC,EAAE,WAAW,CAAA;CACrB,GAAG,OAAO,CAAC,YAAY,CAAC,CA0CxB;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,OAAO,EAAE;IAC5C,MAAM,EAAE,MAAM,CAAA;IACd,KAAK,EAAE,MAAM,CAAA;IACb,MAAM,EAAE,MAAM,CAAA;IACd,MAAM,CAAC,EAAE,WAAW,CAAA;CACrB,GAAG,0BAA0B,CAmD7B"}