snowy-yang
c414db4217
项目由 uv 管理，依赖装在 .venv。此前各处的提示语仍写作 `python main.py ...`，用户若在未激活虚拟环境时照抄，会因缺少 fastapi 而失败——这是正确性问题，不只是措辞。 涉及：main.py（12 行，含 prog 与各条 hint）、ai_player.py（5 行）、farm/client.py、farm/server/__main__.py（3 行，含 prog）、farm/server/__init__.py、farm/server/daemon.py、README（4 处示例与目录树注释）。 刻意保留 main.py:26 与 README:49 的「没装 uv？」兜底说明，那里 `python main.py` 是正确写法。 验收：ruff 全通过 / format 26 文件无改动 / pyright 0 错误 / 97 项测试通过。
90 lines
3.2 KiB
Python
90 lines
3.2 KiB
Python
#!/usr/bin/env python3
|
|
"""外挂式 AI/机器人玩家——完全通过 HTTP + WebSocket 接口参与游戏。
|
|
|
|
这个脚本**不**导入任何游戏规则代码，只用接口。换句话说：
|
|
它对服务器来说和一个远程 LLM Agent 没有任何区别。
|
|
|
|
用法
|
|
----
|
|
uv run ai_player.py # 连本机 default 牧场
|
|
uv run ai_player.py --farm ai-1 --minutes 5 # 经营 ai-1 五分钟
|
|
uv run 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="uv run 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(" 先启动：uv run 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())
|