diff --git a/story/story_data.yaml b/story/story_data.yaml index 968b91e..90f350d 100644 --- a/story/story_data.yaml +++ b/story/story_data.yaml @@ -10,7 +10,7 @@ prompts: apocalypse: "You walk for two hours and take a break. You've left the town you were in and are now in a more rural area. There's a building to your right and you see " action_verbs: - classic: ["You attack", "You tell", "You use", "You go"] + classic: ["You tell", "You use", "You go", "You"] anything: ["You", "You", "You"] diff --git a/story/story_manager.py b/story/story_manager.py index 2bb8ead..29cdce6 100644 --- a/story/story_manager.py +++ b/story/story_manager.py @@ -108,7 +108,7 @@ class UnconstrainedStoryManager(StoryManager): class ConstrainedStoryManager(StoryManager): - def __init__(self, generator, action_verbs_key="anything"): + def __init__(self, generator, action_verbs_key="classic"): super().__init__(generator) self.action_phrases = get_action_verbs(action_verbs_key) self.cache = False @@ -209,7 +209,7 @@ class CTRLStoryManager(ConstrainedStoryManager): def get_action_results_generate(self): results = [] options = {"word_blacklist": {0:[]}} - options["word_whitelist"] = {0: get_allowed_ctrl_verbs()} + #options["word_whitelist"] = {0: get_allowed_ctrl_verbs()} for phrase in self.action_phrases: result = self.generate_action_result(self.story_context(), phrase, options=options) action_verb = result[0].split()[1]