chore: import upstream snapshot with attribution

python/mlc_llm/__main__.py

@@ -0,0 +1,67 @@
+"""Entrypoint of all CLI commands from MLC LLM"""
+
+import sys
+
+from mlc_llm.support import logging
+from mlc_llm.support.argparse import ArgumentParser
+
+logging.enable_logging()
+
+
+def main():
+    """Entrypoint of all CLI commands from MLC LLM"""
+    parser = ArgumentParser("MLC LLM Command Line Interface.")
+    parser.add_argument(
+        "subcommand",
+        type=str,
+        choices=[
+            "compile",
+            "convert_weight",
+            "gen_config",
+            "chat",
+            "serve",
+            "package",
+            "calibrate",
+            "router",
+        ],
+        help="Subcommand to to run. (choices: %(choices)s)",
+    )
+    parsed = parser.parse_args(sys.argv[1:2])
+    if parsed.subcommand == "compile":
+        from mlc_llm.cli import compile as cli
+
+        cli.main(sys.argv[2:])
+    elif parsed.subcommand == "convert_weight":
+        from mlc_llm.cli import convert_weight as cli
+
+        cli.main(sys.argv[2:])
+    elif parsed.subcommand == "gen_config":
+        from mlc_llm.cli import gen_config as cli
+
+        cli.main(sys.argv[2:])
+    elif parsed.subcommand == "chat":
+        from mlc_llm.cli import chat as cli
+
+        cli.main(sys.argv[2:])
+    elif parsed.subcommand == "serve":
+        from mlc_llm.cli import serve as cli
+
+        cli.main(sys.argv[2:])
+    elif parsed.subcommand == "package":
+        from mlc_llm.cli import package as cli
+
+        cli.main(sys.argv[2:])
+    elif parsed.subcommand == "calibrate":
+        from mlc_llm.cli import calibrate as cli
+
+        cli.main(sys.argv[2:])
+    elif parsed.subcommand == "router":
+        from mlc_llm.cli import router as cli
+
+        cli.main(sys.argv[2:])
+    else:
+        raise ValueError(f"Unknown subcommand {parsed.subcommand}")
+
+
+if __name__ == "__main__":
+    main()