ai_review: CLI 标准化(argparse + --host/--port/--log-level/--instance)

- 新增 --host/--port 覆盖 $AI_REVIEW_HOST/$AI_REVIEW_PORT(CLI > ENV > 默认值)
- 新增 --log-level debug|info|warn|error(启动打印,ai_review 无 db 日志客户端故仅作标准接口)
- 新增 --instance 实例名(注册中心用)
- 配置加载从模块顶层移入 __main__,argparse 解析
- services.yaml: ai_review 声明 --instance 和 --log-level
This commit is contained in:
qing committed 2026-08-10 12:02:47 +08:00
1 parent 29eac70f24
commit d0ee948339
2 files changed
+31 -1

No files matched your search

+30
View File
@@ -165,9 +165,39 @@ def rag_list():
return jsonify({"total": len(entries), "entries": entries}), 200 return jsonify({"total": len(entries), "entries": entries}), 200
def _parse_args():
"""解析命令行参数(标准化:CLI > ENV > 默认值)。"""
import argparse
p = argparse.ArgumentParser(
prog="ai_review",
description="AI Review Service — 基于 DeepSeek + RAG 的内容审核微服务",
)
p.add_argument("--host", default=None,
help="绑定地址(默认读取 $AI_REVIEW_HOST,否则 127.0.0.1)")
p.add_argument("--port", type=int, default=None,
help="监听端口(默认读取 $AI_REVIEW_PORT,否则 8007)")
p.add_argument("--log-level", default="info",
choices=["debug", "info", "warn", "error"],
help="日志级别(默认 info)")
p.add_argument("--instance", default="ai_review",
help="实例名(注册中心用,默认 ai_review)")
return p.parse_args()
# ── 配置默认值(ENV 覆盖,CLI 最高优先级)───────────────────────────
ADDR = os.getenv("AI_REVIEW_HOST", "127.0.0.1") ADDR = os.getenv("AI_REVIEW_HOST", "127.0.0.1")
PORT = int(os.getenv("AI_REVIEW_PORT", "8007")) PORT = int(os.getenv("AI_REVIEW_PORT", "8007"))
if __name__ == "__main__": if __name__ == "__main__":
args = _parse_args()
# CLI 覆盖(最高优先级)
if args.host is not None:
ADDR = args.host
if args.port is not None:
PORT = args.port
print(f'AI Review Service listening at http://{ADDR}:{PORT}') print(f'AI Review Service listening at http://{ADDR}:{PORT}')
print(f'[ai_review] instance={args.instance} log-level={args.log_level}')
print(f'[ai_review] 配置来源: CLI > ENV > 默认值')
serve(app, host=ADDR, port=PORT) serve(app, host=ADDR, port=PORT)
+1 -1
View File
@@ -21,7 +21,7 @@ services:
# --- AI 审核 --- # --- AI 审核 ---
- name: ai_review - name: ai_review
cmd: bash /opt/micro_servers/ai_review/_run.sh cmd: bash /opt/micro_servers/ai_review/_run.sh
args: [] args: ["--instance", "ai_review", "--log-level", "info"]
depends_on: [db2] depends_on: [db2]
sleep_before: 2 sleep_before: 2