ag2 0.9.6__py3-none-any.whl → 0.9.8.post1__py3-none-any.whl

autogen/agentchat/group/context_str.py

@@ -2,7 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-from typing import Optional
 
 from pydantic import BaseModel
 
@@ -20,7 +19,7 @@ class ContextStr(BaseModel):
     """The string to be substituted with context variables. It is expected that the string will contain `{var}` placeholders and that string format will be able to replace all values."""
     template: str
 
-    def format(self, context_variables: ContextVariables) -> Optional[str]:
+    def format(self, context_variables: ContextVariables) -> str | None:
         """Substitute context variables into the string.
 
         Args:
@@ -29,7 +28,6 @@ class ContextStr(BaseModel):
         Returns:
             Optional[str]: The formatted string with context variables substituted.
         """
-
         context = context_variables.to_dict()
 
         if not context:

autogen/agentchat/group/context_variables.py

@@ -2,7 +2,8 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-from typing import Any, Generator, Iterable, Optional
+from collections.abc import Generator, Iterable
+from typing import Any
 
 from pydantic import BaseModel, Field
 
@@ -15,8 +16,7 @@ __CONTEXT_VARIABLES_PARAM_NAME__ = "context_variables"
 
 
 class ContextVariables(BaseModel):
-    """
-    Stores and manages context variables for agentic workflows.
+    """Stores and manages context variables for agentic workflows.
 
     Utilises a dictionary-like interface for setting, getting, and removing variables.
     """
@@ -24,7 +24,7 @@ class ContextVariables(BaseModel):
     # Internal storage for context variables
     data: dict[str, Any] = Field(default_factory=dict)
 
-    def __init__(self, data: Optional[dict[str, Any]] = None, **kwargs: Any) -> None:
+    def __init__(self, data: dict[str, Any] | None = None, **kwargs: Any) -> None:
         """Initialize with data dictionary as an optional positional parameter.
 
         Args:
@@ -34,9 +34,8 @@ class ContextVariables(BaseModel):
         init_data = data or {}
         super().__init__(data=init_data, **kwargs)
 
-    def get(self, key: str, default: Optional[Any] = None) -> Optional[Any]:
-        """
-        Get a value from the context by key.
+    def get(self, key: str, default: Any | None = None) -> Any | None:
+        """Get a value from the context by key.
 
         Args:
             key: The key to retrieve
@@ -48,8 +47,7 @@ class ContextVariables(BaseModel):
         return self.data.get(key, default)
 
     def set(self, key: str, value: Any) -> None:
-        """
-        Set a value in the context by key.
+        """Set a value in the context by key.
 
         Args:
             key: The key to set
@@ -58,8 +56,7 @@ class ContextVariables(BaseModel):
         self.data[key] = value
 
     def remove(self, key: str) -> bool:
-        """
-        Remove a key from the context.
+        """Remove a key from the context.
 
         Args:
             key: The key to remove
@@ -73,8 +70,7 @@ class ContextVariables(BaseModel):
         return False
 
     def keys(self) -> Iterable[str]:
-        """
-        Get all keys in the context.
+        """Get all keys in the context.
 
         Returns:
             An iterable of all keys
@@ -82,8 +78,7 @@ class ContextVariables(BaseModel):
         return self.data.keys()
 
     def values(self) -> Iterable[Any]:
-        """
-        Get all values in the context.
+        """Get all values in the context.
 
         Returns:
             An iterable of all values
@@ -91,8 +86,7 @@ class ContextVariables(BaseModel):
         return self.data.values()
 
     def items(self) -> Iterable[tuple[str, Any]]:
-        """
-        Get all key-value pairs in the context.
+        """Get all key-value pairs in the context.
 
         Returns:
             An iterable of all key-value pairs
@@ -104,8 +98,7 @@ class ContextVariables(BaseModel):
         self.data.clear()
 
     def contains(self, key: str) -> bool:
-        """
-        Check if a key exists in the context.
+        """Check if a key exists in the context.
 
         Args:
             key: The key to check
@@ -116,8 +109,7 @@ class ContextVariables(BaseModel):
         return key in self.data
 
     def update(self, other: dict[str, Any]) -> None:
-        """
-        Update context with key-value pairs from another dictionary.
+        """Update context with key-value pairs from another dictionary.
 
         Args:
             other: Dictionary containing key-value pairs to add
