Export sample agents (#1927)

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"]

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",
+]

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"