Updated context variable access methods, update_agent_before_reply pa…

…rameter name changed

Signed-off-by: Mark Sze <mark@sze.family>

autogen/agentchat/contrib/swarm_agent.py

@@ -286,7 +286,7 @@ def __init__(
         human_input_mode: Literal["ALWAYS", "NEVER", "TERMINATE"] = "NEVER",
         description: Optional[str] = None,
         code_execution_config=False,
-        update_state_functions: Optional[Union[List[Callable], Callable]] = None,
+        update_agent_before_reply: Optional[Union[List[Callable], Callable]] = None,
         **kwargs,
     ) -> None:
         super().__init__(
@@ -315,9 +315,9 @@ def __init__(
         # use in the tool execution agent to transfer to the next agent
         self._next_agent = None
 
-        self.register_update_state_functions(update_state_functions)
+        self.register_update_agent_before_reply(update_agent_before_reply)
 
-    def register_update_state_functions(self, functions: Optional[Union[List[Callable], Callable]]):
+    def register_update_agent_before_reply(self, functions: Optional[Union[List[Callable], Callable]]):
         """
         Register functions that will be called when the agent is selected and before it speaks.
         You can add your own validation or precondition functions here.

autogen/agentchat/conversable_agent.py

@@ -530,7 +530,7 @@ def system_message(self) -> str:
         """Return the system message."""
         return self._oai_system_message[0]["content"]
 
-    def get_context(self, key: str, default: Any = None) -> Any:
+    def get_context_value(self, key: str, default: Any = None) -> Any:
         """
         Get a context variable by key.
 
@@ -543,7 +543,7 @@ def get_context(self, key: str, default: Any = None) -> Any:
         """
         return self._context_variables.get(key, default)
 
-    def set_context(self, key: str, value: Any) -> None:
+    def set_context_value(self, key: str, value: Any) -> None:
         """
         Set a context variable.
 
@@ -553,7 +553,7 @@ def set_context(self, key: str, value: Any) -> None:
         """
         self._context_variables[key] = value
 
-    def update_context(self, context_variables: Dict[str, Any]) -> None:
+    def set_context_values(self, context_variables: Dict[str, Any]) -> None:
         """
         Update multiple context variables at once.
 
@@ -562,7 +562,7 @@ def update_context(self, context_variables: Dict[str, Any]) -> None:
         """
         self._context_variables.update(context_variables)
 
-    def pop_context(self, key: str, default: Any = None) -> Any:
+    def pop_context_key(self, key: str, default: Any = None) -> Any:
         """
         Remove and return a context variable.