@@ -125,8 +117,7 @@ class ContextVariables(BaseModel):
         self.data.update(other)
 
     def to_dict(self) -> dict[str, Any]:
-        """
-        Convert context variables to a dictionary.
+        """Convert context variables to a dictionary.
 
         Returns:
             Dictionary representation of all context variables
@@ -176,8 +167,7 @@ class ContextVariables(BaseModel):
     # Utility methods
     @classmethod
     def from_dict(cls, data: dict[str, Any]) -> "ContextVariables":
-        """
-        Create a new ContextVariables instance from a dictionary.
+        """Create a new ContextVariables instance from a dictionary.
 
         E.g.:
         my_context = {"user_id": "12345", "settings": {"theme": "dark"}}

autogen/agentchat/group/group_tool_executor.py

@@ -3,8 +3,9 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import inspect
+from collections.abc import Callable
 from copy import deepcopy
-from typing import Annotated, Any, Callable, Optional
+from typing import Annotated, Any
 
 from ...oai import OpenAIWrapper
 from ...tools import Depends, Tool
@@ -30,7 +31,7 @@ class GroupToolExecutor(ConversableAgent):
         )
 
         # Store the next target from a tool call
-        self._group_next_target: Optional[TransitionTarget] = None
+        self._group_next_target: TransitionTarget | None = None
 
         # Primary tool reply function for handling the tool reply and the ReplyResult and TransitionTarget returns
         self.register_reply([Agent, None], self._generate_group_tool_reply, remove_other_reply_funcs=True)
@@ -91,7 +92,6 @@ class GroupToolExecutor(ConversableAgent):
         self, agent: ConversableAgent, current_tool: Tool, context_variables: ContextVariables
     ) -> None:
         """Checks for the context_variables parameter in the tool and updates it to use dependency injection."""
-
         # If the tool has a context_variables parameter, remove the tool and reregister it without the parameter
         if __CONTEXT_VARIABLES_PARAM_NAME__ in current_tool.tool_schema["function"]["parameters"]["properties"]:
             # We'll replace the tool, so start with getting the underlying function
@@ -129,17 +129,17 @@ class GroupToolExecutor(ConversableAgent):
     def _generate_group_tool_reply(
         self,
         agent: ConversableAgent,
-        messages: Optional[list[dict[str, Any]]] = None,
-        sender: Optional[Agent] = None,
-        config: Optional[OpenAIWrapper] = None,
-    ) -> tuple[bool, Optional[dict[str, Any]]]:
+        messages: list[dict[str, Any]] | None = None,
+        sender: Agent | None = None,
+        config: OpenAIWrapper | None = None,
+    ) -> tuple[bool, dict[str, Any] | None]:
         """Pre-processes and generates tool call replies.
 
         This function:
         1. Adds context_variables back to the tool call for the function, if necessary.
         2. Generates the tool calls reply.
-        3. Updates context_variables and next_agent based on the tool call response."""
-
+        3. Updates context_variables and next_agent based on the tool call response.
+        """
         if config is None:
             config = agent  # type: ignore[assignment]
         if messages is None:
@@ -150,7 +150,7 @@ class GroupToolExecutor(ConversableAgent):
             tool_call_count = len(message["tool_calls"])
 
             # Loop through tool calls individually (so context can be updated after each function call)
-            next_target: Optional[TransitionTarget] = None
+            next_target: TransitionTarget | None = None
             tool_responses_inner = []
             contents = []
             for index in range(tool_call_count):

autogen/agentchat/group/group_utils.py

@@ -3,9 +3,10 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import copy
+from collections.abc import Callable
 from functools import partial
 from types import MethodType
-from typing import TYPE_CHECKING, Any, Callable, Optional, Union
+from typing import TYPE_CHECKING, Any, Optional
 
 from ..agent import Agent
 from ..groupchat import GroupChat, GroupChatManager
@@ -86,7 +87,7 @@ def _evaluate_after_works_conditions(
     agent: "ConversableAgent",
     groupchat: GroupChat,
     user_agent: Optional["ConversableAgent"],
-) -> Optional[Union[Agent, str]]:
+) -> Agent | str | None:
     """Evaluate after_works context conditions for an agent.
 
     Args:
