From 6e8cf907b8acff823d55ec09f9577a4c1aef157c Mon Sep 17 00:00:00 2001
From: Mehrdad Hessar <mhessar@octo.ai>
Date: Fri, 19 Jan 2024 16:56:59 +0000
Subject: [PATCH] Add mlc_serve as a package

---
 setup.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index b9721497c2..ed25a674b3 100644
--- a/setup.py
+++ b/setup.py
@@ -22,6 +22,10 @@ def git_describe_version(original_version):
 
 __version__ = git_describe_version(None)
 
+def find_mlc_serve():
+    packages = find_packages(where="serve")
+    return [package for package in packages if package.startswith("mlc_serve")]
+
 setup(
     name="mlc_llm",
     version=__version__,
@@ -39,8 +43,8 @@ def git_describe_version(original_version):
     ],
     keywords="machine learning",
     zip_safe=False,
-    packages=find_packages(),
-    package_dir={"mlc_llm": "mlc_llm"},
+    packages=find_packages() + find_mlc_serve(),
+    package_dir={"mlc_llm": "mlc_llm", "mlc_serve": "serve/mlc_serve"},
     install_requires=["numpy", "torch", "transformers", "scipy", "timm"],
     entry_points={"console_scripts": ["mlc_llm_build = mlc_llm.build:main"]},
     distclass=Distribution,