@lenne.tech/cli/build/commands/qdrant/stats.js

lenne.Tech CLI: lt

"use strict";
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());
    });
};
Object.defineProperty(exports, "__esModule", { value: true });
const command = {
    alias: ['s'],
    description: 'Shows statistics for Qdrant collections',
    name: 'stats',
    run: (toolbox) => __awaiter(void 0, void 0, void 0, function* () {
        var _a, _b, _c, _d, _e;
        const { http, print } = toolbox;
        const qdrantApi = http.create({
            baseURL: 'http://localhost:6333',
            headers: { 'Content-Type': 'application/json' },
        });
        const spinner = print.spin('Fetching Qdrant statistics...');
        // 1. Fetch all collections
        const collectionsResponse = yield qdrantApi.get('/collections');
        if (!collectionsResponse.ok) {
            spinner.fail('Error fetching collections from Qdrant.');
            print.info('Please ensure Qdrant is running on http://localhost:6333');
            return;
        }
        const collections = (_b = (_a = collectionsResponse.data) === null || _a === void 0 ? void 0 : _a.result) === null || _b === void 0 ? void 0 : _b.collections;
        if (!collections || collections.length === 0) {
            spinner.succeed('No collections found in Qdrant.');
            return;
        }
        spinner.succeed('Fetched collection statistics:');
        const tableData = [
            [
                'Collection',
                'Points',
                'Vectors',
                'Segments',
                'RAM Size (MB)',
                'Disk Size (MB)',
            ],
        ];
        for (const collection of collections) {
            const infoResponse = yield qdrantApi.get(`/collections/${collection.name}`);
            if (infoResponse.ok && ((_c = infoResponse.data) === null || _c === void 0 ? void 0 : _c.result)) {
                const result = infoResponse.data.result;
                tableData.push([
                    collection.name,
                    result.points_count.toString(),
                    (_d = result.vectors_count) === null || _d === void 0 ? void 0 : _d.toString(),
                    (_e = result.segments_count) === null || _e === void 0 ? void 0 : _e.toString(),
                    (result.ram_size / 1024 / 1024).toFixed(2),
                    (result.disk_size / 1024 / 1024).toFixed(2),
                ]);
            }
            else {
                tableData.push([collection.name, 'Error', 'Error', 'Error', 'Error', 'Error']);
            }
        }
        print.table(tableData, {
            format: 'lean',
        });
        // For tests
        return 'qdrant stats';
    }),
};
exports.default = command;
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoic3RhdHMuanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi8uLi9zcmMvY29tbWFuZHMvcWRyYW50L3N0YXRzLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7Ozs7Ozs7O0FBMEJBLE1BQU0sT0FBTyxHQUFtQjtJQUM5QixLQUFLLEVBQUUsQ0FBQyxHQUFHLENBQUM7SUFDWixXQUFXLEVBQUUseUNBQXlDO0lBQ3RELElBQUksRUFBRSxPQUFPO0lBQ2IsR0FBRyxFQUFFLENBQU8sT0FBK0IsRUFBRSxFQUFFOztRQUM3QyxNQUFNLEVBQUUsSUFBSSxFQUFFLEtBQUssRUFBRSxHQUFHLE9BQU8sQ0FBQztRQUVoQyxNQUFNLFNBQVMsR0FBRyxJQUFJLENBQUMsTUFBTSxDQUFDO1lBQzVCLE9BQU8sRUFBRSx1QkFBdUI7WUFDaEMsT0FBTyxFQUFFLEVBQUUsY0FBYyxFQUFFLGtCQUFrQixFQUFFO1NBQ2hELENBQUMsQ0FBQztRQUVILE1BQU0sT0FBTyxHQUFHLEtBQUssQ0FBQyxJQUFJLENBQUMsK0JBQStCLENBQUMsQ0FBQztRQUU1RCwyQkFBMkI7UUFDM0IsTUFBTSxtQkFBbUIsR0FBRyxNQUFNLFNBQVMsQ0FBQyxHQUFHLENBQzdDLGNBQWMsQ0FDZixDQUFDO1FBRUYsSUFBSSxDQUFDLG1CQUFtQixDQUFDLEVBQUUsRUFBRSxDQUFDO1lBQzVCLE9BQU8sQ0FBQyxJQUFJLENBQUMseUNBQXlDLENBQUMsQ0FBQztZQUN4RCxLQUFLLENBQUMsSUFBSSxDQUFDLDBEQUEwRCxDQUFDLENBQUM7WUFDdkUsT0FBTztRQUNULENBQUM7UUFFRCxNQUFNLFdBQVcsR0FBRyxNQUFBLE1BQUEsbUJBQW1CLENBQUMsSUFBSSwwQ0FBRSxNQUFNLDBDQUFFLFdBQVcsQ0FBQztRQUNsRSxJQUFJLENBQUMsV0FBVyxJQUFJLFdBQVcsQ0FBQyxNQUFNLEtBQUssQ0FBQyxFQUFFLENBQUM7WUFDN0MsT0FBTyxDQUFDLE9BQU8sQ0FBQyxpQ0FBaUMsQ0FBQyxDQUFDO1lBQ25ELE9BQU87UUFDVCxDQUFDO1FBRUQsT0FBTyxDQUFDLE9BQU8sQ0FBQyxnQ0FBZ0MsQ0FBQyxDQUFDO1FBRWxELE1BQU0sU0FBUyxHQUFHO1lBQ2hCO2dCQUNFLFlBQVk7Z0JBQ1osUUFBUTtnQkFDUixTQUFTO2dCQUNULFVBQVU7Z0JBQ1YsZUFBZTtnQkFDZixnQkFBZ0I7YUFDakI7U0FDRixDQUFDO1FBRUYsS0FBSyxNQUFNLFVBQVUsSUFBSSxXQUFXLEVBQUUsQ0FBQztZQUNyQyxNQUFNLFlBQVksR0FDZCxNQUFNLFNBQVMsQ0FBQyxHQUFHLENBQ25CLGdCQUFnQixVQUFVLENBQUMsSUFBSSxFQUFFLENBQ2xDLENBQUM7WUFFSixJQUFJLFlBQVksQ0FBQyxFQUFFLEtBQUksTUFBQSxZQUFZLENBQUMsSUFBSSwwQ0FBRSxNQUFNLENBQUEsRUFBRSxDQUFDO2dCQUNqRCxNQUFNLE1BQU0sR0FBRyxZQUFZLENBQUMsSUFBSSxDQUFDLE1BQU0sQ0FBQztnQkFDeEMsU0FBUyxDQUFDLElBQUksQ0FBQztvQkFDYixVQUFVLENBQUMsSUFBSTtvQkFDZixNQUFNLENBQUMsWUFBWSxDQUFDLFFBQVEsRUFBRTtvQkFDOUIsTUFBQSxNQUFNLENBQUMsYUFBYSwwQ0FBRSxRQUFRLEVBQUU7b0JBQ2hDLE1BQUEsTUFBTSxDQUFDLGNBQWMsMENBQUUsUUFBUSxFQUFFO29CQUNqQyxDQUFDLE1BQU0sQ0FBQyxRQUFRLEdBQUcsSUFBSSxHQUFHLElBQUksQ0FBQyxDQUFDLE9BQU8sQ0FBQyxDQUFDLENBQUM7b0JBQzFDLENBQUMsTUFBTSxDQUFDLFNBQVMsR0FBRyxJQUFJLEdBQUcsSUFBSSxDQUFDLENBQUMsT0FBTyxDQUFDLENBQUMsQ0FBQztpQkFDNUMsQ0FBQyxDQUFDO1lBQ0wsQ0FBQztpQkFBTSxDQUFDO2dCQUNOLFNBQVMsQ0FBQyxJQUFJLENBQUMsQ0FBQyxVQUFVLENBQUMsSUFBSSxFQUFFLE9BQU8sRUFBRSxPQUFPLEVBQUUsT0FBTyxFQUFFLE9BQU8sRUFBRSxPQUFPLENBQUMsQ0FBQyxDQUFDO1lBQ2pGLENBQUM7UUFDSCxDQUFDO1FBRUQsS0FBSyxDQUFDLEtBQUssQ0FBQyxTQUFTLEVBQUU7WUFDckIsTUFBTSxFQUFFLE1BQU07U0FDZixDQUFDLENBQUM7UUFFSCxZQUFZO1FBQ1osT0FBTyxjQUFjLENBQUM7SUFDeEIsQ0FBQyxDQUFBO0NBQ0YsQ0FBQztBQUVGLGtCQUFlLE9BQU8sQ0FBQyJ9