@@ -124,10 +125,10 @@ def _evaluate_after_works_conditions(
 
 def _run_oncontextconditions(
     agent: "ConversableAgent",
-    messages: Optional[list[dict[str, Any]]] = None,
-    sender: Optional[Agent] = None,
-    config: Optional[Any] = None,
-) -> tuple[bool, Optional[Union[str, dict[str, Any]]]]:
+    messages: list[dict[str, Any]] | None = None,
+    sender: Agent | None = None,
+    config: Any | None = None,
+) -> tuple[bool, str | dict[str, Any] | None]:
     """Run OnContextConditions for an agent before any other reply function."""
     for on_condition in agent.handoffs.context_conditions:  # type: ignore[attr-defined]
         is_available = (
@@ -267,7 +268,7 @@ def prepare_group_agents(
     tool_execution = GroupToolExecutor()
 
     # Wrap handoff targets in agents that need to be wrapped
-    wrapped_chat_agents: list["ConversableAgent"] = []
+    wrapped_chat_agents: list[ConversableAgent] = []
     for agent in agents:
         wrap_agent_handoff_targets(agent, wrapped_chat_agents)
 
@@ -318,7 +319,7 @@ def wrap_agent_handoff_targets(agent: "ConversableAgent", wrapped_agent_list: li
 
 
 def process_initial_messages(
-    messages: Union[list[dict[str, Any]], str],
+    messages: list[dict[str, Any]] | str,
     user_agent: Optional["ConversableAgent"],
     agents: list["ConversableAgent"],
     wrapped_agents: list["ConversableAgent"],
@@ -346,8 +347,8 @@ def process_initial_messages(
 
     # If there's only one message and there's no identified group agent
     # Start with a user proxy agent, creating one if they haven't passed one in
-    last_agent: Optional[ConversableAgent]
-    temp_user_proxy: Optional[ConversableAgent] = None
+    last_agent: ConversableAgent | None
+    temp_user_proxy: ConversableAgent | None = None
     temp_user_list: list[Agent] = []
     if len(messages) == 1 and "name" not in messages[0] and not user_agent:
         temp_user_proxy = ConversableAgent(name="_User", code_execution_config=False, human_input_mode="ALWAYS")
@@ -425,7 +426,7 @@ def determine_next_agent(
     group_agent_names: list[str],
     user_agent: Optional["ConversableAgent"],
     group_after_work: TransitionTarget,
-) -> Optional[Union[Agent, str]]:
+) -> Agent | str | None:
     """Determine the next agent in the conversation.
 
     Args:
@@ -441,7 +442,6 @@ def determine_next_agent(
     Returns:
         Optional[Union[Agent, str]]: The next agent or speaker selection method.
     """
-
     # Logic for determining the next target (anything based on Transition Target: an agent, wrapped agent, TerminateTarget, StayTarget, RevertToUserTarget, GroupManagerTarget, etc.
     # 1. If it's the first response -> initial agent
     # 2. If the last message is a tool call -> tool execution agent
@@ -513,7 +513,7 @@ def create_group_transition(
     group_agent_names: list[str],
     user_agent: Optional["ConversableAgent"],
     group_after_work: TransitionTarget,
-) -> Callable[["ConversableAgent", GroupChat], Optional[Union[Agent, str]]]:
+) -> Callable[["ConversableAgent", GroupChat], Agent | str | None]:
     """Creates a transition function for group chat with enclosed state for the use_initial_agent.
 
     Args:
@@ -530,7 +530,7 @@ def create_group_transition(
     # of group_transition
     state = {"use_initial_agent": True}
 
-    def group_transition(last_speaker: "ConversableAgent", groupchat: GroupChat) -> Optional[Union[Agent, str]]:
+    def group_transition(last_speaker: "ConversableAgent", groupchat: GroupChat) -> Agent | str | None:
         result = determine_next_agent(
             last_speaker=last_speaker,
             groupchat=groupchat,
@@ -549,7 +549,7 @@ def create_group_transition(
 
 def create_group_manager(
     groupchat: GroupChat,
-    group_manager_args: Optional[dict[str, Any]],
+    group_manager_args: dict[str, Any] | None,
     agents: list["ConversableAgent"],
     group_after_work: TransitionTarget,
 ) -> GroupChatManager:

autogen/agentchat/group/guardrails.py

@@ -5,7 +5,7 @@
 import json
 import re
 from abc import ABC, abstractmethod
-from typing import TYPE_CHECKING, Any, Optional, Union
+from typing import TYPE_CHECKING, Any
 
 from pydantic import BaseModel, Field
 
@@ -46,7 +46,7 @@ class Guardrail(ABC):
     """Abstract base class for guardrails."""
 
     def __init__(
-        self, name: str, condition: str, target: "TransitionTarget", activation_message: Optional[str] = None
+        self, name: str, condition: str, target: "TransitionTarget", activation_message: str | None = None
     ) -> None:
         self.name = name
         self.condition = condition
@@ -58,7 +58,7 @@ class Guardrail(ABC):
     @abstractmethod
     def check(
         self,
-        context: Union[str, list[dict[str, Any]]],
+        context: str | list[dict[str, Any]],
     ) -> GuardrailResult:
         """Checks the text against the guardrail and returns a GuardrailResult.
 
@@ -80,7 +80,7 @@ class LLMGuardrail(Guardrail):
         condition: str,
         target: "TransitionTarget",
         llm_config: "LLMConfig",
-        activation_message: Optional[str] = None,
+        activation_message: str | None = None,
     ) -> None:
         super().__init__(name, condition, target, activation_message)
 
@@ -98,7 +98,7 @@ You will activate the guardrail only if the condition is met.
 
     def check(
         self,
-        context: Union[str, list[dict[str, Any]]],
+        context: str | list[dict[str, Any]],
     ) -> GuardrailResult:
         """Checks the context against the guardrail using an LLM.
 
@@ -131,7 +131,7 @@ class RegexGuardrail(Guardrail):
         name: str,
         condition: str,
         target: "TransitionTarget",
-        activation_message: Optional[str] = None,
+        activation_message: str | None = None,
     ) -> None:
         super().__init__(name, condition, target, activation_message)
         # Compile the regular expression condition
@@ -142,7 +142,7 @@ class RegexGuardrail(Guardrail):
 
     def check(
         self,
-        context: Union[str, list[dict[str, Any]]],
+        context: str | list[dict[str, Any]],
     ) -> GuardrailResult:
         """Checks the context against the guardrail using a regular expression.
 

autogen/agentchat/group/handoffs.py

@@ -2,7 +2,7 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-from typing import Union, overload
+from typing import overload
 
 from pydantic import BaseModel, Field
 
@@ -14,8 +14,7 @@ __all__ = ["Handoffs"]
 
 
 class Handoffs(BaseModel):
-    """
-    Container for all handoff transition conditions of a ConversableAgent.
+    """Container for all handoff transition conditions of a ConversableAgent.
 
     Three types of conditions can be added, each with a different order and time of use:
     1. OnContextConditions (evaluated without an LLM)
@@ -33,8 +32,7 @@ class Handoffs(BaseModel):
     after_works: list[OnContextCondition] = Field(default_factory=list)
 
     def add_context_condition(self, condition: OnContextCondition) -> "Handoffs":
-        """
-        Add a single context condition.
+        """Add a single context condition.
 
         Args:
             condition: The OnContextCondition to add
@@ -50,8 +48,7 @@ class Handoffs(BaseModel):
         return self
 
     def add_context_conditions(self, conditions: list[OnContextCondition]) -> "Handoffs":
-        """
-        Add multiple context conditions.
+        """Add multiple context conditions.
 
         Args:
             conditions: List of OnContextConditions to add
@@ -67,8 +64,7 @@ class Handoffs(BaseModel):
         return self
 
     def add_llm_condition(self, condition: OnCondition) -> "Handoffs":
-        """
-        Add a single LLM condition.
+        """Add a single LLM condition.
 
         Args:
             condition: The OnCondition to add
@@ -84,8 +80,7 @@ class Handoffs(BaseModel):
         return self
 
     def add_llm_conditions(self, conditions: list[OnCondition]) -> "Handoffs":
-        """
-        Add multiple LLM conditions.
+        """Add multiple LLM conditions.
 
         Args:
             conditions: List of OnConditions to add
@@ -101,8 +96,7 @@ class Handoffs(BaseModel):
         return self
 
     def set_after_work(self, target: TransitionTarget) -> "Handoffs":
-        """
-        Set the after work target (replaces all after_works with single entry).
+        """Set the after work target (replaces all after_works with single entry).
 
         For backward compatibility, this creates an OnContextCondition with no condition (always true).
 
@@ -121,8 +115,7 @@ class Handoffs(BaseModel):
         return self
 
     def add_after_work(self, condition: OnContextCondition) -> "Handoffs":
-        """
-        Add a single after-work condition.
+        """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.
@@ -158,8 +151,7 @@ class Handoffs(BaseModel):
         return self
 
     def add_after_works(self, conditions: list[OnContextCondition]) -> "Handoffs":
-        """
-        Add multiple after-work conditions.
+        """Add multiple after-work conditions.
 
         Special handling for condition=None entries:
         - Only one condition=None entry is allowed (the fallback)
@@ -198,9 +190,8 @@ class Handoffs(BaseModel):
     @overload
     def add(self, condition: OnCondition) -> "Handoffs": ...
 
-    def add(self, condition: Union[OnContextCondition, OnCondition]) -> "Handoffs":
-        """
-        Add a single condition (OnContextCondition or OnCondition).
+    def add(self, condition: OnContextCondition | OnCondition) -> "Handoffs":
+        """Add a single condition (OnContextCondition or OnCondition).
 
         Args:
             condition: The condition to add (OnContextCondition or OnCondition)
@@ -220,9 +211,8 @@ class Handoffs(BaseModel):
         else:
             raise TypeError(f"Unsupported condition type: {type(condition).__name__}")
 
-    def add_many(self, conditions: list[Union[OnContextCondition, OnCondition]]) -> "Handoffs":
-        """
-        Add multiple conditions of any supported types (OnContextCondition and OnCondition).
+    def add_many(self, conditions: list[OnContextCondition | OnCondition]) -> "Handoffs":
+        """Add multiple conditions of any supported types (OnContextCondition and OnCondition).
 
         Args:
             conditions: List of conditions to add
@@ -254,8 +244,7 @@ class Handoffs(BaseModel):
         return self
 
     def clear(self) -> "Handoffs":
-        """
-        Clear all handoff conditions.
+        """Clear all handoff conditions.
 
         Returns:
             Self for method chaining
@@ -266,8 +255,7 @@ class Handoffs(BaseModel):
         return self
 
     def get_llm_conditions_by_target_type(self, target_type: type) -> list[OnCondition]:
-        """
-        Get OnConditions for a specific target type.
+        """Get OnConditions for a specific target type.
 
         Args:
             target_type: The type of condition to retrieve
@@ -278,8 +266,7 @@ class Handoffs(BaseModel):
         return [on_condition for on_condition in self.llm_conditions if on_condition.has_target_type(target_type)]
 
     def get_context_conditions_by_target_type(self, target_type: type) -> list[OnContextCondition]:
-        """
-        Get OnContextConditions for a specific target type.
+        """Get OnContextConditions for a specific target type.
 
         Args:
             target_type: The type of condition to retrieve
@@ -294,8 +281,7 @@ class Handoffs(BaseModel):
         ]
 
     def get_llm_conditions_requiring_wrapping(self) -> list[OnCondition]:
-        """
-        Get LLM conditions that have targets that require wrapping.
+        """Get LLM conditions that have targets that require wrapping.
 
         Returns:
             List of LLM conditions that require wrapping
@@ -303,8 +289,7 @@ class Handoffs(BaseModel):
         return [condition for condition in self.llm_conditions if condition.target_requires_wrapping()]
 
     def get_context_conditions_requiring_wrapping(self) -> list[OnContextCondition]:
-        """
-        Get context conditions that have targets that require wrapping.
+        """Get context conditions that have targets that require wrapping.
 
         Returns:
             List of context conditions that require wrapping
@@ -312,9 +297,7 @@ class Handoffs(BaseModel):
         return [condition for condition in self.context_conditions if condition.target_requires_wrapping()]
 
     def set_llm_function_names(self) -> None:
-        """
-        Set the LLM function names for all LLM conditions, creating unique names for each function.
-        """
+        """Set the LLM function names for all LLM conditions, creating unique names for each function."""
         for i, condition in enumerate(self.llm_conditions):
             # Function names are made unique and allow multiple OnCondition's to the same agent
             condition.llm_function_name = f"transfer_to_{condition.target.normalized_name()}_{i + 1}"

autogen/agentchat/group/multi_agent_chat.py

@@ -4,7 +4,7 @@
 
 import asyncio
 import threading
-from typing import TYPE_CHECKING, Any, Union
+from typing import TYPE_CHECKING, Any
 
 from ...doc_utils import export_module
 from ...events.agent_events import ErrorEvent, RunCompletionEvent
@@ -30,7 +30,7 @@ __all__ = [
 @export_module("autogen")
 def initiate_group_chat(
     pattern: "Pattern",
-    messages: Union[list[dict[str, Any]], str],
+    messages: list[dict[str, Any]] | str,
     max_rounds: int = 20,
 ) -> tuple[ChatResult, ContextVariables, "Agent"]:
     """Initialize and run a group chat using a pattern for configuration.
@@ -92,7 +92,7 @@ def initiate_group_chat(
 @export_module("autogen.agentchat")
 async def a_initiate_group_chat(
     pattern: "Pattern",
-    messages: Union[list[dict[str, Any]], str],
+    messages: list[dict[str, Any]] | str,
     max_rounds: int = 20,
 ) -> tuple[ChatResult, ContextVariables, "Agent"]:
     """Initialize and run a group chat using a pattern for configuration, asynchronously.
@@ -154,7 +154,7 @@ async def a_initiate_group_chat(
 @export_module("autogen.agentchat")
 def run_group_chat(
     pattern: "Pattern",
-    messages: Union[list[dict[str, Any]], str],
+    messages: list[dict[str, Any]] | str,
     max_rounds: int = 20,
 ) -> RunResponseProtocol:
     iostream = ThreadIOStream()
@@ -163,7 +163,7 @@ def run_group_chat(
 
     def _initiate_group_chat(
         pattern: "Pattern" = pattern,
-        messages: Union[list[dict[str, Any]], str] = messages,
+        messages: list[dict[str, Any]] | str = messages,
         max_rounds: int = max_rounds,
         iostream: ThreadIOStream = iostream,
         response: RunResponse = response,
@@ -198,7 +198,7 @@ def run_group_chat(
 @export_module("autogen.agentchat")
 async def a_run_group_chat(
     pattern: "Pattern",
-    messages: Union[list[dict[str, Any]], str],
+    messages: list[dict[str, Any]] | str,
     max_rounds: int = 20,
 ) -> AsyncRunResponseProtocol:
     iostream = AsyncThreadIOStream()
@@ -207,7 +207,7 @@ async def a_run_group_chat(
 
     async def _initiate_group_chat(
         pattern: "Pattern" = pattern,
-        messages: Union[list[dict[str, Any]], str] = messages,
+        messages: list[dict[str, Any]] | str = messages,
         max_rounds: int = max_rounds,
         iostream: AsyncThreadIOStream = iostream,
         response: AsyncRunResponse = response,

autogen/agentchat/group/on_condition.py

@@ -2,7 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-from typing import Optional
 
 from pydantic import BaseModel
 
@@ -33,12 +32,11 @@ class OnCondition(BaseModel):  # noqa: N801
 
     target: TransitionTarget
     condition: LLMCondition
-    available: Optional[AvailableCondition] = None
-    llm_function_name: Optional[str] = None
+    available: AvailableCondition | None = None
+    llm_function_name: str | None = None
 
     def has_target_type(self, target_type: type) -> bool:
-        """
-        Check if the target type matches the specified type.
+        """Check if the target type matches the specified type.
 
         Args:
             target_type (type): The target type to check against, which should be a subclass of TransitionTarget
@@ -49,8 +47,7 @@ class OnCondition(BaseModel):  # noqa: N801
         return isinstance(self.target, target_type)
 
     def target_requires_wrapping(self) -> bool:
-        """
-        Check if the target requires wrapping in an agent.
+        """Check if the target requires wrapping in an agent.
 
         Returns:
             bool: True if the target requires wrapping, False otherwise

autogen/agentchat/group/on_context_condition.py

@@ -2,7 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-from typing import Optional
 
 from pydantic import BaseModel
 
@@ -29,12 +28,11 @@ class OnContextCondition(BaseModel):  # noqa: N801
     """
 
     target: TransitionTarget
-    condition: Optional[ContextCondition] = None
-    available: Optional[AvailableCondition] = None
+    condition: ContextCondition | None = None
+    available: AvailableCondition | None = None
 
     def has_target_type(self, target_type: type) -> bool:
-        """
-        Check if the target type matches the specified type.
+        """Check if the target type matches the specified type.
 
         Args:
             target_type (type): The target type to check against. Should be a subclass of TransitionTarget.
@@ -45,8 +43,7 @@ class OnContextCondition(BaseModel):  # noqa: N801
         return isinstance(self.target, target_type)
 
     def target_requires_wrapping(self) -> bool:
-        """
-        Check if the target requires wrapping in an agent.
+        """Check if the target requires wrapping in an agent.
 
         Returns:
             bool: True if the target requires wrapping, False otherwise