@pinecone-database/pinecone
Version:
This is the official Node.js SDK for [Pinecone](https://www.pinecone.io), written in TypeScript.
68 lines • 2.61 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
const upsertRecords_1 = require("../../vectors/upsertRecords");
const errors_1 = require("../../../errors");
let mockFetch;
const mockHostUrl = 'https://host-url.io';
const mockNamepspace = 'mock-namespace';
const setupCmd = (response) => {
mockFetch = jest.fn().mockResolvedValue({
ok: true,
status: 200,
json: async () => response,
text: async () => JSON.stringify(response),
});
const VectorProvider = {
provideHostUrl: async () => mockHostUrl,
};
const config = {
apiKey: 'api-key-test',
fetchApi: mockFetch,
};
const cmd = new upsertRecords_1.UpsertRecordsCommand(VectorProvider, mockNamepspace, config);
return { cmd, mockFetch };
};
describe('upsertRecords', () => {
beforeEach(() => {
jest.clearAllMocks();
});
test('calls fetch with the upsert records endpoint', async () => {
const { cmd, mockFetch } = setupCmd({});
const records = [
{ id: '1', chunk_text: 'test', category: 'test' },
{ id: '2', chunk_text: 'test2', category: 'test' },
];
await cmd.run(records);
expect(mockFetch).toHaveBeenCalledWith(`${mockHostUrl}/records/namespaces/${mockNamepspace}/upsert`, expect.objectContaining({
method: 'POST',
}));
});
test('converts records to ndjson string', async () => {
const { cmd, mockFetch } = setupCmd({});
const records = [
{ id: '1', chunk_text: 'test', category: 'test' },
{ id: '2', chunk_text: 'test2', category: 'test' },
];
await cmd.run(records);
expect(mockFetch).toHaveBeenCalledWith(`${mockHostUrl}/records/namespaces/${mockNamepspace}/upsert`, expect.objectContaining({
body: records.map((record) => JSON.stringify(record)).join('\n'),
}));
});
test('throws PineconeArgumentError if no _id or id is provided with record', async () => {
const { cmd } = setupCmd({});
const records = [
// @ts-ignore
{ chunk_text: 'test', category: 'test' },
// @ts-ignore
{ chunk_text: 'test2', category: 'test' },
];
try {
await cmd.run(records);
}
catch (err) {
expect(err).toBeInstanceOf(errors_1.PineconeArgumentError);
expect(err.message).toContain('Every record must include an `id` or `_id` property in order to upsert.');
}
});
});
//# sourceMappingURL=upsertRecords.test.js.map