aura.js

#!/usr/bin/env node

const server = require('./src/server');
const train = require('./src/train');

var argv = require('yargs/yargs')(process.argv.slice(2))
.usage('Usage: $0 <command> [options]')

// Servidor para receber requisições (e dar respostas com base nos modelos)
.command('serve', 'Inicia o servidor (api REST).', {
    '--port': {
        alias: 'p',
        default: 3000,
        describe: 'Porta do servidor'
    },
    '--engines': {
        alias: 'e',
        default: 'domain:domain:./data/models/domain.model,qna:qna:./data/models/qna.model',
        describe: 'Engines que serão carregadas e disponibilizadas no servidor através de um rota'
    }    
}, function (argv) { server.create(argv) })

// Processo de treinamento de novos modelos
.command('train', 'Treina um modelo em específico.', {
    '--type': {
        alias: 't',
        default: 'domain',
        describe: 'Tipo de modelo a ser treinado. Disponíveis: domain, qna.'
    },
    '--dataset': {
        alias: 'd',
        default: './data/train/aura-domain.tsv',
        describe: 'Arquivo com os dados de treino'
    },
    '--output': {
        alias: 'o',
        default: './data/models/domain.model',
        describe: 'Caminho onde o modelo treinado será escrito.'
    },    
}, function (argv) { train.run(argv) })

// Exemplos de utilização da linha de comando principal
.example('$0 serve -p 3000', 'Inicia o servidor com engines default')
.example('$0 serve -p 3000 -e "domain:my-url/path/to/domain.model"', 'Inicia o servidor com a engine "domain" disponível em /api/my-url usando o modelo "domain.model" (path informado depois do nome da engine).')
.example('$0 train -t domain -d ./data/train/aura-domain.tsv -o ./data/models/domain.model', 'Treina um modelo para a engine domain.')

// Informações gerais sobre o programa
.help('h')
.alias('h', 'help')
.epilog('Aura NLP | github.com/ccuffs/aura-nlp')
.argv;