ag2 0.9.2__tar.gz → 0.9.3__tar.gz

{ag2-0.9.2 → ag2-0.9.3}/.devcontainer/devcontainer.json

@@ -30,7 +30,7 @@
         },
         "AZURE_API_VERSION": {
             "description": "This key is required to specify the version of the Azure API you are using. Set this along with AZURE_OPENAI_API_KEY, AZURE_API_ENDPOINT for Azure OpenAI services. These variables can be configured later as environment variables in the codespace terminal."
-        },
+        }
     },
     "shutdownAction": "stopContainer",
     "workspaceFolder": "/workspaces/ag2",
@@ -46,7 +46,7 @@
         "TOGETHER_API_KEY": "${localEnv:TOGETHER_API_KEY}",
         "GEMINI_API_KEY": "${localEnv:GEMINI_API_KEY}",
         "AZURE_API_ENDPOINT": "${localEnv:AZURE_API_ENDPOINT}",
-        "AZURE_API_VERSION": "${localEnv:AZURE_API_VERSION}",
+        "AZURE_API_VERSION": "${localEnv:AZURE_API_VERSION}"
     },
     "features": {
         "ghcr.io/devcontainers/features/common-utils:2": {

{ag2-0.9.2 → ag2-0.9.3}/.devcontainer/python-3.10/devcontainer.json

@@ -30,7 +30,7 @@
         },
         "AZURE_API_VERSION": {
             "description": "This key is required to specify the version of the Azure API you are using. Set this along with AZURE_OPENAI_API_KEY, AZURE_API_ENDPOINT for Azure OpenAI services. These variables can be configured later as environment variables in the codespace terminal."
-        },
+        }
     },
     "shutdownAction": "stopContainer",
     "workspaceFolder": "/workspaces/ag2",
@@ -46,7 +46,7 @@
         "TOGETHER_API_KEY": "${localEnv:TOGETHER_API_KEY}",
         "GEMINI_API_KEY": "${localEnv:GEMINI_API_KEY}",
         "AZURE_API_ENDPOINT": "${localEnv:AZURE_API_ENDPOINT}",
-        "AZURE_API_VERSION": "${localEnv:AZURE_API_VERSION}",
+        "AZURE_API_VERSION": "${localEnv:AZURE_API_VERSION}"
     },
     "features": {
         "ghcr.io/devcontainers/features/common-utils:2": {

{ag2-0.9.2 → ag2-0.9.3}/.devcontainer/python-3.11/devcontainer.json

@@ -30,7 +30,7 @@
         },
         "AZURE_API_VERSION": {
             "description": "This key is required to specify the version of the Azure API you are using. Set this along with AZURE_OPENAI_API_KEY, AZURE_API_ENDPOINT for Azure OpenAI services. These variables can be configured later as environment variables in the codespace terminal."
-        },
+        }
     },
     "shutdownAction": "stopContainer",
     "workspaceFolder": "/workspaces/ag2",
@@ -46,7 +46,7 @@
         "TOGETHER_API_KEY": "${localEnv:TOGETHER_API_KEY}",
         "GEMINI_API_KEY": "${localEnv:GEMINI_API_KEY}",
         "AZURE_API_ENDPOINT": "${localEnv:AZURE_API_ENDPOINT}",
-        "AZURE_API_VERSION": "${localEnv:AZURE_API_VERSION}",
+        "AZURE_API_VERSION": "${localEnv:AZURE_API_VERSION}"
     },
     "features": {
         "ghcr.io/devcontainers/features/common-utils:2": {

{ag2-0.9.2 → ag2-0.9.3}/.devcontainer/python-3.12/devcontainer.json

@@ -30,7 +30,7 @@
         },
         "AZURE_API_VERSION": {
             "description": "This key is required to specify the version of the Azure API you are using. Set this along with AZURE_OPENAI_API_KEY, AZURE_API_ENDPOINT for Azure OpenAI services. These variables can be configured later as environment variables in the codespace terminal."
-        },
+        }
     },
     "shutdownAction": "stopContainer",
     "workspaceFolder": "/workspaces/ag2",
