edu.cmu.sv.yoda_environment.YodaEnvironment Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of yoda Show documentation
A library that allows rapid prototyping of dialog systems (language understanding, discourse modelling, dialog management, language generation).
There is a newer version: 0.7.0
Show newest version
package edu.cmu.sv.yoda_environment;

import edu.cmu.sv.database.ActionEnumeration;
import edu.cmu.sv.database.Database;
import edu.cmu.sv.database.Ontology;
import edu.cmu.sv.dialog_management.DialogManager;
import edu.cmu.sv.dialog_management.DialogRegistry;
import edu.cmu.sv.dialog_state_tracking.DialogState;
import edu.cmu.sv.dialog_state_tracking.DialogStateTracker;
import edu.cmu.sv.dialog_state_tracking.Turn;
import edu.cmu.sv.domain.DomainSpec;
import edu.cmu.sv.natural_language_generation.Lexicon;
import edu.cmu.sv.natural_language_generation.NaturalLanguageGenerator;
import edu.cmu.sv.spoken_language_understanding.SpokenLanguageUnderstander;
import edu.cmu.sv.spoken_language_understanding.regex_plus_keyword_understander.RegexPlusKeywordUnderstander;
import edu.cmu.sv.system_action.CommandLineExecutor;
import edu.cmu.sv.system_action.Executor;
import edu.cmu.sv.system_action.JsonExecutor;
import edu.cmu.sv.utils.NBestDistribution;
import org.apache.commons.lang3.tuple.Pair;

import java.util.concurrent.BlockingQueue;
import java.util.concurrent.LinkedBlockingDeque;
import java.util.logging.ConsoleHandler;
import java.util.logging.Handler;
import java.util.logging.Logger;

/**
 * Created by David Cohen on 10/14/14.
 *
 * A YODA Environment contains references to all the modules and databases required for a
 * functioning dialog system.
 *
 */
public class YodaEnvironment {
    // remove console logging from the root log handler
    public static boolean mongoLoggingActive = false;
    static{
        mongoLoggingActive = System.getenv().containsKey("YODA_MONGO_LOG_CONNECTION_STRING");
        if (mongoLoggingActive)
            MongoLogHandler.start();

        // don't log all handlers to stdout
        Logger rootLogger = Logger.getLogger("");
        Handler[] handlers = rootLogger.getHandlers();
        if (handlers[0] instanceof ConsoleHandler) {
            rootLogger.removeHandler(handlers[0]);
        }
    }

    public DialogStateTracker dst;
    public DialogManager dm;
    public Database db;
    public NaturalLanguageGenerator nlg;
    public SpokenLanguageUnderstander slu;
    public Executor exe;
    public OutputHandler out;
    public Lexicon lex;

    // turn + time stamp of DST input
    public BlockingQueue> DstInputQueue = new LinkedBlockingDeque<>();
    // DM input
    public BlockingQueue> DmInputQueue = new LinkedBlockingDeque<>();


    public void loadDomain(DomainSpec domainSpec){
        Ontology.loadOntologyRegistry(domainSpec.getOntologyRegistry());
        lex.loadLexicon(domainSpec.getLexicon());
        DialogRegistry.registerNonDialogTasks(domainSpec.getNonDialogTaskRegistry());
        db.updateOntology();
        db.addDatabaseRegistry(domainSpec.getDatabaseRegistry());
    }

    public static YodaEnvironment dialogTestingEnvironment(){
        ActionEnumeration.enumerationType = ActionEnumeration.ENUMERATION_TYPE.EXHAUSTIVE;
        ActionEnumeration.focusConstraint = ActionEnumeration.FOCUS_CONSTRAINT.IN_FOCUS;
        YodaEnvironment ans = new YodaEnvironment();
        ans.dst = new DialogStateTracker(ans);
        ans.db = new Database(ans);
        ans.dm = new DialogManager(ans);
        ans.nlg = new NaturalLanguageGenerator(ans);
        ans.slu = new RegexPlusKeywordUnderstander(ans);
        ans.exe = new CommandLineExecutor(ans);
        ans.out = new FlushingStandardOutOutputHandler();
        ans.lex = new Lexicon();
        return ans;
    }

    public static YodaEnvironment subProcessDialogEnvironment(){
        ActionEnumeration.enumerationType = ActionEnumeration.ENUMERATION_TYPE.EXHAUSTIVE;
        ActionEnumeration.focusConstraint = ActionEnumeration.FOCUS_CONSTRAINT.IN_FOCUS;
        YodaEnvironment ans = new YodaEnvironment();
        ans.dst = new DialogStateTracker(ans);
        ans.db = new Database(ans);
        ans.dm = new DialogManager(ans);
        ans.nlg = new NaturalLanguageGenerator(ans);
        ans.slu = new RegexPlusKeywordUnderstander(ans);
        ans.exe = new JsonExecutor(ans);
        ans.out = new FlushingStandardOutOutputHandler();
        ans.lex = new Lexicon();
        return ans;
    }

    public static YodaEnvironment languageComponentTrainingEnvironment(){
        ActionEnumeration.enumerationType = ActionEnumeration.ENUMERATION_TYPE.SAMPLED;
        ActionEnumeration.focusConstraint = ActionEnumeration.FOCUS_CONSTRAINT.IN_KB;
        YodaEnvironment ans = new YodaEnvironment();
        ans.db = new Database(ans);
        ans.nlg = new NaturalLanguageGenerator(ans);
        ans.lex = new Lexicon();
        return ans;
    }

}