From 00d9c09d810a735b58d77845fed85625a455e87c Mon Sep 17 00:00:00 2001
From: yanrujing <rujing.yan@protagonist-ai.com>
Date: Wed, 28 May 2025 14:32:49 +0800
Subject: [PATCH] chore: chore

---
 pyproject.toml        | 1 +
 src/mcp_server_box.py | 7 ++++++-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index f8267ab..002b7d0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,6 +4,7 @@ version = "0.1.0"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.13"
+scripts = { "mcp-server-box" = "mcp_server_box:main" }
 dependencies = [
     "box-ai-agents-toolkit>=0.0.42",
     "box-sdk-gen>=1.13.0",
diff --git a/src/mcp_server_box.py b/src/mcp_server_box.py
index 04d00ed..09bb7d9 100644
--- a/src/mcp_server_box.py
+++ b/src/mcp_server_box.py
@@ -888,6 +888,11 @@ def _serialize(obj):
         # If all else fails, convert to string
         return str(obj)
 
-if __name__ == "__main__":
+
+def main():
     # Initialize and run the server
     mcp.run(transport="stdio")
+
+
+if __name__ == "__main__":
+    main()