snowy-yang
a7725f8812
在引入 uv 依赖管理、ruff 静态检查与格式化、pyright 类型检查之前， 先把当前可正常运行的状态固定下来，作为后续大规模自动改动的回滚点。 此时状态：97 项测试全绿，HTTP + WebSocket 多牧场服务器可用。
79 lines
3.1 KiB
Python
79 lines
3.1 KiB
Python
#!/usr/bin/env python3
|
|
"""外挂式 AI/机器人玩家——完全通过 HTTP + WebSocket 接口参与游戏。
|
|
|
|
这个脚本**不**导入任何游戏规则代码，只用接口。换句话说：
|
|
它对服务器来说和一个远程 LLM Agent 没有任何区别。
|
|
|
|
用法
|
|
----
|
|
python ai_player.py # 连本机 default 牧场
|
|
python ai_player.py --farm ai-1 --minutes 5 # 经营 ai-1 五分钟
|
|
python ai_player.py --url http://10.0.0.5:8765 --farm my-bot
|
|
|
|
策略在哪里
|
|
----------
|
|
决策逻辑在 :mod:`farm.bot` 的 ``RuleBot.score()``：那是一张"什么该先做"的
|
|
打分表。想换成真 LLM，只需要把「读 /api/actions 之后选哪个」这一步换成模型
|
|
调用——接口调用序列完全一样：
|
|
|
|
读 GET /api/actions → 挑一个 → POST /api/actions → 读回执里的 state
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import json
|
|
import sys
|
|
|
|
from farm.bot import RuleBot, random_bot
|
|
from farm.client import RemoteFarm
|
|
from farm.engine import GameError
|
|
|
|
|
|
def build_parser() -> argparse.ArgumentParser:
|
|
parser = argparse.ArgumentParser(
|
|
prog="python ai_player.py",
|
|
description="通过 HTTP/WebSocket 接口参与 SimpleFarm 的机器人玩家",
|
|
)
|
|
parser.add_argument("--url", default="http://127.0.0.1:8765", help="服务器地址")
|
|
parser.add_argument("--farm", default="ai-1", help="经营哪个牧场，默认 ai-1")
|
|
parser.add_argument("--token", default=None, help="访问令牌（服务器启用时必填）")
|
|
parser.add_argument("--rounds", type=int, default=0, help="最多执行多少轮，0 表示不限")
|
|
parser.add_argument("--minutes", type=float, default=0.0, help="最多跑多少分钟，0 表示不限")
|
|
parser.add_argument("--pause", type=float, default=0.25, help="动作间隔（秒）")
|
|
parser.add_argument("--random", action="store_true", help="用随机策略（对照组）")
|
|
parser.add_argument("--quiet", action="store_true", help="只输出结果，不打过程")
|
|
return parser
|
|
|
|
|
|
def main(argv=None) -> int:
|
|
args = build_parser().parse_args(argv)
|
|
|
|
try:
|
|
client = RemoteFarm(endpoint=args.url, farm=args.farm, token=args.token,
|
|
name=f"bot-{args.farm}", realtime=False)
|
|
except GameError as exc:
|
|
print(f"✗ 连不上服务器：{exc.message}", file=sys.stderr)
|
|
print(f" 先启动：python main.py serve --daemon", file=sys.stderr)
|
|
return 2
|
|
|
|
snapshot = client.snapshot()
|
|
print(f"已连接 {args.url} · 牧场「{args.farm}」· 第 {snapshot.get('day')} 天 · "
|
|
f"金币 {snapshot.get('gold')}")
|
|
|
|
factory = random_bot if args.random else RuleBot
|
|
bot = factory(client, verbose=not args.quiet, pause=args.pause, name=args.farm)
|
|
try:
|
|
summary = bot.run(rounds=args.rounds, max_minutes=args.minutes)
|
|
except KeyboardInterrupt:
|
|
bot.stop()
|
|
summary = {"name": args.farm, "actions": bot.actions_executed,
|
|
"failures": bot.failures, "interrupted": True}
|
|
|
|
print(json.dumps(summary, ensure_ascii=False, indent=2))
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|