From 7305a5e9f4f29b5efbec7416e05494b94bc22e63 Mon Sep 17 00:00:00 2001 From: Zhe Su <360307598@qq.com> Date: Sun, 14 Jan 2024 10:06:34 -0500 Subject: [PATCH 1/2] add workaround to fix multiple agent in single round --- sotopia/messages/message_classes.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/sotopia/messages/message_classes.py b/sotopia/messages/message_classes.py index dc07e1be4..1168df6d1 100644 --- a/sotopia/messages/message_classes.py +++ b/sotopia/messages/message_classes.py @@ -327,7 +327,11 @@ def split_by_turn(self, input_string: str) -> list[str]: dialogues[-1] = "\n".join( dialogues[-1].split("\n")[:2] ) # Discard further input in the last turn - # print("Dialogues: ", dialogues) + + for dialogue in dialogues: + # TODO this is current workaround for the issue of multiple agents in one turn + if len(dialogue.split("\n")) >= 3: + raise ValueError("Only one agent can act per turn.") return dialogues @staticmethod From e06fad6062585330dd8c474062c908a5ca726c72 Mon Sep 17 00:00:00 2001 From: Zhe Su <360307598@qq.com> Date: Mon, 15 Jan 2024 23:25:13 -0500 Subject: [PATCH 2/2] fix pre-commit error --- sotopia/messages/message_classes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sotopia/messages/message_classes.py b/sotopia/messages/message_classes.py index 1168df6d1..9176283f3 100644 --- a/sotopia/messages/message_classes.py +++ b/sotopia/messages/message_classes.py @@ -327,7 +327,7 @@ def split_by_turn(self, input_string: str) -> list[str]: dialogues[-1] = "\n".join( dialogues[-1].split("\n")[:2] ) # Discard further input in the last turn - + for dialogue in dialogues: # TODO this is current workaround for the issue of multiple agents in one turn if len(dialogue.split("\n")) >= 3: