fix gaia bugs (#226)
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 |
|