Prevent entity agents from continuing to talk beyond the end of a dir…

…ect quote. PiperOrigin-RevId: 655915134 Change-Id: I10950fabc11a2bb153de7fb057dac0430b042bbb
google-deepmind · Jul 25, 2024 · abb1375 · abb1375
1 parent bec726c
commit abb1375
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/concordia/components/agent/v2/concat_act_component.py b/concordia/components/agent/v2/concat_act_component.py
@@ -123,6 +123,10 @@ def get_action_attempt(
           call_to_action,
           max_tokens=2200,
           answer_prefix=output,
+          # This terminator protects against the model providing extra context
+          # after the end of a directly spoken response, since it normally
+          # puts a space after a quotation mark only in these cases.
+          terminators=('" ', '\n'),
       )
       self._log(output, prompt)
       return output

diff --git a/concordia/components/agent/v2/legacy_act_component.py b/concordia/components/agent/v2/legacy_act_component.py
@@ -124,6 +124,10 @@ def get_action_attempt(
           call_to_action,
           max_tokens=2200,
           answer_prefix=output,
+          # This terminator protects against the model providing extra context
+          # after the end of a directly spoken response, since it normally
+          # puts a space after a quotation mark only in these cases.
+          terminators=('" ', '\n'),
       )
       self._log(output, prompt)
       return output