@nestbox-ai/cli/dist/commands/document/docSearch.js

The cli tools that helps developers to build agents

"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());
    });
};
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.registerDocSearchCommand = registerDocSearchCommand;
const error_1 = require("../../utils/error");
const chalk_1 = __importDefault(require("chalk"));
const ora_1 = __importDefault(require("ora"));
const project_1 = require("../../utils/project");
const apiUtils_1 = require("./apiUtils");
function registerDocSearchCommand(docCommand) {
    const searchCmd = docCommand
        .command('search')
        .description('Search for documents in a collection')
        .requiredOption('--instance <instanceId>', 'Instance ID')
        .requiredOption('--collection <collectionId>', 'Collection ID')
        .requiredOption('--query <query>', 'Search query')
        .option('--project <projectId>', 'Project ID or name (defaults to the current project)')
        .option('--filter <json>', 'Filter criteria as JSON string');
    searchCmd.action((options) => __awaiter(this, void 0, void 0, function* () {
        yield (0, error_1.withTokenRefresh)(() => __awaiter(this, void 0, void 0, function* () {
            var _a, _b;
            const apis = (0, apiUtils_1.createDocumentApis)();
            const project = yield (0, project_1.resolveProject)(apis.projectsApi, options);
            // Build the request body
            const requestBody = {
                query: options.query,
                params: {},
                filter: {},
                include: ["embedding"]
            };
            // Parse filter JSON if provided
            if (options.filter) {
                try {
                    requestBody.filter = JSON.parse(options.filter);
                }
                catch (e) {
                    console.error(chalk_1.default.red('Error parsing filter JSON:'), e.message);
                    return;
                }
            }
            console.log("REQUEST BODY", requestBody);
            const spinner = (0, ora_1.default)(`Searching for documents in collection "${options.collection}" in instance ${options.instance}...`).start();
            try {
                const response = yield apis.documentsApi.documentControllerSimilaritySearch(project.id, options.instance, options.collection, requestBody);
                const results = response.data;
                spinner.succeed('Successfully retrieved search results');
                console.log(chalk_1.default.blue(`\nSearch results for query "${options.query}" in collection "${options.collection}":\n`));
                console.log(JSON.stringify(results, null, 2));
            }
            catch (error) {
                spinner.fail('Operation failed');
                if (error.response && error.response.status === 401) {
                    console.error(chalk_1.default.red('Authentication token has expired. Please login again using "nestbox login <domain>".'));
                }
                else if ((_b = (_a = error.response) === null || _a === void 0 ? void 0 : _a.data) === null || _b === void 0 ? void 0 : _b.message) {
                    console.error(chalk_1.default.red('API Error:'), error.response.data.message);
                }
                else {
                    console.error(chalk_1.default.red('Error:'), error.message || 'Unknown error');
                }
                throw error;
            }
        }));
    }));
}
//# sourceMappingURL=docSearch.js.map