diff --git a/autogen/messages/agent_messages.py b/autogen/messages/agent_messages.py index 22e827cf3f..a054936072 100644 --- a/autogen/messages/agent_messages.py +++ b/autogen/messages/agent_messages.py @@ -4,7 +4,7 @@ from abc import ABC from copy import deepcopy -from typing import TYPE_CHECKING, Any, Callable, Literal, Optional, Union +from typing import TYPE_CHECKING, Any, Callable, Dict, List, Literal, Optional, Union from uuid import UUID from pydantic import BaseModel @@ -189,7 +189,7 @@ def print(self, f: Optional[Callable[..., Any]] = None) -> None: @wrap_message class TextMessage(BasePrintReceivedMessage): - content: Optional[Union[str, int, float, bool, list[dict[str, str | dict[str, Any]]]]] = None # type: ignore [assignment] + content: Optional[Union[str, int, float, bool, List[Dict[str, Union[str, Dict[str, Any]]]]]] = None # type: ignore [assignment] def print(self, f: Optional[Callable[..., Any]] = None) -> None: f = f or print