@@ -46,7 +46,7 @@
         "TOGETHER_API_KEY": "${localEnv:TOGETHER_API_KEY}",
         "GEMINI_API_KEY": "${localEnv:GEMINI_API_KEY}",
         "AZURE_API_ENDPOINT": "${localEnv:AZURE_API_ENDPOINT}",
-        "AZURE_API_VERSION": "${localEnv:AZURE_API_VERSION}",
+        "AZURE_API_VERSION": "${localEnv:AZURE_API_VERSION}"
     },
     "features": {
         "ghcr.io/devcontainers/features/common-utils:2": {

{ag2-0.9.2 → ag2-0.9.3}/.devcontainer/python-3.13/devcontainer.json

@@ -30,7 +30,7 @@
         },
         "AZURE_API_VERSION": {
             "description": "This key is required to specify the version of the Azure API you are using. Set this along with AZURE_OPENAI_API_KEY, AZURE_API_ENDPOINT for Azure OpenAI services. These variables can be configured later as environment variables in the codespace terminal."
-        },
+        }
     },
     "shutdownAction": "stopContainer",
     "workspaceFolder": "/workspaces/ag2",
@@ -46,7 +46,7 @@
         "TOGETHER_API_KEY": "${localEnv:TOGETHER_API_KEY}",
         "GEMINI_API_KEY": "${localEnv:GEMINI_API_KEY}",
         "AZURE_API_ENDPOINT": "${localEnv:AZURE_API_ENDPOINT}",
