add parochial universalization agent

PiperOrigin-RevId: 693142788 Change-Id: Iabdb4c3660c0ee85c80889a47ab487d9fe86822e
google-deepmind · Nov 5, 2024 · 3ba3ac8 · 3ba3ac8
1 parent f3cc880
commit 3ba3ac8
Show file tree

Hide file tree

Showing 2 changed files with 627 additions and 0 deletions.
diff --git a/concordia/factory/agent/factories_test.py b/concordia/factory/agent/factories_test.py
@@ -28,6 +28,7 @@
 from concordia.factory.agent import basic_agent_without_plan
 from concordia.factory.agent import observe_recall_prompt_agent
 from concordia.factory.agent import paranoid_agent
+from concordia.factory.agent import parochial_universalization_agent
 from concordia.factory.agent import rational_agent
 from concordia.factory.agent import synthetic_user
 from concordia.language_model import no_language_model
@@ -49,6 +50,7 @@
     'basic_agent_without_plan': basic_agent_without_plan,
     'observe_recall_prompt_agent': observe_recall_prompt_agent,
     'paranoid_agent': paranoid_agent,
+    'parochial_universalization_agent': parochial_universalization_agent,
     'rational_agent': rational_agent,
     'synthetic_user': synthetic_user,
 }
@@ -82,6 +84,11 @@ class AgentFactoriesTest(parameterized.TestCase):
           agent_name='paranoid_agent',
           main_role=True,
       ),
+      dict(
+          testcase_name='parochial_universalization_agent',
+          agent_name='parochial_universalization_agent',
+          main_role=True,
+      ),
       dict(
           testcase_name='rational_agent',
           agent_name='rational_agent',