@luzzle/core 0.0.62 → 0.0.63

package/dist/src/llm/google.test.js

@@ -1,253 +0,0 @@
-import { describe, expect, vi, afterEach, test } from 'vitest';
-import { pieceFrontMatterFromPrompt } from './google.js';
-import { GoogleGenAI, createPartFromUri, } from '@google/genai';
-import { fileTypeFromBuffer, fileTypeFromFile } from 'file-type';
-import { readFile } from 'fs/promises';
-// Mock schema created locally to remove dependency on piece.fixtures.ts
-const makeSchema = (name) => {
-    return {
-        type: 'object',
-        title: name,
-        properties: {
-            title: { type: 'string', examples: ['title'] },
-            keywords: { type: 'string', nullable: true },
-        },
-        required: ['title'],
-        additionalProperties: false,
-    };
-};
-vi.mock('@luzzle/core');
-vi.mock('file-type');
-vi.mock('fs/promises');
-vi.mock('@google/genai', () => {
-    const Gemini = vi.fn();
-    Gemini.prototype.models = vi.fn();
-    Gemini.prototype.files = vi.fn();
-    Gemini.prototype.models.generateContent = vi.fn();
-    Gemini.prototype.files.upload = vi.fn();
-    Gemini.prototype.files.get = vi.fn();
-    return {
-        GoogleGenAI: Gemini,
-        HarmCategory: {},
-        HarmBlockThreshold: {},
-        createPartFromUri: vi.fn(),
-    };
-});
-const mocks = {
-    generateContent: vi.mocked(GoogleGenAI.prototype.models.generateContent),
-    uploadFile: vi.mocked(GoogleGenAI.prototype.files.upload),
-    getFile: vi.mocked(GoogleGenAI.prototype.files.get),
-    fileTypeFromFile: vi.mocked(fileTypeFromFile),
-    fileTypeFromBuffer: vi.mocked(fileTypeFromBuffer),
-    createPartFromUri: vi.mocked(createPartFromUri),
-    readFile: vi.mocked(readFile),
-};
-const spies = {};
-describe('lib/llm/google.ts', () => {
-    afterEach(() => {
-        Object.values(mocks).forEach((mock) => {
-            mock.mockReset();
-        });
-        Object.keys(spies).forEach((key) => {
-            spies[key].mockRestore();
-            delete spies[key];
-        });
-    });
-    test('pieceFrontMatterFromPrompt', async () => {
-        const apiKey = 'apiKey';
-        const schema = makeSchema('books');
-        const prompt = 'prompt';
-        const frontmatter = { field: 'value' };
-        const responseText = JSON.stringify(frontmatter);
-        mocks.generateContent.mockResolvedValueOnce({
-            text: responseText,
-        });
-        const generatedFrontmatter = await pieceFrontMatterFromPrompt(apiKey, schema, prompt);
-        expect(mocks.generateContent).toHaveBeenCalledTimes(1);
-        expect(mocks.generateContent).toHaveBeenCalledWith({
-            contents: expect.arrayContaining([prompt]),
-            config: expect.any(Object),
-            model: expect.any(String),
-        });
-        expect(generatedFrontmatter).toEqual(frontmatter);
-    });
-    test('pieceFrontMatterFromPrompt strips empty fields', async () => {
-        const apiKey = 'apiKey';
-        const schema = makeSchema('books');
-        const prompt = 'prompt';
-        const frontmatter = { field: 'value', field2: null, field3: undefined };
-        const responseText = JSON.stringify(frontmatter);
-        mocks.generateContent.mockResolvedValueOnce({
-            text: responseText,
-        });
-        const generatedFrontmatter = await pieceFrontMatterFromPrompt(apiKey, schema, prompt);
-        expect(mocks.generateContent).toHaveBeenCalledTimes(1);
-        expect(mocks.generateContent).toHaveBeenCalledWith({
-            contents: expect.arrayContaining([prompt]),
-            config: expect.any(Object),
-            model: expect.any(String),
-        });
-        expect(generatedFrontmatter).toEqual({ field: 'value' });
-    });
-    test('pieceFrontMatterFromPrompt returns on empty result', async () => {
-        const apiKey = 'apiKey';
-        const schema = makeSchema('books');
-        const prompt = 'prompt';
-        const responseText = undefined;
-        mocks.generateContent.mockResolvedValueOnce({
-            text: responseText,
-        });
-        const generatedFrontmatter = await pieceFrontMatterFromPrompt(apiKey, schema, prompt);
-        expect(mocks.generateContent).toHaveBeenCalledTimes(1);
-        expect(mocks.generateContent).toHaveBeenCalledWith({
-            contents: expect.arrayContaining([prompt]),
-            config: expect.any(Object),
-            model: expect.any(String),
-        });
-        expect(generatedFrontmatter).toEqual({});
-    });
-    test('generatePieceFrontmatter with a binary file', async () => {
-        const apiKey = 'apiKey';
-        const schema = makeSchema('books');
-        const prompt = 'prompt';
-        const file = '/path/to/file.pdf';
-        const uri = 'gs://another/path/to/file.pdf';
-        const name = 'file.pdf';
-        const mimeType = 'application/pdf';
-        const frontmatter = { field: 'value' };
-        const responseText = JSON.stringify(frontmatter);
-        const fileContent = 'fileContent';
-        vi.useFakeTimers();
-        mocks.generateContent.mockResolvedValueOnce({
-            text: responseText,
-        });
-        mocks.uploadFile.mockResolvedValue({ name, uri, mimeType });
-        mocks.getFile.mockResolvedValueOnce({ state: 'PROCESSING' });
-        mocks.getFile.mockResolvedValueOnce({ uri, mimeType, state: 'ACTIVE' });
-        mocks.fileTypeFromFile.mockResolvedValueOnce({ mime: mimeType, ext: 'pdf' });
-        mocks.createPartFromUri.mockReturnValue(fileContent);
-        vi.runAllTimersAsync();
-        const generatedFrontmatter = await pieceFrontMatterFromPrompt(apiKey, schema, prompt, [file]);
-        expect(mocks.uploadFile).toHaveBeenCalledWith({ file, config: { mimeType } });
-        expect(mocks.getFile).toHaveBeenCalledWith({ name });
-        expect(mocks.createPartFromUri).toHaveBeenCalledTimes(1);
-        expect(mocks.generateContent).toHaveBeenCalledTimes(1);
-        expect(mocks.generateContent).toHaveBeenCalledWith({
-            contents: expect.arrayContaining([prompt, fileContent]),
-            config: expect.any(Object),
-            model: expect.any(String),
-        });
-        expect(generatedFrontmatter).toEqual(frontmatter);
-        vi.useRealTimers();
-    });
-    test('generatePieceFrontmatter with a text file', async () => {
-        const apiKey = 'apiKey';
-        const schema = makeSchema('books');
-        const prompt = 'prompt';
-        const file = '/path/to/file.html';
-        const frontmatter = { field: 'value' };
-        const responseText = JSON.stringify(frontmatter);
-        const fileContent = 'fileContent';
-        mocks.generateContent.mockResolvedValueOnce({
-            text: responseText,
-        });
-        mocks.fileTypeFromFile.mockResolvedValueOnce(undefined);
-        mocks.readFile.mockResolvedValueOnce(fileContent);
-        const generatedFrontmatter = await pieceFrontMatterFromPrompt(apiKey, schema, prompt, [file]);
-        expect(mocks.generateContent).toHaveBeenCalledTimes(1);
-        expect(mocks.generateContent).toHaveBeenCalledWith({
-            contents: expect.arrayContaining([prompt, expect.stringContaining(fileContent)]),
-            config: expect.any(Object),
-            model: expect.any(String),
-        });
-        expect(generatedFrontmatter).toEqual(frontmatter);
-        vi.useRealTimers();
-    });
-    test('generatePieceFrontmatter with a text Buffer', async () => {
-        const apiKey = 'apiKey';
-        const schema = makeSchema('books');
-        const prompt = 'prompt';
-        const frontmatter = { field: 'value' };
-        const responseText = JSON.stringify(frontmatter);
-        const buffer = Buffer.from('buffer data');
-        mocks.generateContent.mockResolvedValueOnce({
-            text: responseText,
-        });
-        mocks.fileTypeFromBuffer.mockResolvedValueOnce(undefined);
-        const generatedFrontmatter = await pieceFrontMatterFromPrompt(apiKey, schema, prompt, [buffer]);
-        expect(mocks.generateContent).toHaveBeenCalledTimes(1);
-        expect(mocks.generateContent).toHaveBeenCalledWith({
-            contents: expect.arrayContaining([prompt, expect.stringContaining(buffer.toString())]),
-            config: expect.any(Object),
-            model: expect.any(String),
-        });
-        expect(generatedFrontmatter).toEqual(frontmatter);
-    });
-    test('generatePieceFrontmatter with a binary Buffer', async () => {
-        const apiKey = 'apiKey';
-        const schema = makeSchema('books');
-        const prompt = 'prompt';
-        const name = 'file.pdf';
-        const uri = 'gs://another/path/to/file.pdf';
-        const mimeType = 'application/pdf';
-        const frontmatter = { field: 'value' };
-        const responseText = JSON.stringify(frontmatter);
-        const buffer = Buffer.from('buffer data');
-        const blob = new Blob([buffer]);
-        const fileContent = 'fileContent';
-        mocks.generateContent.mockResolvedValueOnce({
-            text: responseText,
-        });
-        mocks.uploadFile.mockResolvedValue({ name, uri, mimeType });
-        mocks.getFile.mockResolvedValue({ uri, mimeType, state: 'ACTIVE' });
-        mocks.fileTypeFromBuffer.mockResolvedValueOnce({ mime: mimeType, ext: 'pdf' });
-        mocks.createPartFromUri.mockReturnValue(fileContent);
-        const generatedFrontmatter = await pieceFrontMatterFromPrompt(apiKey, schema, prompt, [buffer]);
-        expect(mocks.uploadFile).toHaveBeenCalledWith({ file: blob, config: { mimeType } });
-        expect(mocks.getFile).toHaveBeenCalledWith({ name });
-        expect(mocks.generateContent).toHaveBeenCalledTimes(1);
-        expect(mocks.generateContent).toHaveBeenCalledWith({
-            contents: expect.arrayContaining([prompt, fileContent]),
-            config: expect.any(Object),
-            model: expect.any(String),
-        });
-        expect(generatedFrontmatter).toEqual(frontmatter);
-    });
-    test('generatePieceFrontmatter binary file fails to process upload', async () => {
-        const apiKey = 'apiKey';
-        const schema = makeSchema('books');
-        const prompt = 'prompt';
-        const file = '/path/to/file.pdf';
-        const name = 'file.pdf';
-        const mimeType = 'application/pdf';
-        const frontmatter = { field: 'value' };
-        const responseText = JSON.stringify(frontmatter);
-        mocks.generateContent.mockResolvedValueOnce({
-            text: responseText,
-        });
-        mocks.uploadFile.mockResolvedValue({ name, state: 'FAILED' });
-        mocks.getFile.mockResolvedValue({});
-        mocks.fileTypeFromFile.mockResolvedValueOnce({ mime: mimeType, ext: 'pdf' });
-        const generating = pieceFrontMatterFromPrompt(apiKey, schema, prompt, [file]);
-        expect(generating).rejects.toThrowError();
-    });
-    test('generatePieceFrontmatter file fails to extract', async () => {
-        const apiKey = 'apiKey';
-        const schema = makeSchema('books');
-        const prompt = 'prompt';
-        const file = '/path/to/file.pdf';
-        const name = 'file.pdf';
-        const mimeType = 'application/pdf';
-        const frontmatter = { field: 'value' };
-        const responseText = JSON.stringify(frontmatter);
-        mocks.generateContent.mockResolvedValueOnce({
-            text: responseText,
-        });
-        mocks.uploadFile.mockResolvedValue({ name, uri: undefined });
-        mocks.getFile.mockResolvedValue({});
-        mocks.fileTypeFromFile.mockResolvedValueOnce({ mime: mimeType, ext: 'pdf' });
-        const generating = pieceFrontMatterFromPrompt(apiKey, schema, prompt, [file]);
-        expect(generating).rejects.toThrowError();
-    });
-});
-//# sourceMappingURL=google.test.js.map

