quantnodes 3.0.0__py3-none-any.whl

QuantNodes/cli/commands/init.py

@@ -0,0 +1,190 @@
+# coding=utf-8
+"""``quantnodes init`` command."""
+
+from pathlib import Path
+
+from QuantNodes.cli.command import Command
+from QuantNodes.constants import DEFAULT_LLM_MODEL
+from .._helpers import (
+    confirm_section,
+    create_directory_structure,
+    get_input_with_default,
+    get_model_choice,
+    get_yes_no,
+    init_llmwikify_wiki,
+    install_talib,
+    print_nanobot_install_hint,
+    write_conn_ini,
+    write_env_file,
+)
+
+
+def cmd_init(args) -> int:
+    """Initialize current directory for QuantNodes."""
+    print()
+    print("=" * 50)
+    print("QuantNodes 初始化向导")
+    print("=" * 50)
+
+    current_dir = Path.cwd()
+    print(f"\n✓ 检测当前目录: {current_dir}")
+    print("✓ 检查初始化状态...")
+
+    already_init = False
+    if Path(".env").exists():
+        print("  ✗ .env 已存在")
+        already_init = True
+
+    if Path("conn.ini").exists():
+        print("  ✗ conn.ini 已存在")
+        already_init = True
+
+    if already_init and not args.force:
+        print("\n错误: 当前目录已初始化")
+        print("请先 cd 到其他目录，或使用 --force 强制重新初始化")
+        return 1
+
+    if already_init and args.force:
+        print("  (强制模式: 将覆盖现有配置)")
+
+    print()
+
+    print("-" * 50)
+    print("配置 LLM")
+    print("-" * 50)
+
+    api_key = get_input_with_default(
+        "请输入 OpenAI API Key (sk-...)",
+        "",
+        required=True
+    )
+    while not api_key.startswith("sk-"):
+        print("  错误: API Key 必须以 sk- 开头")
+        api_key = get_input_with_default(
+            "请输入 OpenAI API Key (sk-...)",
+            "",
+            required=True
+        )
+
+    base_url = get_input_with_default(
+        "请输入 API Base URL",
+        "https://api.openai.com/v1"
+    )
+
+    model = get_model_choice()
+    if model == "custom":
+        model = get_input_with_default("请输入自定义模型名称", DEFAULT_LLM_MODEL)
+
+    print()
+    print("-" * 50)
+    print("配置数据源")
+    print("-" * 50)
+
+    duckdb_path = get_input_with_default(
+        "DuckDB 数据库路径",
+        "data/quantnodes.db"
+    )
+
+    configure_clickhouse = get_yes_no("是否配置 ClickHouse", default=False)
+    configure_mysql = get_yes_no("是否配置 MySQL", default=False)
+
+    clickhouse_config = {}
+    mysql_config = {}
+
+    if configure_clickhouse:
+        clickhouse_config = confirm_section(
+            "ClickHouse",
+            [
+                ("Host", "localhost"),
+                ("Port", "8123"),
+                ("User", "default"),
+                ("Password", ""),
+                ("Database", "default"),
+            ],
+        ) or {}
+
+    if configure_mysql:
+        mysql_config = confirm_section(
+            "MySQL",
+            [
+                ("Host", "localhost"),
+                ("Port", "3306"),
+                ("User", "root"),
+                ("Password", ""),
+                ("Database", "quant"),
+            ],
+        ) or {}
+
+    print()
+
+    print("-" * 50)
+    print("初始化 llmwikify Wiki")
+    print("-" * 50)
+    init_llmwikify_wiki(force=args.force)
+    print()
+
+    print("-" * 50)
+    print("创建目录结构")
+    print("-" * 50)
+    create_directory_structure()
+    print()
+
+    print("-" * 50)
+    print("创建配置文件")
+    print("-" * 50)
+
+    write_env_file(api_key, base_url, model, duckdb_path, clickhouse_config, mysql_config)
+    write_conn_ini(duckdb_path, clickhouse_config, mysql_config)
+
+    print()
+
+    install_talib_option = get_yes_no("是否安装 TA-Lib 技术分析库 (可选)", default=True)
+    if install_talib_option:
+        install_talib()
+
+    print()
+    print("=" * 50)
+    print("✓ 初始化完成!")
+    print("=" * 50)
+    print()
+    print("快速启动:")
+    print("  # 启动后端（推荐）")
+    print("  quantnodes serve                  # 前台，Ctrl+C 停止")
+    print("  quantnodes serve --daemon         # 后台，写 .quantnodes.pid")
+    print("  quantnodes serve --frontend       # 同时启动 Vite dev server")
+    print("  quantnodes serve --check-env     # 启动前校验 API key")
+    print()
+    print("  # 服务管理")
+    print("  quantnodes status                 # health + agent state")
+    print("  quantnodes logs -f                # 实时日志")
+    print("  quantnodes stop                   # 停止后台 serve")
+    print()
+    print("  # Agent Chat（HTTP 模式，需后端在跑）")
+    print("  quantnodes agent chat '一句话回答动量因子'")
+    print()
+    print("  # 启动前端 (新终端)")
+    print("  cd frontend && npm run dev")
+    print()
+    print("  # 或使用 quantnodes run 启动全部服务（旧接口，兼容保留）")
+    print()
+    print("访问 http://localhost:5173")
+    print()
+
+    # v3.0.0 Stage 7: 友好提示 nanobot-ai 可选依赖
+    print_nanobot_install_hint()
+
+    return 0
+
+
+class InitCommand(Command):
+    """``quantnodes init`` subcommand."""
+
+    name = "init"
+    description = "初始化当前目录"
+
+    def add_arguments(self, subparsers) -> None:
+        p = subparsers.add_parser(self.name, help=self.description)
+        p.add_argument("--force", action="store_true", help="强制重新初始化")
+
+    def run(self, args) -> int:
+        return cmd_init(args)

