@future-agi/sdk
Version:
We help GenAI teams maintain high-accuracy for their Models in production.
116 lines • 5.59 kB
JavaScript
;
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const fs_1 = __importDefault(require("fs"));
const path_1 = __importDefault(require("path"));
const kb_1 = require("../../kb");
const types_1 = require("../../api/types");
// Temporary directory for test files
const TMP_DIR = path_1.default.join(__dirname, '__tmp__');
const TEST_FILE_1 = path_1.default.join(TMP_DIR, 'test_kb_1.txt');
const TEST_FILE_2 = path_1.default.join(TMP_DIR, 'test_kb_2.txt');
const TEST_FILE_3 = path_1.default.join(TMP_DIR, 'test_kb_3.txt');
const TEST_KB_NAME = `sdk-test-kb-${Date.now()}`;
const KB_ID = 'kb-id-123';
/** Helper to ensure tmp files exist */
function ensureTmpFiles() {
if (!fs_1.default.existsSync(TMP_DIR)) {
fs_1.default.mkdirSync(TMP_DIR, { recursive: true });
}
[TEST_FILE_1, TEST_FILE_2, TEST_FILE_3].forEach((file) => {
if (!fs_1.default.existsSync(file)) {
fs_1.default.writeFileSync(file, 'dummy content');
}
});
}
/** Cleanup tmp directory */
function cleanupTmpFiles() {
if (fs_1.default.existsSync(TMP_DIR)) {
fs_1.default.readdirSync(TMP_DIR).forEach((file) => fs_1.default.unlinkSync(path_1.default.join(TMP_DIR, file)));
fs_1.default.rmdirSync(TMP_DIR);
}
}
describe('KnowledgeBase SDK – happy path', () => {
let kb;
// Mock the HTTP layer once for all tests
const requestMock = jest.spyOn(kb_1.KnowledgeBase.prototype, 'request');
beforeAll(() => {
ensureTmpFiles();
requestMock.mockImplementation((config) => __awaiter(void 0, void 0, void 0, function* () {
switch (config.method) {
case types_1.HttpMethod.POST: {
// createKb
return {
kbId: KB_ID,
kbName: TEST_KB_NAME,
fileIds: ['file1', 'file3'],
};
}
case types_1.HttpMethod.PATCH: {
// updateKb
return {
id: KB_ID,
name: TEST_KB_NAME,
files: ['file1', 'file3', 'file2'],
};
}
case types_1.HttpMethod.DELETE: {
// deleteFiles or deleteKb → just ack
return {};
}
case types_1.HttpMethod.GET: {
// listKbs / getKbFromName
return {
result: {
tableData: [
{ id: KB_ID, name: TEST_KB_NAME, files: ['file1', 'file3'] },
],
},
};
}
default:
return {};
}
}));
});
afterAll(() => {
requestMock.mockRestore();
// Leave temp files on disk; Node will clean up when jest exits
});
it('creates a knowledge base', () => __awaiter(void 0, void 0, void 0, function* () {
var _a, _b, _c, _d;
kb = new kb_1.KnowledgeBase();
yield kb.createKb(TEST_KB_NAME, [TEST_FILE_1, TEST_FILE_3]);
expect(kb).toBeDefined();
expect((_a = kb.kb) === null || _a === void 0 ? void 0 : _a.id).toBe(KB_ID);
expect((_b = kb.kb) === null || _b === void 0 ? void 0 : _b.name).toBe(TEST_KB_NAME);
expect((_d = (_c = kb.kb) === null || _c === void 0 ? void 0 : _c.files) === null || _d === void 0 ? void 0 : _d.length).toBeGreaterThan(0);
}));
it('updates an existing knowledge base by adding files', () => __awaiter(void 0, void 0, void 0, function* () {
var _a, _b, _c, _d;
const originalFiles = ((_b = (_a = kb.kb) === null || _a === void 0 ? void 0 : _a.files) === null || _b === void 0 ? void 0 : _b.length) || 0;
yield kb.updateKb({ kbName: TEST_KB_NAME, filePaths: [TEST_FILE_2] });
expect((_d = (_c = kb.kb) === null || _c === void 0 ? void 0 : _c.files) === null || _d === void 0 ? void 0 : _d.length).toBeGreaterThan(originalFiles);
}));
it('deletes files from a knowledge base', () => __awaiter(void 0, void 0, void 0, function* () {
yield kb.deleteFilesFromKb({ kbName: TEST_KB_NAME, fileNames: ['file3.txt'] });
expect(requestMock).toHaveBeenLastCalledWith(expect.objectContaining({ method: types_1.HttpMethod.DELETE }), expect.anything());
}));
it('deletes the knowledge base', () => __awaiter(void 0, void 0, void 0, function* () {
var _a;
yield kb.deleteKb({ kbIds: (_a = kb.kb) === null || _a === void 0 ? void 0 : _a.id });
expect(kb.kb).toBeUndefined();
}));
});
//# sourceMappingURL=knowledgeBase.test.js.map