diff --git a/owl/run_gaia_roleplaying.py b/owl/run_gaia_roleplaying.py index 5f22aae..cc08cd2 100644 --- a/owl/run_gaia_roleplaying.py +++ b/owl/run_gaia_roleplaying.py @@ -52,6 +52,8 @@ def main(): # Create cache directory cache_dir = "tmp/" os.makedirs(cache_dir, exist_ok=True) + result_dir = "results/" + os.makedirs(result_dir, exist_ok=True) # Create models for different components models = { diff --git a/owl/utils/enhanced_role_playing.py b/owl/utils/enhanced_role_playing.py index df475d3..0cbc2c9 100644 --- a/owl/utils/enhanced_role_playing.py +++ b/owl/utils/enhanced_role_playing.py @@ -300,11 +300,6 @@ class OwlGAIARolePlaying(OwlRolePlaying): ), ) user_msg = self._reduce_message_options(user_response.msgs) - if ( - "n" in self.user_agent.model_config_dict.keys() - and self.user_agent.model_config_dict["n"] > 1 - ): - self.user_agent.record_message(user_msg) modified_user_msg = deepcopy(user_msg)