callanwu commited on
Commit
3cebb1b
·
1 Parent(s): 7b3468b

fix gaia bugs

Browse files
owl/run_gaia_roleplaying.py CHANGED
@@ -52,6 +52,8 @@ def main():
52
  # Create cache directory
53
  cache_dir = "tmp/"
54
  os.makedirs(cache_dir, exist_ok=True)
 
 
55
 
56
  # Create models for different components
57
  models = {
 
52
  # Create cache directory
53
  cache_dir = "tmp/"
54
  os.makedirs(cache_dir, exist_ok=True)
55
+ result_dir = "results/"
56
+ os.makedirs(result_dir, exist_ok=True)
57
 
58
  # Create models for different components
59
  models = {
owl/utils/enhanced_role_playing.py CHANGED
@@ -300,11 +300,6 @@ class OwlGAIARolePlaying(OwlRolePlaying):
300
  ),
301
  )
302
  user_msg = self._reduce_message_options(user_response.msgs)
303
- if (
304
- "n" in self.user_agent.model_config_dict.keys()
305
- and self.user_agent.model_config_dict["n"] > 1
306
- ):
307
- self.user_agent.record_message(user_msg)
308
 
309
  modified_user_msg = deepcopy(user_msg)
310
 
 
300
  ),
301
  )
302
  user_msg = self._reduce_message_options(user_response.msgs)
 
 
 
 
 
303
 
304
  modified_user_msg = deepcopy(user_msg)
305