@pinecone-database/pinecone/dist/inference/__tests__/embed.test.js

This is the official Node.js SDK for [Pinecone](https://www.pinecone.io), written in TypeScript.

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
const embed_1 = require("../embed");
const setupEmbedResponse = (isSuccess) => {
    const fakeEmbed = jest
        .fn()
        .mockImplementation(() => isSuccess ? Promise.resolve({}) : Promise.reject({}));
    const IA = { embed: fakeEmbed };
    return IA;
};
describe('embed', () => {
    test('should format inputs correctly', async () => {
        const model = 'test-model';
        const inputs = ['input1', 'input2'];
        const expectedInputs = [{ text: 'input1' }, { text: 'input2' }];
        const params = { inputType: 'text', truncate: 'END' };
        const IA = setupEmbedResponse(true);
        await (0, embed_1.embed)(IA)(model, inputs, params);
        expect(IA.embed).toHaveBeenCalledWith(expect.objectContaining({
            embedRequest: expect.objectContaining({ inputs: expectedInputs }),
        }));
    });
});
//# sourceMappingURL=embed.test.js.map