From afd8b7ecb45c3188177a077717050af01fa314a2 Mon Sep 17 00:00:00 2001
From: Evan Mattson <35585003+moonbox3@users.noreply.github.com>
Date: Thu, 6 Nov 2025 07:26:18 +0900
Subject: [PATCH] Export sample agents (#1927)

---
 python/packages/ag-ui/examples/__init__.py    |  6 +++++
 .../ag-ui/examples/agents/__init__.py         | 22 +++++++++++++++++++
 python/packages/ag-ui/pyproject.toml          |  1 +
 3 files changed, 29 insertions(+)

diff --git a/python/packages/ag-ui/examples/__init__.py b/python/packages/ag-ui/examples/__init__.py
index 2a50eae894..54c879bbc7 100644
--- a/python/packages/ag-ui/examples/__init__.py
+++ b/python/packages/ag-ui/examples/__init__.py
@@ -1 +1,7 @@
 # Copyright (c) Microsoft. All rights reserved.
+
+"""Example agents for AG-UI demonstration."""
+
+from . import agents
+
+__all__ = ["agents"]
diff --git a/python/packages/ag-ui/examples/agents/__init__.py b/python/packages/ag-ui/examples/agents/__init__.py
index eea1a10956..720a16c765 100644
--- a/python/packages/ag-ui/examples/agents/__init__.py
+++ b/python/packages/ag-ui/examples/agents/__init__.py
@@ -1,3 +1,25 @@
 # Copyright (c) Microsoft. All rights reserved.
 
 """Example agents for AG-UI demonstration."""
+
+from .document_writer_agent import document_writer_agent
+from .human_in_the_loop_agent import human_in_the_loop_agent
+from .recipe_agent import recipe_agent
+from .research_assistant_agent import research_assistant_agent
+from .simple_agent import agent as simple_agent
+from .task_planner_agent import task_planner_agent
+from .task_steps_agent import task_steps_agent_wrapped
+from .ui_generator_agent import ui_generator_agent
+from .weather_agent import weather_agent
+
+__all__ = [
+    "document_writer_agent",
+    "human_in_the_loop_agent",
+    "recipe_agent",
+    "research_assistant_agent",
+    "simple_agent",
+    "task_planner_agent",
+    "task_steps_agent_wrapped",
+    "ui_generator_agent",
+    "weather_agent",
+]
diff --git a/python/packages/ag-ui/pyproject.toml b/python/packages/ag-ui/pyproject.toml
index 30a0518a39..4c2df6d111 100644
--- a/python/packages/ag-ui/pyproject.toml
+++ b/python/packages/ag-ui/pyproject.toml
@@ -41,6 +41,7 @@ build-backend = "hatchling.build"
 
 [tool.hatch.build.targets.wheel]
 packages = ["agent_framework_ag_ui"]
+force-include = { "examples" = "agent_framework_ag_ui_examples" }
 
 [tool.pytest.ini_options]
 asyncio_mode = "auto"