From 00eeadb9dd4bbe6dd0ead55141a9e66e112f40a1 Mon Sep 17 00:00:00 2001
From: yichuan520030910320 <yichuan_wang@berkeley.edu>
Date: Thu, 14 Aug 2025 14:39:45 -0700
Subject: [PATCH] upd pkg

---
 packages/wechat-exporter/__init__.py | 1 +
 packages/wechat-exporter/main.py     | 6 +++++-
 pyproject.toml                       | 6 ++++++
 3 files changed, 12 insertions(+), 1 deletion(-)
 create mode 100644 packages/wechat-exporter/__init__.py

diff --git a/packages/wechat-exporter/__init__.py b/packages/wechat-exporter/__init__.py
new file mode 100644
index 0000000..a9a2c5b
--- /dev/null
+++ b/packages/wechat-exporter/__init__.py
@@ -0,0 +1 @@
+__all__ = []
diff --git a/packages/wechat-exporter/main.py b/packages/wechat-exporter/main.py
index 5b0911e..7561f05 100644
--- a/packages/wechat-exporter/main.py
+++ b/packages/wechat-exporter/main.py
@@ -136,5 +136,9 @@ def export_sqlite(
     connection.commit()
 
 
-if __name__ == "__main__":
+def main():
     app()
+
+
+if __name__ == "__main__":
+    main()
diff --git a/pyproject.toml b/pyproject.toml
index 73dd7a7..9aa3db3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -10,6 +10,7 @@ requires-python = ">=3.9"
 dependencies = [
     "leann-core",
     "leann-backend-hnsw",
+    "typer>=0.12.3",
     "numpy>=1.26.0",
     "torch",
     "tqdm",
@@ -84,6 +85,11 @@ documents = [
 
 [tool.setuptools]
 py-modules = []
+packages = ["wechat_exporter"]
+package-dir = { "wechat_exporter" = "packages/wechat-exporter" }
+
+[project.scripts]
+wechat-exporter = "wechat_exporter.main:main"
 
 
 [tool.uv.sources]