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

autogen/_website/notebook_processor.py

@@ -15,10 +15,11 @@ import sys
 import tempfile
 import threading
 import time
+from collections.abc import Callable
 from dataclasses import dataclass
 from functools import lru_cache
 from pathlib import Path
-from typing import Any, Callable, Optional, TypeVar, Union
+from typing import Any, TypeVar
 
 from ..import_utils import optional_import_block, require_optional_import
 
@@ -37,7 +38,7 @@ with optional_import_block():
 @dataclass
 class NotebookError:
     error_name: str
-    error_value: Optional[str]
+    error_value: str | None
     traceback: str
     cell_source: str
 
@@ -79,7 +80,6 @@ C = TypeVar("C", bound=Callable[..., Any])
 
 def require_quarto_bin(f: C) -> C:
     """Decorator to skip a function if quarto is not installed."""
-
     if check_quarto_bin():
         return f
     else:
@@ -91,13 +91,13 @@ def require_quarto_bin(f: C) -> C:
         return wrapper  # type: ignore[return-value]
 
 
-def load_metadata(notebook: Path) -> dict[str, dict[str, Union[str, list[str], None]]]:
+def load_metadata(notebook: Path) -> dict[str, dict[str, str | list[str] | None]]:
     content = json.load(notebook.open(encoding="utf-8"))
-    metadata: dict[str, dict[str, Union[str, list[str], None]]] = content.get("metadata", {})
+    metadata: dict[str, dict[str, str | list[str] | None]] = content.get("metadata", {})
     return metadata
 
 
-def skip_reason_or_none_if_ok(notebook: Path) -> Union[str, None, dict[str, Any]]:
+def skip_reason_or_none_if_ok(notebook: Path) -> str | None | dict[str, Any]:
     """Return a reason to skip the notebook, or None if it should not be skipped."""
     if notebook.suffix != ".ipynb":
         return "not a notebook"
