Add packaged rank-llm entrypoint baseline (#355)

Author: ronakice

pyproject.toml

@@ -104,6 +104,9 @@ dev = [
 [project.urls]
 Homepage = "https://github.com/castorini/rank_llm"
 
+[project.scripts]
+rank-llm = "rank_llm.cli.main:main"
+
 [tool.bumpver]
 current_version = "0.25.7"
 version_pattern = "MAJOR.MINOR.PATCH"
@@ -138,10 +141,13 @@ where = ["src"]
 include = [
     "rank_llm",
     "rank_llm.analysis",
+    "rank_llm.cli",
     "rank_llm.evaluation",
     "rank_llm.rerank",
     "rank_llm.rerank.*",
     "rank_llm.retrieve",
+    "rank_llm.server",
+    "rank_llm.server.*",
 ]
 namespaces = false
 

src/rank_llm/cli/__init__.py

@@ -0,0 +1 @@
+"""Packaged CLI surface for RankLLM."""

src/rank_llm/cli/main.py

@@ -0,0 +1,22 @@
+from __future__ import annotations
+
+import argparse
+from collections.abc import Sequence
+
+
+def build_parser() -> argparse.ArgumentParser:
+    parser = argparse.ArgumentParser(
+        prog="rank-llm",
+        description="Packaged CLI entrypoint for RankLLM.",
+    )
+    return parser
+
+
+def main(argv: Sequence[str] | None = None) -> int:
+    parser = build_parser()
+    parser.parse_args(argv)
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())

src/rank_llm/server/__init__.py

@@ -0,0 +1 @@
+"""Server entrypoints for RankLLM transports."""

src/rank_llm/server/flask/__init__.py

@@ -0,0 +1 @@
+"""Flask compatibility server for RankLLM."""

src/rank_llm/server/mcp/__init__.py

@@ -0,0 +1 @@
+"""MCP server for RankLLM."""

test/test_cli_packaging.py

@@ -0,0 +1,35 @@
+import importlib
+import subprocess
+import unittest
+from pathlib import Path
+from shutil import which
+
+REPO_ROOT = Path(__file__).resolve().parents[1]
+
+
+class TestCLIPackaging(unittest.TestCase):
+    def test_cli_module_imports(self):
+        module = importlib.import_module("rank_llm.cli.main")
+        self.assertTrue(callable(module.main))
+
+    def test_server_mcp_module_imports(self):
+        module = importlib.import_module("rank_llm.server.mcp")
+        self.assertIsNotNone(module)
+
+    def test_console_entrypoint_help_resolves(self):
+        cli = which("rank-llm")
+        self.assertIsNotNone(cli, msg="rank-llm is not installed in PATH")
+
+        help_result = subprocess.run(
+            [cli, "--help"],
+            cwd=REPO_ROOT,
+            check=False,
+            capture_output=True,
+            text=True,
+        )
+        self.assertEqual(help_result.returncode, 0, msg=help_result.stderr)
+        self.assertIn("Packaged CLI entrypoint for RankLLM", help_result.stdout)
+
+
+if __name__ == "__main__":
+    unittest.main()