@pinecone-database/pinecone
Version:
This is the official Node.js SDK for [Pinecone](https://www.pinecone.io), written in TypeScript.
152 lines • 5.88 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
__setModuleDefault(result, mod);
return result;
};
Object.defineProperty(exports, "__esModule", { value: true });
const pinecone_1 = require("../../pinecone");
const fs = __importStar(require("fs"));
const os = __importStar(require("os"));
const path = __importStar(require("path"));
const test_helpers_1 = require("../test-helpers");
let pinecone;
let assistant;
let assistantName;
let tempFile;
let tempFilePath;
let tempFileWithMetadata;
let tempFileWithMetadataPath;
beforeAll(async () => {
if (!process.env.ASSISTANT_NAME) {
throw new Error('ASSISTANT_NAME environment variable is not set');
}
else {
assistantName = process.env.ASSISTANT_NAME;
}
pinecone = new pinecone_1.Pinecone();
assistant = pinecone.Assistant(assistantName);
// Create two temporary test files
const content = 'This is test content for file upload';
// 1: file without metadata
tempFile = `test-upload-${Date.now()}.txt`;
tempFilePath = path.join(os.tmpdir(), tempFile);
try {
fs.writeFileSync(tempFilePath, content);
console.log('File written:', tempFilePath);
}
catch (err) {
console.error('Error writing file:', err);
}
// 2: file with metadata
tempFileWithMetadata = `test-upload-metadata-${Date.now()}.txt`;
tempFileWithMetadataPath = path.join(os.tmpdir(), tempFileWithMetadata);
try {
fs.writeFileSync(tempFileWithMetadataPath, content);
console.log('File written:', tempFileWithMetadataPath);
}
catch (err) {
console.error('Error writing file:', err);
}
// Add a small delay to ensure file system sync
await (0, test_helpers_1.sleep)(5000);
if (!fs.existsSync(tempFilePath)) {
throw new Error(`Temporary file was not created: ${tempFilePath}`);
}
if (!fs.existsSync(tempFileWithMetadataPath)) {
throw new Error(`Temporary file was not created: ${tempFileWithMetadataPath}`);
}
});
afterAll(() => {
// Cleanup: remove temporary test files
if (fs.existsSync(tempFilePath)) {
fs.unlinkSync(tempFilePath);
}
if (fs.existsSync(tempFileWithMetadataPath)) {
fs.unlinkSync(tempFileWithMetadataPath);
}
});
describe('Upload file happy path', () => {
test('Upload file without metadata', async () => {
const response = await assistant.uploadFile({
path: tempFilePath,
});
expect(response).toBeDefined();
expect(response.name).toEqual(tempFile);
expect(response.id).toBeDefined();
expect(response.createdOn).toBeDefined();
expect(response.updatedOn).toBeDefined();
expect(response.status).toBeDefined();
await (0, test_helpers_1.sleep)(10000);
// Delete file happy path test:
(0, test_helpers_1.assertWithRetries)(() => assistant.deleteFile(response.id), () => {
return;
});
});
test('Upload file with metadata', async () => {
const response = await assistant.uploadFile({
path: tempFileWithMetadataPath,
metadata: {
description: 'Test file',
category: 'integration-test',
},
});
expect(response).toBeDefined();
expect(response.name).toEqual(tempFileWithMetadata);
expect(response.id).toBeDefined();
expect(response.createdOn).toBeDefined();
expect(response.updatedOn).toBeDefined();
expect(response.status).toBeDefined();
expect(response.metadata).toBeDefined();
if (response.metadata) {
expect(response.metadata['description']).toEqual('Test file');
expect(response.metadata['category']).toEqual('integration-test');
}
await (0, test_helpers_1.sleep)(10000);
// Delete file happy path test:
(0, test_helpers_1.assertWithRetries)(() => assistant.deleteFile(response.id), () => {
return;
});
});
});
describe('Upload file error paths', () => {
test('Upload with nonexistent file path', async () => {
await expect(assistant.uploadFile({
path: 'nonexistent-file.txt',
})).rejects.toThrow();
});
test('Upload to nonexistent assistant', async () => {
await expect(pinecone.Assistant('nonexistent').uploadFile({
path: tempFileWithMetadataPath,
})).rejects.toThrow(/404/);
});
test('Upload with empty file path', async () => {
await expect(assistant.uploadFile({
path: '',
})).rejects.toThrow('You must pass an object with required properties (`path`) to upload a file.');
});
});
describe('Delete file error paths', () => {
test('Delete non-existent file', async () => {
await expect(assistant.deleteFile('nonexistent-file-id')).rejects.toThrow(/404/);
});
});
//# sourceMappingURL=uploadAndDeleteFile.test.js.map