package/dist/src/llm/google.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"google.test.js","sourceRoot":"","sources":["../../../src/llm/google.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAgB,MAAM,QAAQ,CAAA;AAC5E,OAAO,EAAE,0BAA0B,EAAE,MAAM,aAAa,CAAA;AACxD,OAAO,EACN,WAAW,EAEX,iBAAiB,GAGjB,MAAM,eAAe,CAAA;AACtB,OAAO,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAA;AAChE,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAA;AAItC,wEAAwE;AACxE,MAAM,UAAU,GAAG,CAClB,IAAY,EACuB,EAAE;IACrC,OAAO;QACN,IAAI,EAAE,QAAQ;QACd,KAAK,EAAE,IAAI;QACX,UAAU,EAAE;YACX,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE;YAC9C,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE;SAC5C;QACD,QAAQ,EAAE,CAAC,OAAO,CAAC;QACnB,oBAAoB,EAAE,KAAK;KACoB,CAAA;AACjD,CAAC,CAAA;AAED,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,CAAA;AACvB,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;AACpB,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;AACtB,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAA;IAEtB,MAAM,CAAC,SAAS,CAAC,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAA;IACjC,MAAM,CAAC,SAAS,CAAC,KAAK,GAAG,EAAE,CAAC,EAAE,EAAE,CAAA;IAChC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,eAAe,GAAG,EAAE,CAAC,EAAE,EAAE,CAAA;IACjD,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAA;IACvC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAA;IAEpC,OAAO;QACN,WAAW,EAAE,MAAM;QACnB,YAAY,EAAE,EAAE;QAChB,kBAAkB,EAAE,EAAE;QACtB,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;KAC1B,CAAA;AACF,CAAC,CAAC,CAAA;AAEF,MAAM,KAAK,GAAG;IACb,eAAe,EAAE,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,MAAM,CAAC,eAAe,CAAC;IACxE,UAAU,EAAE,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,KAAK,CAAC,MAAM,CAAC;IACzD,OAAO,EAAE,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC;IACnD,gBAAgB,EAAE,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC;IAC7C,kBAAkB,EAAE,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC;IACjD,iBAAiB,EAAE,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC;IAC/C,QAAQ,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC;CAC7B,CAAA;AAED,MAAM,KAAK,GAAoC,EAAE,CAAA;AAEjD,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IAClC,SAAS,CAAC,GAAG,EAAE;QACd,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;YACrC,IAAI,CAAC,SAAS,EAAE,CAAA;QACjB,CAAC,CAAC,CAAA;QAEF,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAClC,KAAK,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,CAAA;YACxB,OAAO,KAAK,CAAC,GAAG,CAAC,CAAA;QAClB,CAAC,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,MAAM,GAAG,QAAQ,CAAA;QACvB,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,CAAA;QAClC,MAAM,MAAM,GAAG,QAAQ,CAAA;QACvB,MAAM,WAAW,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,CAAA;QACtC,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAA;QAEhD,KAAK,CAAC,eAAe,CAAC,qBAAqB,CAAC;YAC3C,IAAI,EAAE,YAAY;SACS,CAAC,CAAA;QAE7B,MAAM,oBAAoB,GAAG,MAAM,0BAA0B,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;QAErF,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QACtD,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC;YAClD,QAAQ,EAAE,MAAM,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,CAAC;YAC1C,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;YAC1B,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;SACzB,CAAC,CAAA;QACF,MAAM,CAAC,oBAAoB,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAA;IAClD,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,MAAM,GAAG,QAAQ,CAAA;QACvB,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,CAAA;QAClC,MAAM,MAAM,GAAG,QAAQ,CAAA;QACvB,MAAM,WAAW,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,CAAA;QACvE,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAA;QAEhD,KAAK,CAAC,eAAe,CAAC,qBAAqB,CAAC;YAC3C,IAAI,EAAE,YAAY;SACS,CAAC,CAAA;QAE7B,MAAM,oBAAoB,GAAG,MAAM,0BAA0B,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;QAErF,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QACtD,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC;YAClD,QAAQ,EAAE,MAAM,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,CAAC;YAC1C,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;YAC1B,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;SACzB,CAAC,CAAA;QACF,MAAM,CAAC,oBAAoB,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAA;IACzD,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,MAAM,GAAG,QAAQ,CAAA;QACvB,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,CAAA;QAClC,MAAM,MAAM,GAAG,QAAQ,CAAA;QACvB,MAAM,YAAY,GAAG,SAAS,CAAA;QAE9B,KAAK,CAAC,eAAe,CAAC,qBAAqB,CAAC;YAC3C,IAAI,EAAE,YAAY;SACS,CAAC,CAAA;QAE7B,MAAM,oBAAoB,GAAG,MAAM,0BAA0B,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;QAErF,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QACtD,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC;YAClD,QAAQ,EAAE,MAAM,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,CAAC;YAC1C,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;YAC1B,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;SACzB,CAAC,CAAA;QACF,MAAM,CAAC,oBAAoB,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;IACzC,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,MAAM,GAAG,QAAQ,CAAA;QACvB,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,CAAA;QAClC,MAAM,MAAM,GAAG,QAAQ,CAAA;QACvB,MAAM,IAAI,GAAG,mBAAmB,CAAA;QAChC,MAAM,GAAG,GAAG,+BAA+B,CAAA;QAC3C,MAAM,IAAI,GAAG,UAAU,CAAA;QACvB,MAAM,QAAQ,GAAG,iBAAiB,CAAA;QAClC,MAAM,WAAW,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,CAAA;QACtC,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAA;QAChD,MAAM,WAAW,GAAG,aAAqB,CAAA;QAEzC,EAAE,CAAC,aAAa,EAAE,CAAA;QAElB,KAAK,CAAC,eAAe,CAAC,qBAAqB,CAAC;YAC3C,IAAI,EAAE,YAAY;SACS,CAAC,CAAA;QAC7B,KAAK,CAAC,UAAU,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAC,CAAA;QAC3D,KAAK,CAAC,OAAO,CAAC,qBAAqB,CAAC,EAAE,KAAK,EAAE,YAAyB,EAAE,CAAC,CAAA;QACzE,KAAK,CAAC,OAAO,CAAC,qBAAqB,CAAC,EAAE,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAqB,EAAE,CAAC,CAAA;QACpF,KAAK,CAAC,gBAAgB,CAAC,qBAAqB,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAA;QAC5E,KAAK,CAAC,iBAAiB,CAAC,eAAe,CAAC,WAAW,CAAC,CAAA;QAEpD,EAAE,CAAC,iBAAiB,EAAE,CAAA;QAEtB,MAAM,oBAAoB,GAAG,MAAM,0BAA0B,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAA;QAE7F,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAA;QAC7E,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,EAAE,IAAI,EAAE,CAAC,CAAA;QACpD,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QACxD,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QACtD,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC;YAClD,QAAQ,EAAE,MAAM,CAAC,eAAe,CAAC,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YACvD,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;YAC1B,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;SACzB,CAAC,CAAA;QACF,MAAM,CAAC,oBAAoB,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAA;QAEjD,EAAE,CAAC,aAAa,EAAE,CAAA;IACnB,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,MAAM,GAAG,QAAQ,CAAA;QACvB,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,CAAA;QAClC,MAAM,MAAM,GAAG,QAAQ,CAAA;QACvB,MAAM,IAAI,GAAG,oBAAoB,CAAA;QACjC,MAAM,WAAW,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,CAAA;QACtC,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAA;QAChD,MAAM,WAAW,GAAG,aAAa,CAAA;QAEjC,KAAK,CAAC,eAAe,CAAC,qBAAqB,CAAC;YAC3C,IAAI,EAAE,YAAY;SACS,CAAC,CAAA;QAC7B,KAAK,CAAC,gBAAgB,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAA;QACvD,KAAK,CAAC,QAAQ,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAA;QAEjD,MAAM,oBAAoB,GAAG,MAAM,0BAA0B,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAA;QAE7F,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QACtD,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC;YAClD,QAAQ,EAAE,MAAM,CAAC,eAAe,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC,CAAC;YAChF,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;YAC1B,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;SACzB,CAAC,CAAA;QACF,MAAM,CAAC,oBAAoB,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAA;QAEjD,EAAE,CAAC,aAAa,EAAE,CAAA;IACnB,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,MAAM,GAAG,QAAQ,CAAA;QACvB,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,CAAA;QAClC,MAAM,MAAM,GAAG,QAAQ,CAAA;QACvB,MAAM,WAAW,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,CAAA;QACtC,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAA;QAChD,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;QAEzC,KAAK,CAAC,eAAe,CAAC,qBAAqB,CAAC;YAC3C,IAAI,EAAE,YAAY;SACS,CAAC,CAAA;QAC7B,KAAK,CAAC,kBAAkB,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAA;QAEzD,MAAM,oBAAoB,GAAG,MAAM,0BAA0B,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,MAAM,CAAC,CAAC,CAAA;QAE/F,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QACtD,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC;YAClD,QAAQ,EAAE,MAAM,CAAC,eAAe,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;YACtF,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;YAC1B,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;SACzB,CAAC,CAAA;QAEF,MAAM,CAAC,oBAAoB,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAA;IAClD,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,MAAM,GAAG,QAAQ,CAAA;QACvB,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,CAAA;QAClC,MAAM,MAAM,GAAG,QAAQ,CAAA;QACvB,MAAM,IAAI,GAAG,UAAU,CAAA;QACvB,MAAM,GAAG,GAAG,+BAA+B,CAAA;QAC3C,MAAM,QAAQ,GAAG,iBAAiB,CAAA;QAClC,MAAM,WAAW,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,CAAA;QACtC,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAA;QAChD,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;QACzC,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,CAAA;QAC/B,MAAM,WAAW,GAAG,aAAqB,CAAA;QAEzC,KAAK,CAAC,eAAe,CAAC,qBAAqB,CAAC;YAC3C,IAAI,EAAE,YAAY;SACS,CAAC,CAAA;QAC7B,KAAK,CAAC,UAAU,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAC,CAAA;QAC3D,KAAK,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAqB,EAAE,CAAC,CAAA;QAChF,KAAK,CAAC,kBAAkB,CAAC,qBAAqB,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAA;QAC9E,KAAK,CAAC,iBAAiB,CAAC,eAAe,CAAC,WAAW,CAAC,CAAA;QAEpD,MAAM,oBAAoB,GAAG,MAAM,0BAA0B,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,MAAM,CAAC,CAAC,CAAA;QAE/F,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAA;QACnF,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,EAAE,IAAI,EAAE,CAAC,CAAA;QACpD,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QACtD,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC;YAClD,QAAQ,EAAE,MAAM,CAAC,eAAe,CAAC,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YACvD,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;YAC1B,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;SACzB,CAAC,CAAA;QAEF,MAAM,CAAC,oBAAoB,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAA;IAClD,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC/E,MAAM,MAAM,GAAG,QAAQ,CAAA;QACvB,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,CAAA;QAClC,MAAM,MAAM,GAAG,QAAQ,CAAA;QACvB,MAAM,IAAI,GAAG,mBAAmB,CAAA;QAChC,MAAM,IAAI,GAAG,UAAU,CAAA;QACvB,MAAM,QAAQ,GAAG,iBAAiB,CAAA;QAClC,MAAM,WAAW,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,CAAA;QACtC,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAA;QAEhD,KAAK,CAAC,eAAe,CAAC,qBAAqB,CAAC;YAC3C,IAAI,EAAE,YAAY;SACS,CAAC,CAAA;QAC7B,KAAK,CAAC,UAAU,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,QAAqB,EAAE,CAAC,CAAA;QAC1E,KAAK,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAA;QACnC,KAAK,CAAC,gBAAgB,CAAC,qBAAqB,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAA;QAE5E,MAAM,UAAU,GAAG,0BAA0B,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAA;QAE7E,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,YAAY,EAAE,CAAA;IAC1C,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,MAAM,GAAG,QAAQ,CAAA;QACvB,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,CAAA;QAClC,MAAM,MAAM,GAAG,QAAQ,CAAA;QACvB,MAAM,IAAI,GAAG,mBAAmB,CAAA;QAChC,MAAM,IAAI,GAAG,UAAU,CAAA;QACvB,MAAM,QAAQ,GAAG,iBAAiB,CAAA;QAClC,MAAM,WAAW,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,CAAA;QACtC,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAA;QAEhD,KAAK,CAAC,eAAe,CAAC,qBAAqB,CAAC;YAC3C,IAAI,EAAE,YAAY;SACS,CAAC,CAAA;QAC7B,KAAK,CAAC,UAAU,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,CAAA;QAC5D,KAAK,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAA;QACnC,KAAK,CAAC,gBAAgB,CAAC,qBAAqB,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAA;QAE5E,MAAM,UAAU,GAAG,0BAA0B,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAA;QAE7E,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,YAAY,EAAE,CAAA;IAC1C,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA","sourcesContent":["import { describe, expect, vi, afterEach, test, MockInstance } from 'vitest'\nimport { pieceFrontMatterFromPrompt } from './google.js'\nimport {\n\tGoogleGenAI,\n\tPart,\n\tcreatePartFromUri,\n\tGenerateContentResponse,\n\tFileState,\n} from '@google/genai'\nimport { fileTypeFromBuffer, fileTypeFromFile } from 'file-type'\nimport { readFile } from 'fs/promises'\nimport { JSONSchemaType } from 'ajv'\nimport { PieceFrontmatter } from '../pieces/index.js'\n\n// Mock schema created locally to remove dependency on piece.fixtures.ts\nconst makeSchema = (\n\tname: string\n): JSONSchemaType<PieceFrontmatter> => {\n\treturn {\n\t\ttype: 'object',\n\t\ttitle: name,\n\t\tproperties: {\n\t\t\ttitle: { type: 'string', examples: ['title'] },\n\t\t\tkeywords: { type: 'string', nullable: true },\n\t\t},\n\t\trequired: ['title'],\n\t\tadditionalProperties: false,\n\t} as unknown as JSONSchemaType<PieceFrontmatter>\n}\n\nvi.mock('@luzzle/core')\nvi.mock('file-type')\nvi.mock('fs/promises')\nvi.mock('@google/genai', () => {\n\tconst Gemini = vi.fn()\n\n\tGemini.prototype.models = vi.fn()\n\tGemini.prototype.files = vi.fn()\n\tGemini.prototype.models.generateContent = vi.fn()\n\tGemini.prototype.files.upload = vi.fn()\n\tGemini.prototype.files.get = vi.fn()\n\n\treturn {\n\t\tGoogleGenAI: Gemini,\n\t\tHarmCategory: {},\n\t\tHarmBlockThreshold: {},\n\t\tcreatePartFromUri: vi.fn(),\n\t}\n})\n\nconst mocks = {\n\tgenerateContent: vi.mocked(GoogleGenAI.prototype.models.generateContent),\n\tuploadFile: vi.mocked(GoogleGenAI.prototype.files.upload),\n\tgetFile: vi.mocked(GoogleGenAI.prototype.files.get),\n\tfileTypeFromFile: vi.mocked(fileTypeFromFile),\n\tfileTypeFromBuffer: vi.mocked(fileTypeFromBuffer),\n\tcreatePartFromUri: vi.mocked(createPartFromUri),\n\treadFile: vi.mocked(readFile),\n}\n\nconst spies: { [key: string]: MockInstance } = {}\n\ndescribe('lib/llm/google.ts', () => {\n\tafterEach(() => {\n\t\tObject.values(mocks).forEach((mock) => {\n\t\t\tmock.mockReset()\n\t\t})\n\n\t\tObject.keys(spies).forEach((key) => {\n\t\t\tspies[key].mockRestore()\n\t\t\tdelete spies[key]\n\t\t})\n\t})\n\n\ttest('pieceFrontMatterFromPrompt', async () => {\n\t\tconst apiKey = 'apiKey'\n\t\tconst schema = makeSchema('books')\n\t\tconst prompt = 'prompt'\n\t\tconst frontmatter = { field: 'value' }\n\t\tconst responseText = JSON.stringify(frontmatter)\n\n\t\tmocks.generateContent.mockResolvedValueOnce({\n\t\t\ttext: responseText,\n\t\t} as GenerateContentResponse)\n\n\t\tconst generatedFrontmatter = await pieceFrontMatterFromPrompt(apiKey, schema, prompt)\n\n\t\texpect(mocks.generateContent).toHaveBeenCalledTimes(1)\n\t\texpect(mocks.generateContent).toHaveBeenCalledWith({\n\t\t\tcontents: expect.arrayContaining([prompt]),\n\t\t\tconfig: expect.any(Object),\n\t\t\tmodel: expect.any(String),\n\t\t})\n\t\texpect(generatedFrontmatter).toEqual(frontmatter)\n\t})\n\n\ttest('pieceFrontMatterFromPrompt strips empty fields', async () => {\n\t\tconst apiKey = 'apiKey'\n\t\tconst schema = makeSchema('books')\n\t\tconst prompt = 'prompt'\n\t\tconst frontmatter = { field: 'value', field2: null, field3: undefined }\n\t\tconst responseText = JSON.stringify(frontmatter)\n\n\t\tmocks.generateContent.mockResolvedValueOnce({\n\t\t\ttext: responseText,\n\t\t} as GenerateContentResponse)\n\n\t\tconst generatedFrontmatter = await pieceFrontMatterFromPrompt(apiKey, schema, prompt)\n\n\t\texpect(mocks.generateContent).toHaveBeenCalledTimes(1)\n\t\texpect(mocks.generateContent).toHaveBeenCalledWith({\n\t\t\tcontents: expect.arrayContaining([prompt]),\n\t\t\tconfig: expect.any(Object),\n\t\t\tmodel: expect.any(String),\n\t\t})\n\t\texpect(generatedFrontmatter).toEqual({ field: 'value' })\n\t})\n\n\ttest('pieceFrontMatterFromPrompt returns on empty result', async () => {\n\t\tconst apiKey = 'apiKey'\n\t\tconst schema = makeSchema('books')\n\t\tconst prompt = 'prompt'\n\t\tconst responseText = undefined\n\n\t\tmocks.generateContent.mockResolvedValueOnce({\n\t\t\ttext: responseText,\n\t\t} as GenerateContentResponse)\n\n\t\tconst generatedFrontmatter = await pieceFrontMatterFromPrompt(apiKey, schema, prompt)\n\n\t\texpect(mocks.generateContent).toHaveBeenCalledTimes(1)\n\t\texpect(mocks.generateContent).toHaveBeenCalledWith({\n\t\t\tcontents: expect.arrayContaining([prompt]),\n\t\t\tconfig: expect.any(Object),\n\t\t\tmodel: expect.any(String),\n\t\t})\n\t\texpect(generatedFrontmatter).toEqual({})\n\t})\n\n\ttest('generatePieceFrontmatter with a binary file', async () => {\n\t\tconst apiKey = 'apiKey'\n\t\tconst schema = makeSchema('books')\n\t\tconst prompt = 'prompt'\n\t\tconst file = '/path/to/file.pdf'\n\t\tconst uri = 'gs://another/path/to/file.pdf'\n\t\tconst name = 'file.pdf'\n\t\tconst mimeType = 'application/pdf'\n\t\tconst frontmatter = { field: 'value' }\n\t\tconst responseText = JSON.stringify(frontmatter)\n\t\tconst fileContent = 'fileContent' as Part\n\n\t\tvi.useFakeTimers()\n\n\t\tmocks.generateContent.mockResolvedValueOnce({\n\t\t\ttext: responseText,\n\t\t} as GenerateContentResponse)\n\t\tmocks.uploadFile.mockResolvedValue({ name, uri, mimeType })\n\t\tmocks.getFile.mockResolvedValueOnce({ state: 'PROCESSING' as FileState })\n\t\tmocks.getFile.mockResolvedValueOnce({ uri, mimeType, state: 'ACTIVE' as FileState })\n\t\tmocks.fileTypeFromFile.mockResolvedValueOnce({ mime: mimeType, ext: 'pdf' })\n\t\tmocks.createPartFromUri.mockReturnValue(fileContent)\n\n\t\tvi.runAllTimersAsync()\n\n\t\tconst generatedFrontmatter = await pieceFrontMatterFromPrompt(apiKey, schema, prompt, [file])\n\n\t\texpect(mocks.uploadFile).toHaveBeenCalledWith({ file, config: { mimeType } })\n\t\texpect(mocks.getFile).toHaveBeenCalledWith({ name })\n\t\texpect(mocks.createPartFromUri).toHaveBeenCalledTimes(1)\n\t\texpect(mocks.generateContent).toHaveBeenCalledTimes(1)\n\t\texpect(mocks.generateContent).toHaveBeenCalledWith({\n\t\t\tcontents: expect.arrayContaining([prompt, fileContent]),\n\t\t\tconfig: expect.any(Object),\n\t\t\tmodel: expect.any(String),\n\t\t})\n\t\texpect(generatedFrontmatter).toEqual(frontmatter)\n\n\t\tvi.useRealTimers()\n\t})\n\n\ttest('generatePieceFrontmatter with a text file', async () => {\n\t\tconst apiKey = 'apiKey'\n\t\tconst schema = makeSchema('books')\n\t\tconst prompt = 'prompt'\n\t\tconst file = '/path/to/file.html'\n\t\tconst frontmatter = { field: 'value' }\n\t\tconst responseText = JSON.stringify(frontmatter)\n\t\tconst fileContent = 'fileContent'\n\n\t\tmocks.generateContent.mockResolvedValueOnce({\n\t\t\ttext: responseText,\n\t\t} as GenerateContentResponse)\n\t\tmocks.fileTypeFromFile.mockResolvedValueOnce(undefined)\n\t\tmocks.readFile.mockResolvedValueOnce(fileContent)\n\n\t\tconst generatedFrontmatter = await pieceFrontMatterFromPrompt(apiKey, schema, prompt, [file])\n\n\t\texpect(mocks.generateContent).toHaveBeenCalledTimes(1)\n\t\texpect(mocks.generateContent).toHaveBeenCalledWith({\n\t\t\tcontents: expect.arrayContaining([prompt, expect.stringContaining(fileContent)]),\n\t\t\tconfig: expect.any(Object),\n\t\t\tmodel: expect.any(String),\n\t\t})\n\t\texpect(generatedFrontmatter).toEqual(frontmatter)\n\n\t\tvi.useRealTimers()\n\t})\n\n\ttest('generatePieceFrontmatter with a text Buffer', async () => {\n\t\tconst apiKey = 'apiKey'\n\t\tconst schema = makeSchema('books')\n\t\tconst prompt = 'prompt'\n\t\tconst frontmatter = { field: 'value' }\n\t\tconst responseText = JSON.stringify(frontmatter)\n\t\tconst buffer = Buffer.from('buffer data')\n\n\t\tmocks.generateContent.mockResolvedValueOnce({\n\t\t\ttext: responseText,\n\t\t} as GenerateContentResponse)\n\t\tmocks.fileTypeFromBuffer.mockResolvedValueOnce(undefined)\n\n\t\tconst generatedFrontmatter = await pieceFrontMatterFromPrompt(apiKey, schema, prompt, [buffer])\n\n\t\texpect(mocks.generateContent).toHaveBeenCalledTimes(1)\n\t\texpect(mocks.generateContent).toHaveBeenCalledWith({\n\t\t\tcontents: expect.arrayContaining([prompt, expect.stringContaining(buffer.toString())]),\n\t\t\tconfig: expect.any(Object),\n\t\t\tmodel: expect.any(String),\n\t\t})\n\n\t\texpect(generatedFrontmatter).toEqual(frontmatter)\n\t})\n\n\ttest('generatePieceFrontmatter with a binary Buffer', async () => {\n\t\tconst apiKey = 'apiKey'\n\t\tconst schema = makeSchema('books')\n\t\tconst prompt = 'prompt'\n\t\tconst name = 'file.pdf'\n\t\tconst uri = 'gs://another/path/to/file.pdf'\n\t\tconst mimeType = 'application/pdf'\n\t\tconst frontmatter = { field: 'value' }\n\t\tconst responseText = JSON.stringify(frontmatter)\n\t\tconst buffer = Buffer.from('buffer data')\n\t\tconst blob = new Blob([buffer])\n\t\tconst fileContent = 'fileContent' as Part\n\n\t\tmocks.generateContent.mockResolvedValueOnce({\n\t\t\ttext: responseText,\n\t\t} as GenerateContentResponse)\n\t\tmocks.uploadFile.mockResolvedValue({ name, uri, mimeType })\n\t\tmocks.getFile.mockResolvedValue({ uri, mimeType, state: 'ACTIVE' as FileState })\n\t\tmocks.fileTypeFromBuffer.mockResolvedValueOnce({ mime: mimeType, ext: 'pdf' })\n\t\tmocks.createPartFromUri.mockReturnValue(fileContent)\n\n\t\tconst generatedFrontmatter = await pieceFrontMatterFromPrompt(apiKey, schema, prompt, [buffer])\n\n\t\texpect(mocks.uploadFile).toHaveBeenCalledWith({ file: blob, config: { mimeType } })\n\t\texpect(mocks.getFile).toHaveBeenCalledWith({ name })\n\t\texpect(mocks.generateContent).toHaveBeenCalledTimes(1)\n\t\texpect(mocks.generateContent).toHaveBeenCalledWith({\n\t\t\tcontents: expect.arrayContaining([prompt, fileContent]),\n\t\t\tconfig: expect.any(Object),\n\t\t\tmodel: expect.any(String),\n\t\t})\n\n\t\texpect(generatedFrontmatter).toEqual(frontmatter)\n\t})\n\n\ttest('generatePieceFrontmatter binary file fails to process upload', async () => {\n\t\tconst apiKey = 'apiKey'\n\t\tconst schema = makeSchema('books')\n\t\tconst prompt = 'prompt'\n\t\tconst file = '/path/to/file.pdf'\n\t\tconst name = 'file.pdf'\n\t\tconst mimeType = 'application/pdf'\n\t\tconst frontmatter = { field: 'value' }\n\t\tconst responseText = JSON.stringify(frontmatter)\n\n\t\tmocks.generateContent.mockResolvedValueOnce({\n\t\t\ttext: responseText,\n\t\t} as GenerateContentResponse)\n\t\tmocks.uploadFile.mockResolvedValue({ name, state: 'FAILED' as FileState })\n\t\tmocks.getFile.mockResolvedValue({})\n\t\tmocks.fileTypeFromFile.mockResolvedValueOnce({ mime: mimeType, ext: 'pdf' })\n\n\t\tconst generating = pieceFrontMatterFromPrompt(apiKey, schema, prompt, [file])\n\n\t\texpect(generating).rejects.toThrowError()\n\t})\n\n\ttest('generatePieceFrontmatter file fails to extract', async () => {\n\t\tconst apiKey = 'apiKey'\n\t\tconst schema = makeSchema('books')\n\t\tconst prompt = 'prompt'\n\t\tconst file = '/path/to/file.pdf'\n\t\tconst name = 'file.pdf'\n\t\tconst mimeType = 'application/pdf'\n\t\tconst frontmatter = { field: 'value' }\n\t\tconst responseText = JSON.stringify(frontmatter)\n\n\t\tmocks.generateContent.mockResolvedValueOnce({\n\t\t\ttext: responseText,\n\t\t} as GenerateContentResponse)\n\t\tmocks.uploadFile.mockResolvedValue({ name, uri: undefined })\n\t\tmocks.getFile.mockResolvedValue({})\n\t\tmocks.fileTypeFromFile.mockResolvedValueOnce({ mime: mimeType, ext: 'pdf' })\n\n\t\tconst generating = pieceFrontMatterFromPrompt(apiKey, schema, prompt, [file])\n\n\t\texpect(generating).rejects.toThrowError()\n\t})\n})\n"]}

package/dist/src/pieces/Piece.test.d.ts

@@ -1 +0,0 @@
-export {};