wehub-resource-sync
143 lines
5.3 KiB
Python
143 lines
5.3 KiB
Python
"""`/btw` 命令：side question — 不打断主 Agent 的临时 subagent 问答。
|
||
|
||
- 持锁 deepcopy backend.history → 后台线程 backend.raw_ask 单次拉答
|
||
- 主 agent backend.history 零写入；不入 task_queue
|
||
- 答案 → display_queue 'done'（install 路径）或同步 return（frontend 路径）
|
||
|
||
复用 backend.raw_ask + make_messages，不新建 LLM 实例。
|
||
"""
|
||
from __future__ import annotations
|
||
import copy, os, threading, time
|
||
from typing import Optional
|
||
|
||
|
||
_WRAPPER_ZH = """<system-reminder>
|
||
这是用户的临时插问 (side question)。主 agent 仍在后台运行，**不会被打断**。
|
||
|
||
身份与边界：
|
||
- 你是一个独立的轻量 sub-agent
|
||
- 上下文里能看到主 agent 与用户的完整对话、最近的工具调用与结果
|
||
- 用户在问当前进展或顺便确认某事——基于已有信息**一次性**作答
|
||
- 没有任何工具可用：不要"让我查一下" / "我去试试" / 任何承诺动作
|
||
- 信息不足就坦白说"基于目前对话我不知道"
|
||
|
||
侧问内容如下：
|
||
</system-reminder>
|
||
|
||
{question}"""
|
||
|
||
_WRAPPER_EN = """<system-reminder>
|
||
This is a side question from the user. The main agent is NOT interrupted — it continues in the background.
|
||
|
||
Identity & boundaries:
|
||
- You are an independent lightweight sub-agent
|
||
- You can see the full conversation between the main agent and the user, plus recent tool calls/results
|
||
- The user is asking about current progress or a quick aside — answer in **one shot** from existing info
|
||
- You have NO tools — never say "let me check" / "I'll try" / any action promise
|
||
- If info is missing, just say "based on the conversation I don't know"
|
||
|
||
Question:
|
||
</system-reminder>
|
||
|
||
{question}"""
|
||
|
||
_TIMEOUT_SEC = 120
|
||
|
||
|
||
def _wrapper(): return _WRAPPER_EN if os.environ.get('GA_LANG') == 'en' else _WRAPPER_ZH
|
||
|
||
|
||
def _strip_cmd(query):
|
||
s = (query or '').strip()
|
||
return s[len('/btw'):].strip() if s.startswith('/btw') else s
|
||
|
||
|
||
def _help_text():
|
||
return ('**/btw 用法**：side question — 临时问主 agent 当前进展，不打断主线\n\n'
|
||
'`/btw <你的问题>`\n\n'
|
||
'行为：抓取当前对话上下文 → 单轮纯文本作答（无工具）→ 主 agent 历史不变。')
|
||
|
||
|
||
def _snapshot_history(backend):
|
||
"""Lock + deepcopy: defends against concurrent compress_history_tags mutating inner blocks."""
|
||
with backend.lock:
|
||
return copy.deepcopy(list(backend.history))
|
||
|
||
|
||
def _build_wire(backend, history, sidequest_msg):
|
||
"""history + sidequest → wire-format. Dispatches: BaseSession subclasses → make_messages,
|
||
Native* → raw pairs (raw_ask runs _fix/_drop/_ensure transforms itself)."""
|
||
msgs = history + [sidequest_msg]
|
||
if hasattr(backend, 'make_messages'):
|
||
return backend.make_messages(msgs)
|
||
return [{"role": m["role"], "content": list(m.get("content", []))} for m in msgs]
|
||
|
||
|
||
def _ask(agent, question, deadline):
|
||
"""One-shot raw_ask against current backend; never mutates backend.history."""
|
||
backend = agent.llmclient.backend
|
||
user_msg = {"role": "user",
|
||
"content": [{"type": "text", "text": _wrapper().format(question=question)}]}
|
||
wire = _build_wire(backend, _snapshot_history(backend), user_msg)
|
||
text = ''
|
||
for chunk in backend.raw_ask(wire):
|
||
text += chunk
|
||
if time.time() > deadline:
|
||
return text + '\n\n⚠️ /btw 超时，仅返回部分回复。'
|
||
return text
|
||
|
||
|
||
def _format(question, body, took):
|
||
head = f'> 🟡 /btw {question}\n\n'
|
||
return head + (body.strip() or '*(空回复)*') + f'\n\n*({took:.1f}s)*'
|
||
|
||
|
||
def _run(agent, question, deadline):
|
||
"""Catches errors at the boundary so neither caller path needs its own try/except."""
|
||
try: return _ask(agent, question, deadline)
|
||
except Exception as e: return f'❌ /btw 失败: {type(e).__name__}: {e}'
|
||
|
||
|
||
def handle(agent, query, display_queue) -> Optional[str]:
|
||
"""Slash-cmd entry (server-side, install path). Spawn worker; return None to consume."""
|
||
question = _strip_cmd(query)
|
||
if not question or question in ('help', '?', '-h', '--help'):
|
||
display_queue.put({'done': _help_text(), 'source': 'system'})
|
||
return None
|
||
started = time.time()
|
||
deadline = started + _TIMEOUT_SEC
|
||
|
||
def worker():
|
||
body = _run(agent, question, deadline)
|
||
display_queue.put({'done': _format(question, body, time.time() - started), 'source': 'system'})
|
||
|
||
threading.Thread(target=worker, daemon=True, name='btw-sidequest').start()
|
||
return None
|
||
|
||
|
||
def handle_frontend_command(agent, query) -> str:
|
||
"""Sync entry for frontends wanting a string back (tg/wx/stapp/...)."""
|
||
question = _strip_cmd(query)
|
||
if not question or question in ('help', '?', '-h', '--help'):
|
||
return _help_text()
|
||
started = time.time()
|
||
body = _run(agent, question, started + _TIMEOUT_SEC)
|
||
return _format(question, body, time.time() - started)
|
||
|
||
|
||
def install(cls):
|
||
"""Idempotent monkey-patch: intercept /btw before original dispatch."""
|
||
orig = cls._handle_slash_cmd
|
||
if getattr(orig, '_btw_patched', False): return
|
||
|
||
def patched(self, raw_query, display_queue):
|
||
s = (raw_query or '').strip()
|
||
if s == '/btw' or s.startswith('/btw ') or s.startswith('/btw\t'):
|
||
r = handle(self, raw_query, display_queue)
|
||
if r is None: return None
|
||
return r
|
||
return orig(self, raw_query, display_queue)
|
||
|
||
patched._btw_patched = True
|
||
cls._handle_slash_cmd = patched
|