docs: add missing TextMessage import in custom agents documentation

tejas-dharani · tejas-dharani · commit d07898bfc077 · 2025-06-02T11:30:34.000+05:30
Fix issue #6277 where TextMessage was used but not imported in three code cells of the custom agents documentation, causing NameError when users run the examples. Changes: - Add TextMessage to imports in ArithmeticAgent section - Add TextMessage to imports in GeminiAssistantAgent section - Add TextMessage to imports in Declarative GeminiAssistantAgent section The CountDownAgent section already had the correct import. Fixes #6277
diff --git a/python/packages/autogen-core/docs/src/user-guide/agentchat-user-guide/custom-agents.ipynb b/python/packages/autogen-core/docs/src/user-guide/agentchat-user-guide/custom-agents.ipynb
@@ -137,7 +137,7 @@
                 "from autogen_agentchat.agents import BaseChatAgent\n",
                 "from autogen_agentchat.base import Response\n",
                 "from autogen_agentchat.conditions import MaxMessageTermination\n",
-                "from autogen_agentchat.messages import BaseChatMessage\n",
+                "from autogen_agentchat.messages import BaseChatMessage, TextMessage\n",
                 "from autogen_agentchat.teams import SelectorGroupChat\n",
                 "from autogen_agentchat.ui import Console\n",
                 "from autogen_core import CancellationToken\n",
@@ -321,7 +321,7 @@
                 "\n",
                 "from autogen_agentchat.agents import BaseChatAgent\n",
                 "from autogen_agentchat.base import Response\n",
-                "from autogen_agentchat.messages import BaseAgentEvent, BaseChatMessage\n",
+                "from autogen_agentchat.messages import BaseAgentEvent, BaseChatMessage, TextMessage\n",
                 "from autogen_core import CancellationToken\n",
                 "from autogen_core.model_context import UnboundedChatCompletionContext\n",
                 "from autogen_core.models import AssistantMessage, RequestUsage, UserMessage\n",
@@ -552,7 +552,7 @@
                 "\n",
                 "from autogen_agentchat.agents import BaseChatAgent\n",
                 "from autogen_agentchat.base import Response\n",
-                "from autogen_agentchat.messages import BaseAgentEvent, BaseChatMessage\n",
+                "from autogen_agentchat.messages import BaseAgentEvent, BaseChatMessage, TextMessage\n",
                 "from autogen_core import CancellationToken, Component\n",
                 "from pydantic import BaseModel\n",
                 "from typing_extensions import Self\n",
