From 6fa26f1e47c73b245ab62458d418a8a2340d5b7d Mon Sep 17 00:00:00 2001
From: Mark Sze <mark@sze.family>
Date: Sat, 23 Nov 2024 21:20:19 +0000
Subject: [PATCH] Add CaptainAgent tools to pip install

---
 MANIFEST.in | 2 ++
 setup.py    | 6 ++++++
 2 files changed, 8 insertions(+)
 create mode 100644 MANIFEST.in

diff --git a/MANIFEST.in b/MANIFEST.in
new file mode 100644
index 0000000000..25700f369d
--- /dev/null
+++ b/MANIFEST.in
@@ -0,0 +1,2 @@
+# Include CaptainAgent tools
+recursive-include autogen/agentchat/contrib/captainagent/tools *
diff --git a/setup.py b/setup.py
index 16abda3298..f2bb9c7548 100644
--- a/setup.py
+++ b/setup.py
@@ -131,6 +131,12 @@
     long_description_content_type="text/markdown",
     url="https://github.com/ag2ai/ag2",
     packages=setuptools.find_packages(include=["autogen*"], exclude=["test"]),
+    package_data={
+        "autogen.agentchat.contrib.captainagent": [
+            "tools/**/*",  # CaptainAgent tools
+        ]
+    },
+    include_package_data=True,
     install_requires=install_requires,
     extras_require=extra_require,
     classifiers=[