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

This commit is contained in:
2026-04-02 15:05:07 +08:00
parent 9f1561afe7
commit 475de51c7d
12 changed files with 616 additions and 99 deletions
+37
View File
@@ -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 autoreload 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()