-        "AZURE_API_VERSION": "${localEnv:AZURE_API_VERSION}",
+        "AZURE_API_VERSION": "${localEnv:AZURE_API_VERSION}"
     },
     "features": {
         "ghcr.io/devcontainers/features/common-utils:2": {

{ag2-0.9.2 → ag2-0.9.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ag2
-Version: 0.9.2
+Version: 0.9.3
 Summary: A programming framework for agentic AI
 Project-URL: Homepage, https://ag2.ai/
 Project-URL: Documentation, https://docs.ag2.ai
@@ -72,7 +72,7 @@ Requires-Dist: azure-cosmos>=4.2.0; extra == 'cosmosdb'
 Provides-Extra: crawl4ai
 Requires-Dist: crawl4ai<0.5,>=0.4.247; extra == 'crawl4ai'
 Provides-Extra: deepseek
-Requires-Dist: openai>=1.66.2; extra == 'deepseek'
+Requires-Dist: openai>=1.87.0; extra == 'deepseek'
 Provides-Extra: dev
 Requires-Dist: cairosvg; extra == 'dev'
 Requires-Dist: codespell==2.4.1; extra == 'dev'
@@ -96,7 +96,7 @@ Requires-Dist: mypy==1.15.0; extra == 'dev'
 Requires-Dist: nbclient==0.10.2; extra == 'dev'
 Requires-Dist: nbconvert==7.16.6; extra == 'dev'
 Requires-Dist: nbformat==5.10.4; extra == 'dev'
-Requires-Dist: openai>=1.66.2; extra == 'dev'
+Requires-Dist: openai>=1.87.0; extra == 'dev'
 Requires-Dist: pandas==2.2.3; extra == 'dev'
 Requires-Dist: pdoc3==0.11.6; extra == 'dev'
 Requires-Dist: pillow; extra == 'dev'
@@ -130,6 +130,8 @@ Requires-Dist: pillow; extra == 'docs'
 Requires-Dist: pyyaml==6.0.2; extra == 'docs'
 Requires-Dist: termcolor==3.0.1; extra == 'docs'
 Requires-Dist: typer==0.15.2; extra == 'docs'
+Provides-Extra: duckduckgo
+Requires-Dist: duckduckgo-search>=8.0.2; extra == 'duckduckgo'
 Provides-Extra: flaml
 Requires-Dist: flaml; extra == 'flaml'
 Requires-Dist: numpy<2.0.0,>=1.24.0; (python_version < '3.13') and extra == 'flaml'
@@ -138,14 +140,14 @@ Provides-Extra: gemini
 Requires-Dist: google-api-core; extra == 'gemini'
 Requires-Dist: google-auth; extra == 'gemini'
 Requires-Dist: google-cloud-aiplatform; extra == 'gemini'
-Requires-Dist: google-genai>=1.2.0; extra == 'gemini'
+Requires-Dist: google-genai>=1.20.0; extra == 'gemini'
 Requires-Dist: jsonschema; extra == 'gemini'
 Requires-Dist: pillow; extra == 'gemini'
 Provides-Extra: gemini-realtime
 Requires-Dist: google-api-core; extra == 'gemini-realtime'
 Requires-Dist: google-auth; extra == 'gemini-realtime'
 Requires-Dist: google-cloud-aiplatform; extra == 'gemini-realtime'
-Requires-Dist: google-genai>=1.2.0; extra == 'gemini-realtime'
+Requires-Dist: google-genai>=1.20.0; extra == 'gemini-realtime'
 Requires-Dist: jsonschema; extra == 'gemini-realtime'
 Requires-Dist: pillow; extra == 'gemini-realtime'
 Requires-Dist: websockets<16,>=14.0; extra == 'gemini-realtime'
@@ -198,7 +200,7 @@ Provides-Extra: mathchat
 Requires-Dist: sympy; extra == 'mathchat'
 Requires-Dist: wolframalpha; extra == 'mathchat'
 Provides-Extra: mcp
-Requires-Dist: mcp<1.6,>=1.4.0; (python_version >= '3.10') and extra == 'mcp'
+Requires-Dist: mcp>=1.9.4; (python_version >= '3.10') and extra == 'mcp'
 Provides-Extra: mcp-proxy-gen
 Requires-Dist: fastapi-code-generator>=0.5.4; extra == 'mcp-proxy-gen'
 Requires-Dist: fastapi<1,>=0.112; extra == 'mcp-proxy-gen'
@@ -216,9 +218,9 @@ Provides-Extra: ollama
 Requires-Dist: fix-busted-json>=0.0.18; extra == 'ollama'
 Requires-Dist: ollama>=0.4.7; extra == 'ollama'
 Provides-Extra: openai
-Requires-Dist: openai>=1.66.2; extra == 'openai'
+Requires-Dist: openai>=1.87.0; extra == 'openai'
 Provides-Extra: openai-realtime
-Requires-Dist: openai>=1.66.2; extra == 'openai-realtime'
+Requires-Dist: openai>=1.87.0; extra == 'openai-realtime'
 Requires-Dist: openai[realtime]; extra == 'openai-realtime'
 Provides-Extra: rag
 Requires-Dist: chromadb<1,>=0.5; extra == 'rag'
@@ -282,6 +284,8 @@ Requires-Dist: protobuf==5.29.3; extra == 'retrievechat-qdrant'
 Requires-Dist: pypdf; extra == 'retrievechat-qdrant'
 Requires-Dist: qdrant-client; extra == 'retrievechat-qdrant'
 Requires-Dist: sentence-transformers<=4.1.0; extra == 'retrievechat-qdrant'
+Provides-Extra: tavily
+Requires-Dist: tavily-python>=0.7.4; extra == 'tavily'
 Provides-Extra: teachable
 Requires-Dist: chromadb; extra == 'teachable'
 Provides-Extra: test
@@ -309,7 +313,7 @@ Requires-Dist: mock==5.2.0; extra == 'types'
 Requires-Dist: mypy==1.15.0; extra == 'types'
 Requires-Dist: nbconvert==7.16.6; extra == 'types'
 Requires-Dist: nbformat==5.10.4; extra == 'types'
-Requires-Dist: openai>=1.66.2; extra == 'types'
+Requires-Dist: openai>=1.87.0; extra == 'types'
 Requires-Dist: pandas==2.2.3; extra == 'types'
 Requires-Dist: pytest-asyncio==0.26.0; extra == 'types'
 Requires-Dist: pytest-cov==6.1.1; extra == 'types'
@@ -678,7 +682,7 @@ AG2 supports more advanced concepts to help you build your AI agent workflows. Y
 
 - [Structured Output](https://docs.ag2.ai/latest/docs/user-guide/basic-concepts/structured-outputs)
 - [Ending a conversation](https://docs.ag2.ai/latest/docs/user-guide/advanced-concepts/orchestration/ending-a-chat/)
-- [Retrieval Augmented Generation (RAG)](https://docs.ag2.ai/docs/user-guide/advanced-concepts/rag)
+- [Retrieval Augmented Generation (RAG)](https://docs.ag2.ai/latest/docs/user-guide/advanced-concepts/rag/)
 - [Code Execution](https://docs.ag2.ai/latest/docs/user-guide/advanced-concepts/code-execution)
 - [Tools with Secrets](https://docs.ag2.ai/latest/docs/user-guide/advanced-concepts/tools/tools-with-secrets/)
 

{ag2-0.9.2 → ag2-0.9.3}/README.md

@@ -351,7 +351,7 @@ AG2 supports more advanced concepts to help you build your AI agent workflows. Y
 
 - [Structured Output](https://docs.ag2.ai/latest/docs/user-guide/basic-concepts/structured-outputs)
 - [Ending a conversation](https://docs.ag2.ai/latest/docs/user-guide/advanced-concepts/orchestration/ending-a-chat/)
-- [Retrieval Augmented Generation (RAG)](https://docs.ag2.ai/docs/user-guide/advanced-concepts/rag)
+- [Retrieval Augmented Generation (RAG)](https://docs.ag2.ai/latest/docs/user-guide/advanced-concepts/rag/)
 - [Code Execution](https://docs.ag2.ai/latest/docs/user-guide/advanced-concepts/code-execution)
 - [Tools with Secrets](https://docs.ag2.ai/latest/docs/user-guide/advanced-concepts/tools/tools-with-secrets/)
 

{ag2-0.9.2 → ag2-0.9.3}/autogen/agentchat/contrib/capabilities/transforms.py

@@ -60,15 +60,23 @@ class MessageHistoryLimiter:
     It trims the conversation history by removing older messages, retaining only the most recent messages.
     """
 
-    def __init__(self, max_messages: Optional[int] = None, keep_first_message: bool = False):
+    def __init__(
+        self,
+        max_messages: Optional[int] = None,
+        keep_first_message: bool = False,
+        exclude_names: Optional[list[str]] = None,
+    ):
         """Args:
         max_messages Optional[int]: Maximum number of messages to keep in the context. Must be greater than 0 if not None.
         keep_first_message bool: Whether to keep the original first message in the conversation history.
             Defaults to False.
+        exclude_names Optional[list[str]]: List of message sender names to exclude from the message history.
+            Messages from these senders will be filtered out before applying the message limit. Defaults to None.
         """
         self._validate_max_messages(max_messages)
         self._max_messages = max_messages
         self._keep_first_message = keep_first_message
+        self._exclude_names = exclude_names
 
     def apply_transform(self, messages: list[dict[str, Any]]) -> list[dict[str, Any]]:
         """Truncates the conversation history to the specified maximum number of messages.
@@ -83,25 +91,30 @@ class MessageHistoryLimiter:
         Returns:
             List[Dict]: A new list containing the most recent messages up to the specified maximum.
         """
-        if self._max_messages is None or len(messages) <= self._max_messages:
-            return messages
+
+        exclude_names = getattr(self, "_exclude_names", None)
+
+        filtered = [msg for msg in messages if msg.get("name") not in exclude_names] if exclude_names else messages
+
+        if self._max_messages is None or len(filtered) <= self._max_messages:
+            return filtered
 
         truncated_messages = []
         remaining_count = self._max_messages
 
         # Start with the first message if we need to keep it
-        if self._keep_first_message:
-            truncated_messages = [messages[0]]
+        if self._keep_first_message and filtered:
+            truncated_messages = [filtered[0]]
             remaining_count -= 1
 
         # Loop through messages in reverse
-        for i in range(len(messages) - 1, 0, -1):
+        for i in range(len(filtered) - 1, 0, -1):
             if remaining_count > 1:
-                truncated_messages.insert(1 if self._keep_first_message else 0, messages[i])
+                truncated_messages.insert(1 if self._keep_first_message else 0, filtered[i])
             if remaining_count == 1:  # noqa: SIM102
                 # If there's only 1 slot left and it's a 'tools' message, ignore it.
-                if messages[i].get("role") != "tool":
-                    truncated_messages.insert(1, messages[i])
+                if filtered[i].get("role") != "tool":
+                    truncated_messages.insert(1, filtered[i])
 
             remaining_count -= 1
             if remaining_count == 0:

{ag2-0.9.2 → ag2-0.9.3}/autogen/agentchat/group/group_utils.py

@@ -82,6 +82,46 @@ def link_agents_to_group_manager(agents: list[Agent], group_chat_manager: Agent)
         agent._group_manager = group_chat_manager  # type: ignore[attr-defined]
 
 
+def _evaluate_after_works_conditions(
+    agent: "ConversableAgent",
+    groupchat: GroupChat,
+    user_agent: Optional["ConversableAgent"],
+) -> Optional[Union[Agent, str]]:
+    """Evaluate after_works context conditions for an agent.
+
+    Args:
+        agent: The agent to evaluate after_works conditions for
+        groupchat: The current group chat
+        user_agent: Optional user proxy agent
+
+    Returns:
+        The resolved speaker selection result if a condition matches, None otherwise
+    """
+    if not hasattr(agent, "handoffs") or not agent.handoffs.after_works:  # type: ignore[attr-defined]
+        return None
+
+    for after_work_condition in agent.handoffs.after_works:  # type: ignore[attr-defined]
+        # Check if condition is available
+        is_available = (
+            after_work_condition.available.is_available(agent, groupchat.messages)
+            if after_work_condition.available
+            else True
+        )
+
+        # Evaluate the condition (None condition means always true)
+        if is_available and (
+            after_work_condition.condition is None or after_work_condition.condition.evaluate(agent.context_variables)
+        ):
+            # Condition matched, resolve and return
+            return after_work_condition.target.resolve(
+                groupchat,
+                agent,
+                user_agent,
+            ).get_speaker_selection_result(groupchat)
+
+    return None
+
+
 def _run_oncontextconditions(
     agent: "ConversableAgent",
     messages: Optional[list[dict[str, Any]]] = None,
@@ -94,7 +134,9 @@ def _run_oncontextconditions(
             on_condition.available.is_available(agent, messages if messages else []) if on_condition.available else True
         )
 
-        if is_available and on_condition.condition.evaluate(agent.context_variables):
+        if is_available and (
+            on_condition.condition is None or on_condition.condition.evaluate(agent.context_variables)
+        ):
             # Condition has been met, we'll set the Tool Executor's next target
             # attribute and that will be picked up on the next iteration when
             # _determine_next_agent is called
@@ -161,12 +203,13 @@ def ensure_handoff_agents_in_group(agents: list["ConversableAgent"]) -> None:
                 and context_conditions.target.agent_name not in agent_names
             ):
                 raise ValueError("Agent in OnContextCondition Hand-offs must be in the agents list")
-        if (
-            agent.handoffs.after_work is not None
-            and isinstance(agent.handoffs.after_work, (AgentTarget, AgentNameTarget))
-            and agent.handoffs.after_work.agent_name not in agent_names
-        ):
-            raise ValueError("Agent in after work target Hand-offs must be in the agents list")
+        # Check after_works targets
+        for after_work_condition in agent.handoffs.after_works:
+            if (
+                isinstance(after_work_condition.target, (AgentTarget, AgentNameTarget))
+                and after_work_condition.target.agent_name not in agent_names
+            ):
+                raise ValueError("Agent in after work target Hand-offs must be in the agents list")
 
 
 def prepare_exclude_transit_messages(agents: list["ConversableAgent"]) -> None:
@@ -320,17 +363,19 @@ def setup_context_variables(
     tool_execution: "ConversableAgent",
     agents: list["ConversableAgent"],
     manager: GroupChatManager,
+    user_agent: Optional["ConversableAgent"],
     context_variables: ContextVariables,
 ) -> None:
-    """Assign a common context_variables reference to all agents in the group, including the tool executor and group chat manager.
+    """Assign a common context_variables reference to all agents in the group, including the tool executor, group chat manager, and user proxy agent.
 
     Args:
         tool_execution: The tool execution agent.
         agents: List of all agents in the conversation.
         manager: GroupChatManager instance.
+        user_agent: Optional user proxy agent.
         context_variables: Context variables to assign to all agents.
     """
-    for agent in agents + [tool_execution] + [manager]:
+    for agent in agents + [tool_execution] + [manager] + ([user_agent] if user_agent else []):
         agent.context_variables = context_variables
 
 
@@ -426,22 +471,25 @@ def determine_next_agent(
 
     # If the user last spoke, return to the agent prior to them (if they don't have an after work, otherwise it's treated like any other agent)
     if user_agent and last_speaker == user_agent:
-        if user_agent.handoffs.after_work is None:
+        if not user_agent.handoffs.after_works:
             return last_agent_speaker
         else:
             last_agent_speaker = user_agent
 
     # AFTER WORK:
 
-    # Get the appropriate After Work condition (from the agent if they have one, otherwise the group level one)
-    after_work_condition = (
-        last_agent_speaker.handoffs.after_work  # type: ignore[attr-defined]
-        if last_agent_speaker.handoffs.after_work is not None  # type: ignore[attr-defined]
-        else group_after_work
+    # First, try to evaluate after_works context conditions
+    after_works_result = _evaluate_after_works_conditions(
+        last_agent_speaker,  # type: ignore[arg-type]
+        groupchat,
+        user_agent,
     )
+    if after_works_result is not None:
+        return after_works_result
 
+    # If no after_works conditions matched, use the group-level after_work
     # Resolve the next agent, termination, or speaker selection method
-    resolved_speaker_selection_result = after_work_condition.resolve(
+    resolved_speaker_selection_result = group_after_work.resolve(
         groupchat,
         last_agent_speaker,  # type: ignore[arg-type]
         user_agent,
@@ -525,10 +573,7 @@ def create_group_manager(
                 if (
                     len(agent.handoffs.get_context_conditions_by_target_type(GroupManagerTarget)) > 0
                     or len(agent.handoffs.get_llm_conditions_by_target_type(GroupManagerTarget)) > 0
-                    or (
-                        agent.handoffs.after_work is not None
-                        and isinstance(agent.handoffs.after_work, GroupManagerTarget)
-                    )
+                    or any(isinstance(aw.target, GroupManagerTarget) for aw in agent.handoffs.after_works)
                 ):
                     has_group_manager_target = True
                     break

{ag2-0.9.2 → ag2-0.9.3}/autogen/agentchat/group/handoffs.py

@@ -2,7 +2,7 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-from typing import Optional, Union, overload
+from typing import Union, overload
 
 from pydantic import BaseModel, Field
 
@@ -30,7 +30,7 @@ class Handoffs(BaseModel):
 
     context_conditions: list[OnContextCondition] = Field(default_factory=list)
     llm_conditions: list[OnCondition] = Field(default_factory=list)
-    after_work: Optional[TransitionTarget] = None
+    after_works: list[OnContextCondition] = Field(default_factory=list)
 
     def add_context_condition(self, condition: OnContextCondition) -> "Handoffs":
         """
@@ -102,7 +102,9 @@ class Handoffs(BaseModel):
 
     def set_after_work(self, target: TransitionTarget) -> "Handoffs":
         """
-        Set the after work target (only one allowed).
+        Set the after work target (replaces all after_works with single entry).
+
+        For backward compatibility, this creates an OnContextCondition with no condition (always true).
 
         Args:
             target: The after work TransitionTarget to set
@@ -113,7 +115,81 @@ class Handoffs(BaseModel):
         if not isinstance(target, TransitionTarget):
             raise TypeError(f"Expected a TransitionTarget instance, got {type(target).__name__}")
 
-        self.after_work = target
+        # Create OnContextCondition with no condition (always true)
+        after_work_condition = OnContextCondition(target=target, condition=None)
+        self.after_works = [after_work_condition]
+        return self
+
+    def add_after_work(self, condition: OnContextCondition) -> "Handoffs":
+        """
+        Add a single after-work condition.
+
+        If the condition has condition=None, it will replace any existing
+        condition=None entry and be placed at the end.
+
+        Args:
+            condition: The OnContextCondition to add
+
+        Returns:
+            Self for method chaining
+        """
+        if not isinstance(condition, OnContextCondition):
+            raise TypeError(f"Expected an OnContextCondition instance, got {type(condition).__name__}")
+
+        if condition.condition is None:
+            # Remove any existing condition=None entries
+            self.after_works = [c for c in self.after_works if c.condition is not None]
+            # Add the new one at the end
+            self.after_works.append(condition)
+        else:
+            # For regular conditions, check if we need to move condition=None to the end
+            none_conditions = [c for c in self.after_works if c.condition is None]
+            if none_conditions:
+                # Remove the None condition temporarily
+                self.after_works = [c for c in self.after_works if c.condition is not None]
+                # Add the new regular condition
+                self.after_works.append(condition)
+                # Re-add the None condition at the end
+                self.after_works.append(none_conditions[0])
+            else:
+                # No None condition exists, just append
+                self.after_works.append(condition)
+
+        return self
+
+    def add_after_works(self, conditions: list[OnContextCondition]) -> "Handoffs":
+        """
+        Add multiple after-work conditions.
+
+        Special handling for condition=None entries:
+        - Only one condition=None entry is allowed (the fallback)
+        - It will always be placed at the end of the list
+        - If multiple condition=None entries are provided, only the last one is kept
+
+        Args:
+            conditions: List of OnContextConditions to add
+
+        Returns:
+            Self for method chaining
+        """
+        # Validate that it is a list of OnContextConditions
+        if not all(isinstance(condition, OnContextCondition) for condition in conditions):
+            raise TypeError("All conditions must be of type OnContextCondition")
+
+        # Separate conditions with None and without None
+        none_conditions = [c for c in conditions if c.condition is None]
+        regular_conditions = [c for c in conditions if c.condition is not None]
+
+        # Remove any existing condition=None entries
+        self.after_works = [c for c in self.after_works if c.condition is not None]
+
+        # Add regular conditions
+        self.after_works.extend(regular_conditions)
+
+        # Add at most one None condition at the end
+        if none_conditions:
+            self.after_works.append(none_conditions[-1])  # Use the last one if multiple provided
+
         return self
 
     @overload
@@ -186,7 +262,7 @@ class Handoffs(BaseModel):
         """
         self.context_conditions.clear()
         self.llm_conditions.clear()
-        self.after_work = None
+        self.after_works.clear()
         return self
 
     def get_llm_conditions_by_target_type(self, target_type: type) -> list[OnCondition]:

{ag2-0.9.2 → ag2-0.9.3}/autogen/agentchat/group/on_context_condition.py

@@ -24,12 +24,12 @@ class OnContextCondition(BaseModel):  # noqa: N801
 
     Args:
         target (TransitionTarget): The transition (essentially an agent) to hand off to.
-        condition (ContextCondition): The context variable based condition for transitioning to the target agent.
+        condition (Optional[ContextCondition]): The context variable based condition for transitioning to the target agent. If None, the condition always evaluates to True.
         available (AvailableCondition): Optional condition to determine if this OnCondition is included for the LLM to evaluate based on context variables using classes like StringAvailableCondition and ContextExpressionAvailableCondition.
     """
 
     target: TransitionTarget
-    condition: ContextCondition
+    condition: Optional[ContextCondition] = None
     available: Optional[AvailableCondition] = None
 
     def has_target_type(self, target_type: type) -> bool:

{ag2-0.9.2 → ag2-0.9.3}/autogen/agentchat/group/patterns/pattern.py

@@ -152,7 +152,13 @@ class Pattern(ABC):
         manager = create_group_manager(groupchat, self.group_manager_args, self.agents, self.group_after_work)
 
         # Point all agent's context variables to this function's context_variables
-        setup_context_variables(tool_executor, self.agents, manager, self.context_variables)
+        setup_context_variables(
+            tool_execution=tool_executor,
+            agents=self.agents,
+            manager=manager,
+            user_agent=self.user_agent,
+            context_variables=self.context_variables,
+        )
 
         # Link all agents with the GroupChatManager to allow access to the group chat
         link_agents_to_group_manager(groupchat.agents, manager)

{ag2-0.9.2 → ag2-0.9.3}/autogen/agentchat/groupchat.py

@@ -1489,10 +1489,10 @@ class GroupChatManager(ConversableAgent):
                 for agent in self._groupchat.agents:
                     if agent.name == message["name"]:
                         # An agent`s message is sent to the Group Chat Manager
-                        agent.a_send(message, self, request_reply=False, silent=True)
+                        await agent.a_send(message, self, request_reply=False, silent=True)
                     else:
                         # Otherwise, messages are sent from the Group Chat Manager to the agent
-                        self.a_send(message, agent, request_reply=False, silent=True)
+                        await self.a_send(message, agent, request_reply=False, silent=True)
 
                 # Add previous message to the new groupchat, if it's an admin message the name may not match so add the message directly
                 if message_speaker_agent:

{ag2-0.9.2 → ag2-0.9.3}/autogen/agentchat/realtime/experimental/realtime_swarm.py

@@ -104,7 +104,7 @@ def parse_oai_message(message: Union[dict[str, Any], str], role: str, adressee:
     return oai_message
 
 
-class SwarmableAgent:
+class SwarmableAgent(Agent):
     """A class for an agent that can participate in a swarm chat."""
 
     def __init__(
@@ -239,7 +239,7 @@ class SwarmableAgent:
         sender: Optional["Agent"] = None,
         **kwargs: Any,
     ) -> Union[str, dict[str, Any], None]:
-        raise NotImplementedError
+        return self.generate_reply(messages=messages, sender=sender, **kwargs)
 
     async def a_receive(
         self,
@@ -247,7 +247,7 @@ class SwarmableAgent:
         sender: "Agent",
         request_reply: Optional[bool] = None,
     ) -> None:
-        raise NotImplementedError
+        self.receive(message, sender, request_reply)
 
     async def a_send(
         self,
@@ -255,7 +255,7 @@ class SwarmableAgent:
         recipient: "Agent",
         request_reply: Optional[bool] = None,
     ) -> None:
-        raise NotImplementedError
+        self.send(message, recipient, request_reply)
 
     @property
     def chat_messages(self) -> dict[Agent, list[dict[str, Any]]]:
@@ -293,6 +293,14 @@ class SwarmableAgent:
     def _raise_exception_on_async_reply_functions(self) -> None:
         pass
 
+    def set_ui_tools(self, tools: Optional[list] = None) -> None:
+        """Set UI tools for the agent."""
+        pass
+
+    def unset_ui_tools(self) -> None:
+        """Unset UI tools for the agent."""
+        pass
+
     @staticmethod
     def _last_msg_as_summary(sender: Agent, recipient: Agent, summary_args: Optional[dict[str, Any]]) -> str:
         """Get a chat summary from the last message of the recipient."""