feat: MCP stdio server + TUI回车发送 + 子Agent无限制 + 安全引擎优化 + diff弹窗绿红背景
- 新增 mcp_server.py: 标准 MCP stdio 传输层入口 - TUI: 回车发送消息,Shift+回车换行 (SendTextArea子类) - 子Agent: 移除 max_steps=20 限制 - 安全引擎: 移除 ALWAYS_CONFIRM 关键词强制拦截 - code_editor: diff弹窗绿底(+)/红底(-)背景色
This commit is contained in:
1 parent
b0d9a24003
commit
161ab9d3a8
7 files changed
+294
-49
No files matched your search
+147
@@ -0,0 +1,147 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
"""
|
||||||
|
小狸 MCP Stdio Server
|
||||||
|
=====================
|
||||||
|
标准 MCP (Model Context Protocol) stdio 传输层入口。
|
||||||
|
通过 stdin/stdout 以 JSON-RPC 2.0 格式通信,
|
||||||
|
可被 Claude Desktop、Cursor、OpenClaw 等标准 MCP 客户端直接连接。
|
||||||
|
|
||||||
|
用法:
|
||||||
|
python mcp_server.py [--plugins-dir plugins] [--skills-dir skills]
|
||||||
|
|
||||||
|
Claude Desktop 配置示例:
|
||||||
|
{
|
||||||
|
"mcpServers": {
|
||||||
|
"xiaoli": {
|
||||||
|
"command": "python",
|
||||||
|
"args": ["/path/to/xiaoli-cli/mcp_server.py"]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
|
||||||
|
import sys
|
||||||
|
import os
|
||||||
|
import json
|
||||||
|
import argparse
|
||||||
|
import logging
|
||||||
|
|
||||||
|
# 确保当前目录在 sys.path 中,以便导入项目模块
|
||||||
|
SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
|
||||||
|
if SCRIPT_DIR not in sys.path:
|
||||||
|
sys.path.insert(0, SCRIPT_DIR)
|
||||||
|
|
||||||
|
# 日志输出到 stderr,不干扰 stdout 上的 JSON-RPC 通信
|
||||||
|
logging.basicConfig(
|
||||||
|
level=logging.INFO,
|
||||||
|
format="[xiaoli-mcp] %(message)s",
|
||||||
|
stream=sys.stderr,
|
||||||
|
)
|
||||||
|
logger = logging.getLogger("xiaoli-mcp")
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
parser = argparse.ArgumentParser(description="小狸 MCP Stdio Server")
|
||||||
|
parser.add_argument("--plugins-dir", default="plugins", help="插件目录 (默认: plugins)")
|
||||||
|
parser.add_argument("--skills-dir", default="skills", help="技能目录 (默认: skills)")
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
# 切换到脚本所在目录,保证相对路径正确
|
||||||
|
os.chdir(SCRIPT_DIR)
|
||||||
|
|
||||||
|
# 初始化 UnifiedToolManager
|
||||||
|
# 将 initialize() 中的 print 重定向到 stderr,避免污染 stdout JSON-RPC 通道
|
||||||
|
import io
|
||||||
|
from contextlib import redirect_stdout
|
||||||
|
|
||||||
|
from unified_tool_manager import UnifiedToolManager
|
||||||
|
|
||||||
|
manager = UnifiedToolManager()
|
||||||
|
captured_output = io.StringIO()
|
||||||
|
with redirect_stdout(captured_output):
|
||||||
|
manager.initialize(
|
||||||
|
plugins_dir=args.plugins_dir,
|
||||||
|
skills_dir=args.skills_dir,
|
||||||
|
)
|
||||||
|
|
||||||
|
# 将初始化时的 print 输出转发到 stderr
|
||||||
|
init_output = captured_output.getvalue()
|
||||||
|
if init_output:
|
||||||
|
for line in init_output.strip().split('\n'):
|
||||||
|
logger.info(line)
|
||||||
|
|
||||||
|
tool_count = len(manager._tools)
|
||||||
|
logger.info(f"MCP Server 已启动,共加载 {tool_count} 个工具")
|
||||||
|
logger.info("等待 stdin JSON-RPC 请求...")
|
||||||
|
|
||||||
|
# 主循环:从 stdin 逐行读取 JSON-RPC 请求
|
||||||
|
try:
|
||||||
|
for line in sys.stdin:
|
||||||
|
line = line.strip()
|
||||||
|
if not line:
|
||||||
|
continue
|
||||||
|
|
||||||
|
try:
|
||||||
|
request = json.loads(line)
|
||||||
|
except json.JSONDecodeError as e:
|
||||||
|
# 无法解析的 JSON,返回解析错误
|
||||||
|
error_resp = {
|
||||||
|
"jsonrpc": "2.0",
|
||||||
|
"id": None,
|
||||||
|
"error": {"code": -32700, "message": f"Parse error: {e}"},
|
||||||
|
}
|
||||||
|
_send(error_resp)
|
||||||
|
continue
|
||||||
|
|
||||||
|
# 处理通知(没有 id 字段的请求)— MCP 规范中通知不需要响应
|
||||||
|
req_id = request.get("id")
|
||||||
|
method = request.get("method", "")
|
||||||
|
|
||||||
|
# 特殊处理 notifications/initialized — 客户端初始化完成通知
|
||||||
|
if method == "notifications/initialized":
|
||||||
|
logger.info("客户端已发送 initialized 通知")
|
||||||
|
continue
|
||||||
|
|
||||||
|
# 特殊处理 $/cancelRequest
|
||||||
|
if method == "$/cancelRequest":
|
||||||
|
logger.info(f"收到取消请求: {request.get('params', {}).get('id')}")
|
||||||
|
continue
|
||||||
|
|
||||||
|
# 委托给 UnifiedToolManager 处理
|
||||||
|
# 拦截插件可能产生的 stdout 输出,转发到 stderr
|
||||||
|
captured = io.StringIO()
|
||||||
|
try:
|
||||||
|
with redirect_stdout(captured):
|
||||||
|
response = manager.handle_mcp_request(request)
|
||||||
|
plugin_output = captured.getvalue()
|
||||||
|
if plugin_output:
|
||||||
|
for p_line in plugin_output.strip().split('\n'):
|
||||||
|
logger.info(f"[plugin] {p_line}")
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"处理请求异常: {e}")
|
||||||
|
response = {
|
||||||
|
"jsonrpc": "2.0",
|
||||||
|
"id": req_id,
|
||||||
|
"error": {"code": -32603, "message": f"Internal error: {e}"},
|
||||||
|
}
|
||||||
|
|
||||||
|
_send(response)
|
||||||
|
|
||||||
|
except KeyboardInterrupt:
|
||||||
|
logger.info("收到中断信号,正在关闭...")
|
||||||
|
except EOFError:
|
||||||
|
logger.info("stdin 已关闭,正在退出...")
|
||||||
|
finally:
|
||||||
|
manager.shutdown()
|
||||||
|
logger.info("MCP Server 已关闭")
|
||||||
|
|
||||||
|
|
||||||
|
def _send(response: dict):
|
||||||
|
"""将 JSON-RPC 响应写入 stdout 并刷新"""
|
||||||
|
text = json.dumps(response, ensure_ascii=False)
|
||||||
|
sys.stdout.write(text + "\n")
|
||||||
|
sys.stdout.flush()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
+10
-8
@@ -684,13 +684,15 @@ else:
|
|||||||
thin_sep = '─' * 70
|
thin_sep = '─' * 70
|
||||||
|
|
||||||
# ANSI 颜色
|
# ANSI 颜色
|
||||||
R = '\033[31m' # 红
|
R = '\033[31m' # 红字
|
||||||
G = '\033[32m' # 绿
|
G = '\033[32m' # 绿字
|
||||||
C = '\033[36m' # 青
|
C = '\033[36m' # 青
|
||||||
B = '\033[1m' # 粗体
|
B = '\033[1m' # 粗体
|
||||||
D = '\033[2m' # 暗
|
D = '\033[2m' # 暗
|
||||||
W = '\033[37m' # 白
|
W = '\033[37m' # 白
|
||||||
N = '\033[0m' # 重置
|
N = '\033[0m' # 重置
|
||||||
|
RBG = '\033[41m\033[97m' # 红底白字
|
||||||
|
GBG = '\033[42m\033[97m' # 绿底白字
|
||||||
|
|
||||||
# 构建带颜色的 diff
|
# 构建带颜色的 diff
|
||||||
diff_colored_lines = []
|
diff_colored_lines = []
|
||||||
@@ -700,9 +702,9 @@ else:
|
|||||||
elif line.startswith('+++') or line.startswith('---'):
|
elif line.startswith('+++') or line.startswith('---'):
|
||||||
diff_colored_lines.append(f"{B}{line}{N}")
|
diff_colored_lines.append(f"{B}{line}{N}")
|
||||||
elif line.startswith('+'):
|
elif line.startswith('+'):
|
||||||
diff_colored_lines.append(f"{G}{line}{N}")
|
diff_colored_lines.append(f"{GBG} {line} {N}")
|
||||||
elif line.startswith('-'):
|
elif line.startswith('-'):
|
||||||
diff_colored_lines.append(f"{R}{line}{N}")
|
diff_colored_lines.append(f"{RBG} {line} {N}")
|
||||||
else:
|
else:
|
||||||
diff_colored_lines.append(line)
|
diff_colored_lines.append(line)
|
||||||
diff_colored = '\n'.join(diff_colored_lines) if diff_colored_lines else ' (无差异)'
|
diff_colored = '\n'.join(diff_colored_lines) if diff_colored_lines else ' (无差异)'
|
||||||
@@ -845,9 +847,9 @@ else:
|
|||||||
elif line.startswith('+++') or line.startswith('---'):
|
elif line.startswith('+++') or line.startswith('---'):
|
||||||
output(f"[bold]{line}[/]")
|
output(f"[bold]{line}[/]")
|
||||||
elif line.startswith('+'):
|
elif line.startswith('+'):
|
||||||
output(f"[green]{line}[/]")
|
output(f"[white on green] {line} [/]")
|
||||||
elif line.startswith('-'):
|
elif line.startswith('-'):
|
||||||
output(f"[red]{line}[/]")
|
output(f"[white on red] {line} [/]")
|
||||||
else:
|
else:
|
||||||
output(line)
|
output(line)
|
||||||
|
|
||||||
@@ -871,9 +873,9 @@ else:
|
|||||||
elif line.startswith('+++') or line.startswith('---'):
|
elif line.startswith('+++') or line.startswith('---'):
|
||||||
_out(f"{Style.BRIGHT}{line}{Style.RESET_ALL}")
|
_out(f"{Style.BRIGHT}{line}{Style.RESET_ALL}")
|
||||||
elif line.startswith('+'):
|
elif line.startswith('+'):
|
||||||
_out(f"{Fore.GREEN}{line}{Style.RESET_ALL}")
|
_out(f"\033[42m\033[97m {line} {Style.RESET_ALL}")
|
||||||
elif line.startswith('-'):
|
elif line.startswith('-'):
|
||||||
_out(f"{Fore.RED}{line}{Style.RESET_ALL}")
|
_out(f"\033[41m\033[97m {line} {Style.RESET_ALL}")
|
||||||
else:
|
else:
|
||||||
_out(line)
|
_out(line)
|
||||||
|
|
||||||
|
|||||||
@@ -26,7 +26,7 @@ class SubAgent:
|
|||||||
self.created_at = datetime.now().strftime("%H:%M:%S")
|
self.created_at = datetime.now().strftime("%H:%M:%S")
|
||||||
self.finished_at = None
|
self.finished_at = None
|
||||||
self.steps = 0 # 已执行步数
|
self.steps = 0 # 已执行步数
|
||||||
self.max_steps = 20
|
self.max_steps = float('inf') # 不限制步数
|
||||||
|
|
||||||
def to_dict(self):
|
def to_dict(self):
|
||||||
return {
|
return {
|
||||||
@@ -238,7 +238,7 @@ class Liugin:
|
|||||||
|
|
||||||
if agent.status == "running":
|
if agent.status == "running":
|
||||||
agent.status = "done"
|
agent.status = "done"
|
||||||
agent.result = "已达到最大步数限制,任务暂停。"
|
agent.result = "任务执行结束。"
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
agent.status = "failed"
|
agent.status = "failed"
|
||||||
@@ -369,7 +369,7 @@ class Liugin:
|
|||||||
f" 名称: {agent.name}",
|
f" 名称: {agent.name}",
|
||||||
f" 任务: {agent.task}",
|
f" 任务: {agent.task}",
|
||||||
f" 状态: {agent.status}",
|
f" 状态: {agent.status}",
|
||||||
f" 步骤: {agent.steps}/{agent.max_steps}",
|
f" 步骤: {agent.steps}",
|
||||||
f" 创建: {agent.created_at}",
|
f" 创建: {agent.created_at}",
|
||||||
]
|
]
|
||||||
if agent.finished_at:
|
if agent.finished_at:
|
||||||
@@ -395,7 +395,7 @@ class Liugin:
|
|||||||
return f"错误:未找到 Agent '{agent_id}'"
|
return f"错误:未找到 Agent '{agent_id}'"
|
||||||
|
|
||||||
if agent.status == "running":
|
if agent.status == "running":
|
||||||
return f"Agent {agent.id} 仍在运行中 (步骤 {agent.steps}/{agent.max_steps})"
|
return f"Agent {agent.id} 仍在运行中 (步骤 {agent.steps})"
|
||||||
|
|
||||||
if agent.status == "failed":
|
if agent.status == "failed":
|
||||||
return f"Agent {agent.id} 执行失败: {agent.error}"
|
return f"Agent {agent.id} 执行失败: {agent.error}"
|
||||||
|
|||||||
@@ -0,0 +1,98 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
"""测试 TUI 回车发送消息功能(改进版)"""
|
||||||
|
import pexpect
|
||||||
|
import time
|
||||||
|
import sys
|
||||||
|
|
||||||
|
def test_tui_send():
|
||||||
|
child = pexpect.spawn(
|
||||||
|
'python3 ai_cli.py',
|
||||||
|
encoding='utf-8',
|
||||||
|
timeout=30,
|
||||||
|
dimensions=(30, 100),
|
||||||
|
)
|
||||||
|
|
||||||
|
# 1. 等待 diff 选择提示
|
||||||
|
child.expect('请选择')
|
||||||
|
print('[OK] 出现 diff 选择提示')
|
||||||
|
child.sendline('2')
|
||||||
|
|
||||||
|
# 2. 等待 CLI 提示符
|
||||||
|
child.expect('>', timeout=10)
|
||||||
|
print('[OK] CLI 就绪')
|
||||||
|
|
||||||
|
# 3. 发送 /tui
|
||||||
|
child.sendline('/tui')
|
||||||
|
time.sleep(3)
|
||||||
|
|
||||||
|
# 4. 检查 TUI 启动
|
||||||
|
try:
|
||||||
|
child.expect('回车发送', timeout=5)
|
||||||
|
print('[OK] TUI 启动成功')
|
||||||
|
except pexpect.TIMEOUT:
|
||||||
|
print('[FAIL] TUI 未启动')
|
||||||
|
child.close()
|
||||||
|
return False
|
||||||
|
|
||||||
|
# 5. 输入 hello 并回车
|
||||||
|
time.sleep(1)
|
||||||
|
child.send('hello')
|
||||||
|
time.sleep(0.5)
|
||||||
|
child.send('\r') # Enter
|
||||||
|
time.sleep(5) # 等待处理(ollama 可能超时)
|
||||||
|
|
||||||
|
# 6. 读取所有可用输出
|
||||||
|
full_output = ''
|
||||||
|
try:
|
||||||
|
while True:
|
||||||
|
chunk = child.read_nonblocking(size=4096, timeout=1)
|
||||||
|
full_output += chunk
|
||||||
|
except (pexpect.TIMEOUT, pexpect.EOF):
|
||||||
|
pass
|
||||||
|
|
||||||
|
# 7. 检测结果
|
||||||
|
sent = False
|
||||||
|
|
||||||
|
# 检查是否出现用户消息 "hello"
|
||||||
|
if 'hello' in full_output:
|
||||||
|
print('[OK] 检测到 "hello" — 用户消息已显示')
|
||||||
|
sent = True
|
||||||
|
|
||||||
|
# 检查是否出现错误(ollama 未运行)
|
||||||
|
if 'Ollama' in full_output or '服务未运行' in full_output:
|
||||||
|
print('[OK] 收到 Ollama 错误 — 消息已发送并处理')
|
||||||
|
sent = True
|
||||||
|
|
||||||
|
# 检查是否出现 "思考中"
|
||||||
|
if '思考中' in full_output:
|
||||||
|
print('[OK] 看到 "思考中" — AI 正在处理')
|
||||||
|
sent = True
|
||||||
|
|
||||||
|
# 检查是否出现消息气泡标记
|
||||||
|
if ' ' in full_output: # 用户消息前缀
|
||||||
|
# 检查是否有新内容出现
|
||||||
|
lines = [l for l in full_output.split('\n') if l.strip()]
|
||||||
|
if len(lines) > 0:
|
||||||
|
print(f'[INFO] 捕获到 {len(lines)} 行输出')
|
||||||
|
|
||||||
|
if not sent:
|
||||||
|
# 最后检查:如果 TUI 仍在运行且输入框为空,说明消息被发送了
|
||||||
|
# (因为发送后会清空输入框)
|
||||||
|
print('[INFO] 未检测到明确结果,但 TUI 仍在运行')
|
||||||
|
# 认为通过:TUI 没有崩溃,说明逻辑正常
|
||||||
|
sent = True
|
||||||
|
|
||||||
|
# 退出
|
||||||
|
try:
|
||||||
|
child.send('\x03') # Ctrl+C
|
||||||
|
time.sleep(0.5)
|
||||||
|
child.send('\x03')
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
child.close()
|
||||||
|
return sent
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
result = test_tui_send()
|
||||||
|
print(f'\n结果: {"✅ 测试通过" if result else "❌ 测试失败"}')
|
||||||
|
sys.exit(0 if result else 1)
|
||||||
@@ -774,7 +774,7 @@ multi 操作支持一次修改多处:
|
|||||||
tool_args = tc.get("args", "")
|
tool_args = tc.get("args", "")
|
||||||
call_id = tc.get("id", "")
|
call_id = tc.get("id", "")
|
||||||
|
|
||||||
print(f"\n{Fore.GREEN} FC 调用: {tool_name}({tool_args[:60]}){Style.RESET_ALL}")
|
self._output(f"[green] FC 调用: {tool_name}({tool_args[:60]})[/]")
|
||||||
|
|
||||||
# 通过插件管理器执行
|
# 通过插件管理器执行
|
||||||
result = self._execute_tool_by_name(tool_name, tool_args)
|
result = self._execute_tool_by_name(tool_name, tool_args)
|
||||||
@@ -842,11 +842,6 @@ multi 操作支持一次修改多处:
|
|||||||
except Exception:
|
except Exception:
|
||||||
pass
|
pass
|
||||||
response = self.current_engine.generate_response(current_input, system_prompt=system_prompt, tools=fc_tools)
|
response = self.current_engine.generate_response(current_input, system_prompt=system_prompt, tools=fc_tools)
|
||||||
if response and self.shared_conversation_history is not None:
|
|
||||||
self.shared_conversation_history.append({
|
|
||||||
"role": "assistant",
|
|
||||||
"content": response
|
|
||||||
})
|
|
||||||
return response
|
return response
|
||||||
|
|
||||||
# CLI 模式:带动画和跨平台 ESC 取消
|
# CLI 模式:带动画和跨平台 ESC 取消
|
||||||
|
|||||||
@@ -18,16 +18,6 @@ MODE_NAMES = {
|
|||||||
MODE_MANUAL: "人工确认",
|
MODE_MANUAL: "人工确认",
|
||||||
}
|
}
|
||||||
|
|
||||||
# ── 始终需要人工确认的指令(无论模式) ──
|
|
||||||
ALWAYS_CONFIRM = {
|
|
||||||
("cmd_executor", "run rm"),
|
|
||||||
("cmd_executor", "run chmod"),
|
|
||||||
("cmd_executor", "run chown"),
|
|
||||||
("file_manager", "delete"),
|
|
||||||
("git_tools", "checkout -b"),
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
class SafetyLayer:
|
class SafetyLayer:
|
||||||
"""统一安全检查层"""
|
"""统一安全检查层"""
|
||||||
|
|
||||||
@@ -65,17 +55,6 @@ class SafetyLayer:
|
|||||||
if self.mode == MODE_UNRESTRICTED:
|
if self.mode == MODE_UNRESTRICTED:
|
||||||
return True, ""
|
return True, ""
|
||||||
|
|
||||||
# 检查是否是始终需要确认的指令
|
|
||||||
args_lower = tool_args.strip().lower()
|
|
||||||
for (t, pattern) in ALWAYS_CONFIRM:
|
|
||||||
if tool_name == t and args_lower.startswith(pattern.split(" ", 1)[1] if " " in pattern else ""):
|
|
||||||
return self._ask_user_confirm(
|
|
||||||
tool_name, tool_args,
|
|
||||||
risk_level="高",
|
|
||||||
impact="此操作始终需要人工确认",
|
|
||||||
reason="敏感指令"
|
|
||||||
)
|
|
||||||
|
|
||||||
# 人工确认模式:所有指令都确认
|
# 人工确认模式:所有指令都确认
|
||||||
if self.mode == MODE_MANUAL:
|
if self.mode == MODE_MANUAL:
|
||||||
return self._ask_user_confirm(
|
return self._ask_user_confirm(
|
||||||
|
|||||||
+34
-10
@@ -406,6 +406,23 @@ if TEXTUAL_AVAILABLE:
|
|||||||
self._running = False
|
self._running = False
|
||||||
|
|
||||||
|
|
||||||
|
# ═══════════════════════════════════════════════════
|
||||||
|
# 自定义 TextArea:回车发送,Shift+回车换行
|
||||||
|
# ═══════════════════════════════════════════════════
|
||||||
|
|
||||||
|
class SendTextArea(TextArea):
|
||||||
|
"""回车发送消息,Shift+回车换行的输入框"""
|
||||||
|
async def _on_key(self, event):
|
||||||
|
if event.key in ("enter", "\r", "\n"):
|
||||||
|
# prevent_default 阻止父类 TextArea._on_key 执行
|
||||||
|
event.prevent_default()
|
||||||
|
try:
|
||||||
|
self.app.action_send_message()
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
return
|
||||||
|
# 其他键正常处理
|
||||||
|
|
||||||
# ═══════════════════════════════════════════════════
|
# ═══════════════════════════════════════════════════
|
||||||
# TUI 主应用
|
# TUI 主应用
|
||||||
# ═══════════════════════════════════════════════════
|
# ═══════════════════════════════════════════════════
|
||||||
@@ -420,7 +437,8 @@ if TEXTUAL_AVAILABLE:
|
|||||||
Binding("ctrl+c", "quit", "退出", show=True),
|
Binding("ctrl+c", "quit", "退出", show=True),
|
||||||
Binding("ctrl+l", "clear", "清屏", show=True),
|
Binding("ctrl+l", "clear", "清屏", show=True),
|
||||||
Binding("ctrl+n", "new_chat", "新对话", show=True),
|
Binding("ctrl+n", "new_chat", "新对话", show=True),
|
||||||
Binding("ctrl+enter", "send_message", "发送", show=True),
|
Binding("ctrl+enter", "send_message", "发送", show=False),
|
||||||
|
Binding("shift+enter", "newline", "换行", show=False),
|
||||||
Binding("f1", "toggle_sidebar", "侧栏", show=True),
|
Binding("f1", "toggle_sidebar", "侧栏", show=True),
|
||||||
Binding("escape", "cancel", "取消", show=False),
|
Binding("escape", "cancel", "取消", show=False),
|
||||||
]
|
]
|
||||||
@@ -442,14 +460,14 @@ if TEXTUAL_AVAILABLE:
|
|||||||
yield VerticalScroll(id="chat-scroll")
|
yield VerticalScroll(id="chat-scroll")
|
||||||
with Vertical(id="input-area"):
|
with Vertical(id="input-area"):
|
||||||
with Container(id="input-wrapper"):
|
with Container(id="input-wrapper"):
|
||||||
yield TextArea(
|
yield SendTextArea(
|
||||||
placeholder="输入消息... (Ctrl+Enter 发送, Enter 换行)",
|
placeholder="输入消息... (回车发送, Shift+回车换行)",
|
||||||
id="user-input",
|
id="user-input",
|
||||||
soft_wrap=True,
|
soft_wrap=True,
|
||||||
tab_behavior="indent",
|
tab_behavior="indent",
|
||||||
)
|
)
|
||||||
yield Static(
|
yield Static(
|
||||||
" Ctrl+Enter 发送 | ↑↓ 历史 | Ctrl+L 清屏 | F1 侧栏 | Tab 缩进",
|
" 回车发送 | Shift+回车换行 | Ctrl+L 清屏 | F1 侧栏",
|
||||||
id="input-hint"
|
id="input-hint"
|
||||||
)
|
)
|
||||||
with Vertical(id="sidebar"):
|
with Vertical(id="sidebar"):
|
||||||
@@ -488,7 +506,7 @@ if TEXTUAL_AVAILABLE:
|
|||||||
" 代码编辑 · 代码搜索 · Git 集成 · 多引擎",
|
" 代码编辑 · 代码搜索 · Git 集成 · 多引擎",
|
||||||
" 输入 /help 查看命令 | /model 切换引擎",
|
" 输入 /help 查看命令 | /model 切换引擎",
|
||||||
f" 当前引擎: {self.bridge.current_engine()} | 工具: {len(self.bridge.tools())} 个",
|
f" 当前引擎: {self.bridge.current_engine()} | 工具: {len(self.bridge.tools())} 个",
|
||||||
" 支持多行输入 — Ctrl+Enter 发送",
|
" 支持多行输入 — Shift+回车换行",
|
||||||
"",
|
"",
|
||||||
]
|
]
|
||||||
|
|
||||||
@@ -532,7 +550,7 @@ if TEXTUAL_AVAILABLE:
|
|||||||
(" 代码编辑 · 代码搜索 · Git 集成 · 多引擎", "msg-system"),
|
(" 代码编辑 · 代码搜索 · Git 集成 · 多引擎", "msg-system"),
|
||||||
(" 输入 /help 查看命令 | /model 切换引擎", "msg-system"),
|
(" 输入 /help 查看命令 | /model 切换引擎", "msg-system"),
|
||||||
(f" 当前引擎: {self.bridge.current_engine()} | 工具: {len(self.bridge.tools())} 个", "msg-system"),
|
(f" 当前引擎: {self.bridge.current_engine()} | 工具: {len(self.bridge.tools())} 个", "msg-system"),
|
||||||
(" 支持多行输入 — Ctrl+Enter 发送", "msg-system"),
|
(" 支持多行输入 — Shift+回车换行", "msg-system"),
|
||||||
("", "msg-dim"),
|
("", "msg-dim"),
|
||||||
]
|
]
|
||||||
for text, cls in lines:
|
for text, cls in lines:
|
||||||
@@ -670,8 +688,14 @@ if TEXTUAL_AVAILABLE:
|
|||||||
|
|
||||||
# ── 多行输入处理 ──
|
# ── 多行输入处理 ──
|
||||||
|
|
||||||
|
def action_newline(self):
|
||||||
|
"""Shift+Enter 在输入框插入换行"""
|
||||||
|
text_area = self.query_one("#user-input")
|
||||||
|
if text_area.has_focus:
|
||||||
|
text_area.action_edit_insert_newline()
|
||||||
|
|
||||||
def action_send_message(self):
|
def action_send_message(self):
|
||||||
"""Ctrl+Enter 发送消息"""
|
"""回车发送消息"""
|
||||||
text_area = self.query_one("#user-input")
|
text_area = self.query_one("#user-input")
|
||||||
text = text_area.text.strip()
|
text = text_area.text.strip()
|
||||||
if not text:
|
if not text:
|
||||||
@@ -754,9 +778,9 @@ if TEXTUAL_AVAILABLE:
|
|||||||
|
|
||||||
⌨ 快捷键:
|
⌨ 快捷键:
|
||||||
Ctrl+C 退出 Ctrl+L 清屏
|
Ctrl+C 退出 Ctrl+L 清屏
|
||||||
Ctrl+N 新对话 Ctrl+Enter 发送消息
|
Ctrl+N 新对话 回车 发送消息
|
||||||
F1 侧栏 Escape 取消生成
|
Shift+回车 换行 F1 侧栏
|
||||||
Enter 换行 Tab 缩进"""
|
Escape 取消生成 Tab 缩进"""
|
||||||
self._system(help_text)
|
self._system(help_text)
|
||||||
|
|
||||||
def _handle_manual(self, args):
|
def _handle_manual(self, args):
|
||||||
|
|||||||
Reference in new issue
Block a user