QuantNodes/cli/commands/run.py

@@ -0,0 +1,259 @@
+# coding=utf-8
+"""``quantnodes run`` command + server start helpers."""
+
+import os
+import subprocess
+import sys
+from datetime import datetime
+from pathlib import Path
+from typing import Optional, Tuple, Any, List
+
+from QuantNodes.core.path_utils import ensure_parent
+from QuantNodes.cli.command import Command
+
+from .._helpers import (
+    DEFAULT_API_PORT,
+    DEFAULT_FRONTEND_PORT,
+    DEFAULT_GATEWAY_PORT,
+    DEFAULT_HOST,
+    get_project_root,
+    is_initialized,
+)
+
+
+def start_api_server(
+    host: str,
+    port: int,
+    log_file: Optional[Path] = None,
+    gateway_port: Optional[int] = None,
+) -> Tuple[subprocess.Popen, Optional[Any]]:
+    """Start the API server. Returns (process, log_file_handle).
+
+    v3.0.0 Stage 7: ``gateway_port`` (default ``DEFAULT_GATEWAY_PORT=18090``)
+    is injected into the subprocess env as ``NANOBOT_GATEWAY_PORT`` so the
+    nanobot WebSocket gateway binds to a port not occupied by gpustack
+    (which defaults to 18080).
+    """
+    cmd = [
+        sys.executable, "-m", "uvicorn",
+        "api.main:app",
+        "--host", host,
+        "--port", str(port),
+        "--reload"
+    ]
+
+    env = os.environ.copy()
+    env["NANOBOT_GATEWAY_HOST"] = host
+    env["NANOBOT_GATEWAY_PORT"] = str(gateway_port or DEFAULT_GATEWAY_PORT)
+
+    if log_file:
+        ensure_parent(log_file)
+        log_fd = open(log_file, "w", encoding="utf-8")
+        proc = subprocess.Popen(
+            cmd,
+            stdout=log_fd,
+            stderr=subprocess.STDOUT,
+            cwd=get_project_root(),
+            env=env,
+        )
+        return proc, log_fd
+    else:
+        proc = subprocess.Popen(
+            cmd,
+            cwd=get_project_root(),
+            env=env,
+        )
+        return proc, None
+
+
+def start_frontend_server(
+    host: str,
+    port: int,
+    api_port: int = DEFAULT_API_PORT,
+    log_file: Optional[Path] = None,
+) -> Tuple[subprocess.Popen, Optional[Any]]:
+    """Start the frontend server. Returns (process, log_file_handle)."""
+    cmd = ["npm", "run", "dev"]
+
+    env = os.environ.copy()
+    env["HOST"] = host
+    env["PORT"] = str(port)
+    env["API_PORT"] = str(api_port)
+
+    if log_file:
+        ensure_parent(log_file)
+        log_fd = open(log_file, "w", encoding="utf-8")
+        proc = subprocess.Popen(
+            cmd,
+            stdout=log_fd,
+            stderr=subprocess.STDOUT,
+            cwd=str(get_project_root() / "frontend"),
+            env=env
+        )
+        return proc, log_fd
+    else:
+        proc = subprocess.Popen(
+            cmd,
+            cwd=str(get_project_root() / "frontend"),
+            env=env
+        )
+        return proc, None
+
+
+def cmd_run(args) -> int:
+    """Start QuantNodes services."""
+    if not is_initialized():
+        print("错误: 当前目录未初始化")
+        print("请先运行: quantnodes init")
+        return 1
+
+    host = args.host or DEFAULT_HOST
+    frontend_port = args.port or DEFAULT_FRONTEND_PORT
+    # 联动：如果只设置 --port，则 api_port = port + 1000
+    if args.port and not args.api_port:
+        api_port = args.port + 1000
+    else:
+        api_port = args.api_port or DEFAULT_API_PORT
+
+    if args.daemon:
+        if sys.platform != "linux":
+            print("错误: daemon 模式仅支持 Linux")
+            return 1
+
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        log_dir = Path("logs")
+        log_dir.mkdir(exist_ok=True)
+
+        api_log = log_dir / f"quantnodes_api_{timestamp}.log"
+        frontend_log = log_dir / f"quantnodes_frontend_{timestamp}.log"
+
+        print("=" * 50)
+        print("QuantNodes 服务 (后台运行)")
+        print("=" * 50)
+        print(f"  后端: http://{host}:{api_port}")
+        print(f"  前端: http://{host}:{frontend_port}")
+        print(f"  API 日志: {api_log}")
+        print(f"  前端日志: {frontend_log}")
+        print()
+
+        api_proc, api_fd = start_api_server(host, api_port, api_log,
+                                             gateway_port=args.gateway_port)
+        frontend_proc, frontend_fd = start_frontend_server(
+            host, frontend_port, api_port, frontend_log,
+        )
+
+        print("✓ 服务已后台启动")
+        print(f"  API 进程: {api_proc.pid}")
+        print(f"  前端进程: {frontend_proc.pid}")
+        print(f"  nanobot gateway: ws://{host}:{args.gateway_port}")
+        print()
+        print("查看日志:")
+        print(f"  tail -f {api_log}")
+        print(f"  tail -f {frontend_log}")
+        print()
+        print("停止服务:")
+        print(f"  kill {api_proc.pid} {frontend_proc.pid}")
+        print(f"  (quantnodes stop 仅作用于 serve --daemon 启动的服务)")
+
+        return 0
+
+    print("=" * 50)
+    print("QuantNodes 服务")
+    print("=" * 50)
+
+    processes: List[Tuple[str, subprocess.Popen]] = []
+    log_fds: List[Any] = []
+
+    try:
+        if not args.frontend_only:
+            print(f"\n启动后端: http://{host}:{api_port}")
+            print(f"  nanobot gateway: ws://{host}:{args.gateway_port}")
+            api_proc, api_fd = start_api_server(host, api_port,
+                                                 gateway_port=args.gateway_port)
+            processes.append(("API", api_proc))
+            log_fds.append(api_fd)
+            print(f"  进程 PID: {api_proc.pid}")
+
+        if not args.api_only:
+            print(f"\n启动前端: http://{host}:{frontend_port}")
+            # Wait for backend to be ready before starting frontend
+            import time
+            import urllib.request
+            import urllib.error
+            print("  等待后端就绪...")
+            for i in range(30):
+                try:
+                    urllib.request.urlopen(f"http://localhost:{api_port}/docs", timeout=2)
+                    print("  ✓ 后端已就绪")
+                    break
+                except (urllib.error.URLError, OSError):
+                    time.sleep(1)
+            else:
+                print("  ⚠ 后端未就绪，继续启动前端")
+            frontend_proc, frontend_fd = start_frontend_server(host, frontend_port, api_port)
+            processes.append(("Frontend", frontend_proc))
+            log_fds.append(frontend_fd)
+            print(f"  进程 PID: {frontend_proc.pid}")
+
+        print()
+        print("=" * 50)
+        print("✓ 服务已启动")
+        print("=" * 50)
+        print()
+        print("访问:")
+        if not args.frontend_only:
+            print(f"  后端: http://localhost:{api_port}/docs")
+        if not args.api_only:
+            print(f"  前端: http://localhost:{frontend_port}")
+        print()
+        print("按 Ctrl+C 停止服务")
+        print()
+
+        try:
+            for name, proc in processes:
+                proc.wait()
+        except KeyboardInterrupt:
+            print("\n\n正在停止服务...")
+            for name, proc in processes:
+                proc.terminate()
+                proc.wait()
+            for fd in log_fds:
+                if fd:
+                    fd.close()
+            print("✓ 服务已停止")
+
+    except Exception as e:
+        print(f"错误: {e}")
+        for name, proc in processes:
+            proc.terminate()
+        for fd in log_fds:
+            if fd:
+                fd.close()
+        return 1
+
+    return 0
+
+
+class RunCommand(Command):
+    """``quantnodes run`` subcommand."""
+
+    name = "run"
+    description = "启动服务"
+
+    def add_arguments(self, subparsers) -> None:
+        p = subparsers.add_parser(self.name, help=self.description)
+        p.add_argument("--host", help="绑定主机")
+        p.add_argument("--port", type=int, help="前端端口")
+        p.add_argument("--api-port", type=int, dest="api_port", help="后端端口")
+        # v3.0.0 Stage 7: nanobot WebSocket gateway 端口（注入到子进程 env）
+        p.add_argument("--gateway-port", type=int, dest="gateway_port",
+                       default=DEFAULT_GATEWAY_PORT,
+                       help=f"nanobot WebSocket gateway 端口 (默认 {DEFAULT_GATEWAY_PORT})")
+        p.add_argument("--daemon", action="store_true", help="后台运行 (仅 Linux)")
+        p.add_argument("--api-only", action="store_true", dest="api_only", help="仅启动后端")
+        p.add_argument(
+            "--frontend-only", action="store_true", dest="frontend_only", help="仅启动前端"
+        )
+
+    def run(self, args) -> int:
+        return cmd_run(args)