Rename ComponentsContext to ComponentContextMapping.

PiperOrigin-RevId: 648671422
Change-Id: Ib23b6d78ef44112ad0da3128f201208fdb8b05df

concordia/components/agent/v2/simple_act_component.py

@@ -29,7 +29,8 @@ class SimpleActComponent(component_v2.ActingComponent):
   This component will receive the contexts from `pre_act` from all the
   components, and assemble them in the order specified to `__init__`. If the
   component order is not specified, then components will be assembled in the
-  iteration order of the `ComponentsContext` passed to `get_action_attempt`.
+  iteration order of the `ComponentContextMapping` passed to
+  `get_action_attempt`.
   """
 
   def __init__(
@@ -43,13 +44,13 @@ def __init__(
       model: The language model to use for generating the action attempt.
       component_order: The order in which the component contexts will be
         assembled when calling the act component. If None, the contexts will be
-        assembled in the iteration order of the `ComponentsContext` passed to
-        `get_action_attempt`. If the component order is specified, but does not
-        contain all the components passed to `get_action_attempt`, the missing
-        components will be appended at the end in the iteration order of the
-        `ComponentsContext` passed to `get_action_attempt`. The same component
-        cannot appear twice in the component order. All components in the
-        component order must be in the `ComponentsContext` passed to
+        assembled in the iteration order of the `ComponentContextMapping` passed
+        to `get_action_attempt`. If the component order is specified, but does
+        not contain all the components passed to `get_action_attempt`, the
+        missing components will be appended at the end in the iteration order of
+        the `ComponentContextMapping` passed to `get_action_attempt`. The same
+        component cannot appear twice in the component order. All components in
+        the component order must be in the `ComponentContextMapping` passed to
         `get_action_attempt`.
 
     Raises:
@@ -70,7 +71,7 @@ def __init__(
 
   def _context_for_action(
       self,
-      contexts: component_v2.ComponentsContext,
+      contexts: component_v2.ComponentContextMapping,
   ) -> str:
     if self._component_order is None:
       return "\n".join(
@@ -87,7 +88,7 @@ def _context_for_action(
   @override
   def get_action_attempt(
       self,
-      contexts: component_v2.ComponentsContext,
+      contexts: component_v2.ComponentContextMapping,
       action_spec: entity_lib.ActionSpec,
   ) -> str:
     context = self._context_for_action(contexts)

concordia/typing/component_v2.py

@@ -21,7 +21,7 @@
 
 ComponentName = str
 ComponentContext = str
-ComponentsContext = Mapping[ComponentName, ComponentContext]
+ComponentContextMapping = Mapping[ComponentName, ComponentContext]
 
 
 class BaseComponent:
@@ -144,7 +144,7 @@ class ActingComponent(BaseComponent, metaclass=abc.ABCMeta):
   @abc.abstractmethod
   def get_action_attempt(
       self,
-      context: ComponentsContext,
+      context: ComponentContextMapping,
       action_spec: entity_lib.ActionSpec,
   ) -> str:
     """Decides the action of an entity.