218 lines
9.8 KiB
Python
218 lines
9.8 KiB
Python
"""Agent 编排层：OpenAI 兼容模型（DeepSeek 等）经 MCP 工具控制手机。
|
||
|
||
支持两种运行模式：
|
||
- run_stream()：流式（SSE 逐 token + 工具调用实时回调）——Web AI 控制台用
|
||
- run()：非流式收集结果——CLI 用（内部调 run_stream）
|
||
|
||
流式细节（OpenAI 兼容）：
|
||
- content/reasoning_content 增量逐 chunk 回调（kind 区分）
|
||
- tool_calls 分片累积（arguments 按 index 拼接），流结束后统一执行
|
||
- 截图（de_screenshot）图像转 image_url 追加下一轮，同时 on_tool 回调带缩略
|
||
"""
|
||
import base64
|
||
import json
|
||
import logging
|
||
|
||
import httpx
|
||
from fastmcp import Client
|
||
|
||
from mcp_agent.config import AgentSettings
|
||
|
||
_log = logging.getLogger("agent")
|
||
|
||
S = AgentSettings()
|
||
|
||
SYSTEM_PROMPT = """你是手机自动化控制助手。你通过工具实时操作 Android 手机。
|
||
|
||
工作规范：
|
||
1. 先 de_list_devices 确定目标设备（在线才可操作）
|
||
2. 观察屏幕：先 de_screenshot 获取截图（图像会随后给你），基于截图理解当前界面
|
||
3. 操作：de_tap/de_swipe 的坐标必须与最近一次 de_screenshot 图像一致（直接看图给坐标，服务器自动换算）
|
||
4. 元素操作优先：能用 de_ui_tree/de_tap_element（text/id/desc 定位）就不用裸坐标
|
||
5. 每次关键操作后再次 de_screenshot 验证结果，直到完成用户目标
|
||
6. 完成或失败时用中文总结：做了什么、当前状态、需要用户注意的事项
|
||
7. 设备不可用/操作失败时如实报告错误，不要臆测成功
|
||
|
||
可用工具清单将由系统提供。"""
|
||
|
||
|
||
class Agent:
|
||
def __init__(self, settings: AgentSettings = None):
|
||
self.s = settings or S
|
||
self.tools_schema = [] # OpenAI function schema
|
||
self.messages = []
|
||
# 回调（Web 展示用，均可选）：
|
||
# on_delta(text, kind) kind: content | reasoning —— 流式文本增量
|
||
# on_tool(step) step: {tool, args, result, image} —— 工具调用完成
|
||
self.on_delta = None
|
||
self.on_tool = None
|
||
self._mcp = None
|
||
|
||
# ---------- MCP 工具桥 ----------
|
||
async def _load_tools(self):
|
||
"""从 MCP Server 拉工具，转 OpenAI function schema。"""
|
||
# 短连接超时：MCP 不可用时快速失败（默认会无限重试卡死线程）
|
||
self._mcp = Client(self.s.mcp_url, timeout=10.0, init_timeout=10.0)
|
||
await self._mcp.__aenter__()
|
||
tools = await self._mcp.list_tools()
|
||
self.tools_schema = []
|
||
for t in tools:
|
||
# MCP SDK v2 改名 input_schema，兼容新旧字段
|
||
schema = getattr(t, "input_schema", None) or getattr(t, "inputSchema", {})
|
||
name = getattr(t, "name", "")
|
||
desc = getattr(t, "description", "") or ""
|
||
self.tools_schema.append({
|
||
"type": "function",
|
||
"function": {"name": name, "description": desc,
|
||
"parameters": schema}})
|
||
_log.info("MCP 工具已加载: %s", [s["function"]["name"] for s in self.tools_schema])
|
||
|
||
async def close(self):
|
||
if self._mcp:
|
||
try:
|
||
await self._mcp.__aexit__(None, None, None)
|
||
except Exception:
|
||
pass
|
||
|
||
# ---------- 模型调用（流式） ----------
|
||
async def _chat_stream(self):
|
||
"""流式 chat/completions：逐 chunk 产出 JSON（async generator）。"""
|
||
body = {
|
||
"model": self.s.model,
|
||
"messages": self.messages,
|
||
"tools": self.tools_schema if self.tools_schema else None,
|
||
"max_tokens": 4096,
|
||
"stream": True,
|
||
}
|
||
headers = {"Authorization": f"Bearer {self.s.api_key}",
|
||
"Content-Type": "application/json"}
|
||
url = f"{self.s.api_base.rstrip('/')}/chat/completions"
|
||
async with httpx.AsyncClient(timeout=self.s.request_timeout) as client:
|
||
async with client.stream("POST", url, json=body, headers=headers) as r:
|
||
if r.status_code != 200:
|
||
text = (await r.aread()).decode(errors="replace")
|
||
raise RuntimeError(f"模型 API HTTP {r.status_code}: {text[:300]}")
|
||
async for line in r.aiter_lines():
|
||
if not line.startswith("data:"):
|
||
continue
|
||
data = line[5:].strip()
|
||
if data == "[DONE]":
|
||
break
|
||
try:
|
||
yield json.loads(data)
|
||
except json.JSONDecodeError:
|
||
continue
|
||
|
||
# ---------- 工具执行 ----------
|
||
async def _execute_tool(self, name, arguments):
|
||
"""执行 MCP 工具，返回 (文本结果, image_data_or_None)。"""
|
||
args = json.loads(arguments) if isinstance(arguments, str) else (arguments or {})
|
||
_log.info("执行工具 %s %s", name, args)
|
||
try:
|
||
result = await self._mcp.call_tool(name, args)
|
||
data = getattr(result, "data", result)
|
||
except Exception as e:
|
||
return {"ok": False, "error": f"工具执行失败: {e}"}, None
|
||
# de_screenshot：图像分离（作为 image_url 追加给模型看 + on_tool 缩略展示）
|
||
image_b64 = None
|
||
text_result = data
|
||
if name == "de_screenshot" and isinstance(data, dict) and data.get("ok"):
|
||
img = (data.get("data") or {}).get("image") or {}
|
||
if img.get("data"):
|
||
text_result = {k: v for k, v in (data.get("data") or {}).items()
|
||
if k != "image"}
|
||
image_b64 = img["data"]
|
||
if self.on_tool:
|
||
try:
|
||
self.on_tool({"tool": name, "args": args,
|
||
"result": text_result, "image": image_b64})
|
||
except Exception:
|
||
pass
|
||
return text_result, image_b64
|
||
|
||
# ---------- 主循环（流式） ----------
|
||
async def run_stream(self, prompt: str, serial: str = "",
|
||
on_delta=None, on_tool=None):
|
||
"""流式执行一轮指令，返回最终完整文本。
|
||
|
||
on_delta(text, kind)：content/reasoning 文本增量（实时推给前端）
|
||
on_tool(step)：工具调用完成（实时显示 MCP 步骤）
|
||
"""
|
||
self.on_delta = on_delta
|
||
self.on_tool = on_tool
|
||
target = serial or self.s.default_serial
|
||
sys_txt = SYSTEM_PROMPT
|
||
if target:
|
||
sys_txt += f"\n\n本次默认目标设备 serial：{target}（未指定设备时用它）。"
|
||
self.messages = [{"role": "system", "content": sys_txt},
|
||
{"role": "user", "content": prompt}]
|
||
|
||
for _step in range(self.s.max_steps):
|
||
content_parts = []
|
||
tool_acc = {} # index -> {id, name, args}
|
||
has_tool = False
|
||
async for chunk in self._chat_stream():
|
||
choice = (chunk.get("choices") or [{}])[0]
|
||
delta = choice.get("delta") or {}
|
||
text = delta.get("content")
|
||
if text:
|
||
content_parts.append(text)
|
||
if on_delta:
|
||
on_delta(text, "content")
|
||
rtext = delta.get("reasoning_content")
|
||
if rtext:
|
||
if on_delta:
|
||
on_delta(rtext, "reasoning")
|
||
for tc in delta.get("tool_calls") or []:
|
||
has_tool = True
|
||
idx = tc.get("index", 0)
|
||
acc = tool_acc.setdefault(idx, {"id": "", "name": "", "args": ""})
|
||
if tc.get("id"):
|
||
acc["id"] = tc["id"]
|
||
fn = tc.get("function") or {}
|
||
if fn.get("name"):
|
||
acc["name"] += fn["name"]
|
||
if fn.get("arguments"):
|
||
acc["args"] += fn["arguments"]
|
||
|
||
full_content = "".join(content_parts)
|
||
|
||
if has_tool:
|
||
# 组装 assistant 消息（含 tool_calls）并执行工具
|
||
tcs = []
|
||
for idx in sorted(tool_acc):
|
||
acc = tool_acc[idx]
|
||
tcs.append({"id": acc["id"] or f"call_{idx}",
|
||
"type": "function",
|
||
"function": {"name": acc["name"],
|
||
"arguments": acc["args"]}})
|
||
self.messages.append({"role": "assistant",
|
||
"content": full_content,
|
||
"tool_calls": tcs})
|
||
for tc in tcs:
|
||
fn = tc["function"]
|
||
text_result, image_b64 = await self._execute_tool(
|
||
fn["name"], fn["arguments"])
|
||
self.messages.append({
|
||
"role": "tool", "tool_call_id": tc["id"],
|
||
"content": json.dumps(text_result, ensure_ascii=False)[:4000]})
|
||
if image_b64:
|
||
self.messages.append({
|
||
"role": "user",
|
||
"content": [{"type": "text",
|
||
"text": "这是最新屏幕截图，请基于它继续判断"},
|
||
{"type": "image_url",
|
||
"image_url": {"url":
|
||
f"data:image/jpeg;base64,{image_b64}"}}]})
|
||
continue
|
||
|
||
# 无工具调用：本轮即最终回答
|
||
return full_content
|
||
|
||
return "（达到最大步骤数未完成，请检查操作是否卡在循环）"
|
||
|
||
# ---------- 非流式（CLI） ----------
|
||
async def run(self, prompt: str, serial: str = "") -> str:
|
||
"""非流式执行，返回最终文本（CLI 用，内部走流式收集）。"""
|
||
return await self.run_stream(prompt, serial)
|