@@ -149,7 +149,7 @@ def skip_reason_or_none_if_ok(notebook: Path) -> Union[str, None, dict[str, Any]
     return None
 
 
-def extract_title(notebook: Path) -> Optional[str]:
+def extract_title(notebook: Path) -> str | None:
     """Extract the title of the notebook."""
     with open(notebook, encoding="utf-8") as f:
         content = f.read()
@@ -196,7 +196,7 @@ def fmt_ok(notebook: Path) -> str:
 
 
 @require_optional_import("termcolor", "docs")
-def fmt_error(notebook: Path, error: Union[NotebookError, str]) -> str:
+def fmt_error(notebook: Path, error: NotebookError | str) -> str:
     if isinstance(error, str):
         return f"{colored('[Error]', 'red')} {colored(notebook.name, 'blue')}: {error}"
     elif isinstance(error, NotebookError):
@@ -207,7 +207,7 @@ def fmt_error(notebook: Path, error: Union[NotebookError, str]) -> str:
 
 @require_quarto_bin
 @require_optional_import("nbclient", "docs")
-def test_notebook(notebook_path: Path, timeout: int = 300) -> tuple[Path, Optional[Union[NotebookError, NotebookSkip]]]:
+def test_notebook(notebook_path: Path, timeout: int = 300) -> tuple[Path, NotebookError | NotebookSkip | None]:
     nb = nbformat.read(str(notebook_path), NB_VERSION)  # type: ignore[arg-type,no-untyped-call]
 
     if "skip_test" in nb.metadata:
@@ -240,7 +240,7 @@ def test_notebook(notebook_path: Path, timeout: int = 300) -> tuple[Path, Option
 @require_optional_import("nbclient", "docs")
 def get_timeout_info(
     nb: NotebookNode,
-) -> Optional[NotebookError]:
+) -> NotebookError | None:
     for i, cell in enumerate(nb.cells):
         if cell.cell_type != "code":
             continue
@@ -256,7 +256,7 @@ def get_timeout_info(
 
 
 @require_optional_import("nbclient", "docs")
-def get_error_info(nb: NotebookNode) -> Optional[NotebookError]:
+def get_error_info(nb: NotebookNode) -> NotebookError | None:
     for cell in nb["cells"]:  # get LAST error
         if cell["cell_type"] != "code":
             continue
@@ -288,7 +288,7 @@ def process_notebook(
     quarto_bin: str,
     dry_run: bool,
     target_dir_func: Callable[[Path], Path],
-    post_processor: Optional[Callable[[Path, Path, dict[str, Any], Path], None]] = None,
+    post_processor: Callable[[Path, Path, dict[str, Any], Path], None] | None = None,
 ) -> str:
     """Process a single notebook.
 
@@ -301,7 +301,6 @@ def process_notebook(
         target_dir_func: Function to determine target directory for notebooks
         post_processor: Optional callback for post-processing
     """
-
     in_notebook_dir = "notebook" in src_notebook.parts
 
     metadata = load_metadata(src_notebook)
@@ -400,7 +399,7 @@ def create_base_argument_parser() -> argparse.ArgumentParser:
 
 def process_notebooks_core(
     args: argparse.Namespace,
-    post_process_func: Optional[Callable[[Path, Path, dict[str, Any], Path], None]],
+    post_process_func: Callable[[Path, Path, dict[str, Any], Path], None] | None,
     target_dir_func: Callable[[Path], Path],
 ) -> list[Path]:
     """Core logic for processing notebooks shared across build systems.

autogen/_website/process_notebooks.py

@@ -12,9 +12,9 @@ import json
 import re
 import shutil
 import sys
+from collections.abc import Sequence
 from copy import deepcopy
 from pathlib import Path
-from typing import Sequence, Union
 
 from ..import_utils import optional_import_block, require_optional_import
 from .notebook_processor import (
@@ -41,7 +41,7 @@ def notebooks_target_dir(website_build_directory: Path) -> Path:
 
 
 def add_front_matter_to_metadata_mdx(
-    front_matter: dict[str, Union[str, list[str], None]], website_build_directory: Path, rendered_mdx: Path
+    front_matter: dict[str, str | list[str] | None], website_build_directory: Path, rendered_mdx: Path
 ) -> None:
     source = front_matter.get("source_notebook")
     if isinstance(source, str) and source.startswith("/website/docs/"):
@@ -239,7 +239,7 @@ def extract_img_tag_from_figure_tag(content: str, img_rel_path: Path) -> str:
 def post_process_mdx(
     rendered_mdx: Path,
     source_notebooks: Path,
-    front_matter: dict[str, Union[str, list[str], None]],
+    front_matter: dict[str, str | list[str] | None],
     website_build_directory: Path,
 ) -> None:
     with open(rendered_mdx, encoding="utf-8") as f:
@@ -309,7 +309,7 @@ def post_process_mdx(
     # Convert callout blocks
     content = convert_callout_blocks(content)
 
-    # Convert mdx image syntax to mintly image syntax
+    # Convert mdx image syntax to mintlify image syntax
     content = convert_mdx_image_blocks(content, rendered_mdx, website_build_directory)
 
     # ensure editUrl is present
@@ -340,7 +340,7 @@ def get_sorted_files(input_dir: Path, prefix: str) -> list[str]:
     return [f"{prefix}/{f.parent.relative_to(input_dir)}/index".replace("\\", "/") for f in reversed_files]
 
 
-def generate_nav_group(input_dir: Path, group_header: str, prefix: str) -> dict[str, Union[str, list[str]]]:
+def generate_nav_group(input_dir: Path, group_header: str, prefix: str) -> dict[str, str | list[str]]:
     """Generate navigation group for a directory.
 
     Args:
@@ -381,15 +381,15 @@ def extract_example_group(metadata_path: Path) -> list[str]:
 
 
 def update_group_pages(
-    mint_navigation: list[dict[str, Union[str, list[Union[str, dict[str, Union[str, list[str]]]]]]]],
+    mint_navigation: list[dict[str, str | list[str | dict[str, str | list[str]]]]],
     target_group: str,
-    new_value: Sequence[Union[str, dict[str, Union[str, Sequence[str]]]]],
-) -> list[dict[str, Union[str, list[Union[str, dict[str, Union[str, list[str]]]]]]]]:
+    new_value: Sequence[str | dict[str, str | Sequence[str]]],
+) -> list[dict[str, str | list[str | dict[str, str | list[str]]]]]:
     """Update mint.json navigation group with new pages."""
     nav_copy = deepcopy(mint_navigation)
 
     def update_recursively(
-        items: list[dict[str, Union[str, list[Union[str, dict[str, Union[str, list[str]]]]]]]],
+        items: list[dict[str, str | list[str | dict[str, str | list[str]]]]],
     ) -> None:
         for item in items:
             if isinstance(item, dict):
@@ -572,7 +572,7 @@ def get_files_path_from_navigation(navigation: list[NavigationGroup]) -> list[Pa
     """
     file_paths = []
 
-    def extract_paths(items: Union[Sequence[Union[str, NavigationGroup]]]) -> None:
+    def extract_paths(items: Sequence[str | NavigationGroup]) -> None:
         for item in items:
             if isinstance(item, str):
                 file_paths.append(Path(item))

autogen/_website/utils.py

@@ -9,7 +9,7 @@ import sys
 from datetime import datetime
 from pathlib import Path
 from textwrap import dedent, indent
-from typing import Literal, Optional, TypedDict, Union
+from typing import Literal, TypedDict, Union
 
 from ..import_utils import optional_import_block, require_optional_import
 
@@ -51,7 +51,7 @@ def copy_files(src_dir: Path, dst_dir: Path, files_to_copy: list[Path]) -> None:
             shutil.copy2(file, dst)
 
 
-def copy_only_git_tracked_and_untracked_files(src_dir: Path, dst_dir: Path, ignore_dir: Optional[str] = None) -> None:
+def copy_only_git_tracked_and_untracked_files(src_dir: Path, dst_dir: Path, ignore_dir: str | None = None) -> None:
     """Copy only the files that are tracked by git or newly added from src_dir to dst_dir."""
     tracked_and_new_files = get_git_tracked_and_untracked_files_in_directory(src_dir)
 
@@ -120,6 +120,7 @@ def construct_authors_html(authors_list: list[str], authors_dict: dict[str, dict
         authors_list: list of author identifiers
         authors_dict: Dictionary containing author information keyed by author identifier
         build_system: The build system being used (mkdocs or mintlify)
+
     Returns:
         str: Formatted HTML string containing author cards
     """
@@ -204,6 +205,7 @@ def separate_front_matter_and_content(file_path: Path) -> tuple[str, str]:
 
 def ensure_edit_url(content: str, file_path: Path) -> str:
     """Ensure editUrl is present in the content.
+
     Args:
         content (str): Content of the file
         file_path (Path): Path to the file
@@ -223,7 +225,6 @@ def add_authors_and_social_preview(
     build_system: build_system = "mintlify",
 ) -> None:
     """Add authors info and social share image to mdx files in the target directory."""
-
     social_img_html = (
         """\n<div>
 <img noZoom className="social-share-img"
@@ -302,7 +303,7 @@ def render_gallery_html(gallery_file_path: Path) -> str:
     """
     try:
         # Load gallery items from YAML file
-        with open(gallery_file_path, "r") as file:
+        with open(gallery_file_path) as file:
             gallery_items = yaml.safe_load(file)
 
         # Ensure gallery_items is a list

autogen/agentchat/agent.py

@@ -4,7 +4,7 @@
 #
 # Portions derived from  https://github.com/microsoft/autogen are under the MIT License.
 # SPDX-License-Identifier: MIT
-from typing import TYPE_CHECKING, Any, Optional, Protocol, TypeVar, Union, runtime_checkable
+from typing import TYPE_CHECKING, Any, Optional, Protocol, TypeVar, runtime_checkable
 
 from ..doc_utils import export_module
 
@@ -40,9 +40,9 @@ class Agent(Protocol):
 
     def send(
         self,
-        message: Union[dict[str, Any], str],
+        message: dict[str, Any] | str,
         recipient: "Agent",
-        request_reply: Optional[bool] = None,
+        request_reply: bool | None = None,
     ) -> None:
         """Send a message to another agent.
 
@@ -56,9 +56,9 @@ class Agent(Protocol):
 
     async def a_send(
         self,
-        message: Union[dict[str, Any], str],
+        message: dict[str, Any] | str,
         recipient: "Agent",
-        request_reply: Optional[bool] = None,
+        request_reply: bool | None = None,
     ) -> None:
         """(Async) Send a message to another agent.
 
@@ -72,9 +72,9 @@ class Agent(Protocol):
 
     def receive(
         self,
-        message: Union[dict[str, Any], str],
+        message: dict[str, Any] | str,
         sender: "Agent",
-        request_reply: Optional[bool] = None,
+        request_reply: bool | None = None,
     ) -> None:
         """Receive a message from another agent.
 
@@ -87,9 +87,9 @@ class Agent(Protocol):
 
     async def a_receive(
         self,
-        message: Union[dict[str, Any], str],
+        message: dict[str, Any] | str,
         sender: "Agent",
-        request_reply: Optional[bool] = None,
+        request_reply: bool | None = None,
     ) -> None:
         """(Async) Receive a message from another agent.
 
@@ -103,10 +103,10 @@ class Agent(Protocol):
 
     def generate_reply(
         self,
-        messages: Optional[list[dict[str, Any]]] = None,
+        messages: list[dict[str, Any]] | None = None,
         sender: Optional["Agent"] = None,
         **kwargs: Any,
-    ) -> Union[str, dict[str, Any], None]:
+    ) -> str | dict[str, Any] | None:
         """Generate a reply based on the received messages.
 
         Args:
@@ -122,10 +122,10 @@ class Agent(Protocol):
 
     async def a_generate_reply(
         self,
-        messages: Optional[list[dict[str, Any]]] = None,
+        messages: list[dict[str, Any]] | None = None,
         sender: Optional["Agent"] = None,
         **kwargs: Any,
-    ) -> Union[str, dict[str, Any], None]:
+    ) -> str | dict[str, Any] | None:
         """(Async) Generate a reply based on the received messages.
 
         Args:

autogen/agentchat/assistant_agent.py

@@ -4,7 +4,8 @@
 #
 # Portions derived from  https://github.com/microsoft/autogen are under the MIT License.
 # SPDX-License-Identifier: MIT
-from typing import Any, Callable, Literal, Optional, Union
+from collections.abc import Callable
+from typing import Any, Literal
 
 from ..doc_utils import export_module
 from ..llm_config import LLMConfig
@@ -42,12 +43,12 @@ Reply "TERMINATE" in the end when everything is done.
     def __init__(
         self,
         name: str,
-        system_message: Optional[str] = DEFAULT_SYSTEM_MESSAGE,
-        llm_config: Optional[Union[LLMConfig, dict[str, Any], Literal[False]]] = None,
-        is_termination_msg: Optional[Callable[[dict[str, Any]], bool]] = None,
-        max_consecutive_auto_reply: Optional[int] = None,
+        system_message: str | None = DEFAULT_SYSTEM_MESSAGE,
+        llm_config: LLMConfig | dict[str, Any] | Literal[False] | None = None,
+        is_termination_msg: Callable[[dict[str, Any]], bool] | None = None,
+        max_consecutive_auto_reply: int | None = None,
         human_input_mode: Literal["ALWAYS", "NEVER", "TERMINATE"] = "NEVER",
-        description: Optional[str] = None,
+        description: str | None = None,
         **kwargs: Any,
     ):
         """Args:

autogen/agentchat/contrib/agent_eval/agent_eval.py

@@ -4,7 +4,7 @@
 #
 # Portions derived from  https://github.com/microsoft/autogen are under the MIT License.
 # SPDX-License-Identifier: MIT
-from typing import Any, Literal, Optional, Union
+from typing import Any, Literal
 
 from .... import GroupChat, GroupChatManager, UserProxyAgent
 from ....llm_config import LLMConfig
@@ -16,7 +16,7 @@ from .task import Task
 
 
 def generate_criteria(
-    llm_config: Optional[Union[LLMConfig, dict[str, Any], Literal[False]]] = None,
+    llm_config: LLMConfig | dict[str, Any] | Literal[False] | None = None,
     task: Task = None,
     additional_instructions: str = "",
     max_round=2,
@@ -67,7 +67,7 @@ def generate_criteria(
 
 
 def quantify_criteria(
-    llm_config: Optional[Union[LLMConfig, dict[str, Any], Literal[False]]] = None,
+    llm_config: LLMConfig | dict[str, Any] | Literal[False] | None = None,
     criteria: list[Criterion] = None,
     task: Task = None,
     test_case: str = "",

autogen/agentchat/contrib/agent_eval/critic_agent.py

@@ -4,7 +4,7 @@
 #
 # Portions derived from  https://github.com/microsoft/autogen are under the MIT License.
 # SPDX-License-Identifier: MIT
-from typing import Any, Optional
+from typing import Any
 
 from ...conversable_agent import ConversableAgent
 
@@ -24,8 +24,8 @@ class CriticAgent(ConversableAgent):
     def __init__(
         self,
         name="critic",
-        system_message: Optional[str] = DEFAULT_SYSTEM_MESSAGE,
-        description: Optional[str] = DEFAULT_DESCRIPTION,
+        system_message: str | None = DEFAULT_SYSTEM_MESSAGE,
+        description: str | None = DEFAULT_DESCRIPTION,
         **kwargs: Any,
     ):
         """Args:

autogen/agentchat/contrib/agent_eval/quantifier_agent.py

@@ -4,7 +4,7 @@
 #
 # Portions derived from  https://github.com/microsoft/autogen are under the MIT License.
 # SPDX-License-Identifier: MIT
-from typing import Any, Optional
+from typing import Any
 
 from ...conversable_agent import ConversableAgent
 
@@ -24,8 +24,8 @@ class QuantifierAgent(ConversableAgent):
     def __init__(
         self,
         name="quantifier",
-        system_message: Optional[str] = DEFAULT_SYSTEM_MESSAGE,
-        description: Optional[str] = DEFAULT_DESCRIPTION,
+        system_message: str | None = DEFAULT_SYSTEM_MESSAGE,
+        description: str | None = DEFAULT_DESCRIPTION,
         **kwargs: Any,
     ):
         """Args:

autogen/agentchat/contrib/agent_eval/subcritic_agent.py

@@ -4,7 +4,7 @@
 #
 # Portions derived from  https://github.com/microsoft/autogen are under the MIT License.
 # SPDX-License-Identifier: MIT
-from typing import Any, Optional
+from typing import Any
 
 from ...conversable_agent import ConversableAgent
 
@@ -25,8 +25,8 @@ class SubCriticAgent(ConversableAgent):
     def __init__(
         self,
         name="subcritic",
-        system_message: Optional[str] = DEFAULT_SYSTEM_MESSAGE,
-        description: Optional[str] = DEFAULT_DESCRIPTION,
+        system_message: str | None = DEFAULT_SYSTEM_MESSAGE,
+        description: str | None = DEFAULT_DESCRIPTION,
         **kwargs: Any,
     ):
         """Args:

autogen/agentchat/contrib/agent_optimizer.py

@@ -6,7 +6,7 @@
 # SPDX-License-Identifier: MIT
 import copy
 import json
-from typing import Any, Optional, Union
+from typing import Any
 
 from ... import OpenAIWrapper, filter_config
 from ...code_utils import execute_code
@@ -176,8 +176,8 @@ class AgentOptimizer:
     def __init__(
         self,
         max_actions_per_step: int,
-        llm_config: Optional[Union[LLMConfig, dict[str, Any]]] = None,
-        optimizer_model: Optional[str] = "gpt-4-1106-preview",
+        llm_config: LLMConfig | dict[str, Any] | None = None,
+        optimizer_model: str | None = "gpt-4-1106-preview",
     ):
         """(These APIs are experimental and may change in the future.)
 

autogen/agentchat/contrib/capabilities/generate_images.py

@@ -5,7 +5,7 @@
 # Portions derived from  https://github.com/microsoft/autogen are under the MIT License.
 # SPDX-License-Identifier: MIT
 import re
-from typing import Any, Literal, Optional, Protocol, Union
+from typing import Any, Literal, Optional, Protocol
 
 from .... import Agent, ConversableAgent, code_utils
 from ....cache import AbstractCache
@@ -78,7 +78,7 @@ class DalleImageGenerator:
 
     def __init__(
         self,
-        llm_config: Union[LLMConfig, dict[str, Any]],
+        llm_config: LLMConfig | dict[str, Any],
         resolution: Literal["256x256", "512x512", "1024x1024", "1792x1024", "1024x1792"] = "1024x1024",
         quality: Literal["standard", "hd"] = "standard",
         num_images: int = 1,
@@ -153,8 +153,8 @@ class ImageGeneration(AgentCapability):
     def __init__(
         self,
         image_generator: ImageGenerator,
-        cache: Optional[AbstractCache] = None,
-        text_analyzer_llm_config: Optional[Union[LLMConfig, dict[str, Any]]] = None,
+        cache: AbstractCache | None = None,
+        text_analyzer_llm_config: LLMConfig | dict[str, Any] | None = None,
         text_analyzer_instructions: str = PROMPT_INSTRUCTIONS,
         verbosity: int = 0,
         register_reply_position: int = 2,
@@ -183,8 +183,8 @@ class ImageGeneration(AgentCapability):
         self._verbosity = verbosity
         self._register_reply_position = register_reply_position
 
-        self._agent: Optional[ConversableAgent] = None
-        self._text_analyzer: Optional[TextAnalyzerAgent] = None
+        self._agent: ConversableAgent | None = None
+        self._text_analyzer: TextAnalyzerAgent | None = None
 
     def add_to_agent(self, agent: ConversableAgent):
         """Adds the Image Generation capability to the specified ConversableAgent.
@@ -216,10 +216,10 @@ class ImageGeneration(AgentCapability):
     def _image_gen_reply(
         self,
         recipient: ConversableAgent,
-        messages: Optional[list[dict[str, Any]]],
-        sender: Optional[Agent] = None,
-        config: Optional[Any] = None,
-    ) -> tuple[bool, Optional[Union[str, dict[str, Any]]]]:
+        messages: list[dict[str, Any]] | None,
+        sender: Agent | None = None,
+        config: Any | None = None,
+    ) -> tuple[bool, str | dict[str, Any] | None]:
         if messages is None:
             return False, None
 
@@ -272,7 +272,7 @@ class ImageGeneration(AgentCapability):
             key = self._image_generator.cache_key(prompt)
             self._cache.set(key, img_utils.pil_to_data_uri(image))
 
-    def _extract_analysis(self, analysis: Optional[Union[str, dict[str, Any]]]) -> str:
+    def _extract_analysis(self, analysis: str | dict[str, Any] | None) -> str:
         if isinstance(analysis, dict):
             return code_utils.content_str(analysis["content"])
         else:

autogen/agentchat/contrib/capabilities/teachability.py

@@ -6,7 +6,7 @@
 # SPDX-License-Identifier: MIT
 import os
 import pickle
-from typing import Any, Optional, Union
+from typing import Any
 
 from ....formatting_utils import colored
 from ....import_utils import optional_import_block, require_optional_import
@@ -38,12 +38,12 @@ class Teachability(AgentCapability):
 
     def __init__(
         self,
-        verbosity: Optional[int] = 0,
-        reset_db: Optional[bool] = False,
-        path_to_db_dir: Optional[str] = "./tmp/teachable_agent_db",
-        recall_threshold: Optional[float] = 1.5,
-        max_num_retrievals: Optional[int] = 10,
-        llm_config: Optional[Union[LLMConfig, dict[str, Any], bool]] = None,
+        verbosity: int | None = 0,
+        reset_db: bool | None = False,
+        path_to_db_dir: str | None = "./tmp/teachable_agent_db",
+        recall_threshold: float | None = 1.5,
+        max_num_retrievals: int | None = 10,
+        llm_config: LLMConfig | dict[str, Any] | bool | None = None,
     ):
         """Args:
         verbosity (Optional, int): # 0 (default) for basic info, 1 to add memory operations, 2 for analyzer messages, 3 for memo lists.
@@ -92,7 +92,7 @@ class Teachability(AgentCapability):
         """Adds a few arbitrary memos to the DB."""
         self.memo_store.prepopulate()
 
-    def process_last_received_message(self, text: Union[dict[str, Any], str]):
+    def process_last_received_message(self, text: dict[str, Any] | str):
         """Appends any relevant memos to the message text, and stores any apparent teachings in new memos.
         Uses TextAnalyzerAgent to make decisions about memo storage and retrieval.
         """
@@ -107,7 +107,7 @@ class Teachability(AgentCapability):
         # Return the (possibly) expanded message text.
         return expanded_text
 
-    def _consider_memo_storage(self, comment: Union[dict[str, Any], str]):
+    def _consider_memo_storage(self, comment: dict[str, Any] | str):
         """Decides whether to store something from one user comment in the DB."""
         memo_added = False
 
@@ -165,7 +165,7 @@ class Teachability(AgentCapability):
             # Yes. Save them to disk.
             self.memo_store._save_memos()
 
-    def _consider_memo_retrieval(self, comment: Union[dict[str, Any], str]):
+    def _consider_memo_retrieval(self, comment: dict[str, Any] | str):
         """Decides whether to retrieve memos from the DB, and add them to the chat context."""
         # First, use the comment directly as the lookup key.
         if self.verbosity >= 1:
@@ -228,7 +228,7 @@ class Teachability(AgentCapability):
             memo_texts = memo_texts + "\n" + info
         return memo_texts
 
-    def _analyze(self, text_to_analyze: Union[dict[str, Any], str], analysis_instructions: Union[dict[str, Any], str]):
+    def _analyze(self, text_to_analyze: dict[str, Any] | str, analysis_instructions: dict[str, Any] | str):
         """Asks TextAnalyzerAgent to analyze the given text according to specific instructions."""
         self.analyzer.reset()  # Clear the analyzer's list of messages.
         self.teachable_agent.send(
@@ -251,9 +251,9 @@ class MemoStore:
 
     def __init__(
         self,
-        verbosity: Optional[int] = 0,
-        reset: Optional[bool] = False,
-        path_to_db_dir: Optional[str] = "./tmp/teachable_agent_db",
+        verbosity: int | None = 0,
+        reset: bool | None = False,
+        path_to_db_dir: str | None = "./tmp/teachable_agent_db",
     ):
         """Args:
         - verbosity (Optional, int): 1 to print memory operations, 0 to omit them. 3+ to print memo lists.
@@ -342,7 +342,7 @@ class MemoStore:
             )
         return input_text, output_text, distance
 
-    def get_related_memos(self, query_text: str, n_results: int, threshold: Union[int, float]):
+    def get_related_memos(self, query_text: str, n_results: int, threshold: int | float):
         """Retrieves memos that are related to the given query text within the specified distance threshold."""
         if n_results > len(self.uid_text_dict):
             n_results = len(self.uid_text_dict)