feat: 增加server, 并且修改测试单元

2026-04-02 15:05:07 +08:00
parent 9f1561afe7
commit 475de51c7d
12 changed files with 616 additions and 99 deletions
@@ -0,0 +1,37 @@
+import argparse
+from pathlib import Path
+from astrai.inference.server import run_server
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Start AstrAI inference HTTP server")
+    parser.add_argument(
+        "--host", default="0.0.0.0", help="Host address (default: 0.0.0.0)"
+    )
+    parser.add_argument(
+        "--port", type=int, default=8000, help="Port number (default: 8000)"
+    )
+    parser.add_argument(
+        "--reload", action="store_true", help="Enable auto‑reload for development"
+    )
+    parser.add_argument(
+        "--param-path",
+        type=Path,
+        default=None,
+        help="Path to model parameters (default: project_root/params)",
+    )
+    args = parser.parse_args()
+
+    # If param_path is provided, set environment variable or modify global?
+    # Currently the server loads from default location on startup.
+    # We could pass it via an environment variable, but for simplicity we assume
+    # the default location is correct.
+    project_root = Path(__file__).parent.parent
+    param_path = args.param_path or (project_root / "params")
+    print(f"Starting AstrAI inference server on http://{args.host}:{args.port}")
+    print(f"Model parameters expected at: {[param_path]}")
+    run_server(host=args.host, port=args.port, reload=args.reload)
+
+
+if __name__ == "__main__":
+    main()