mirror of
https://gitcode.com/JianFeeeee/TrulyMEM-TrueHumanMEM.git
synced 2026-09-21 01:18:15 +00:00
jianf
ddc84382fd
1. 修复配置保存问题：config_section.py 现在正确发送 API 配置和工具限制两个消息
2. 修复代理环境变量问题：graph_client.py 自动修正缺少 scheme 前缀的代理 URL
3. 修复响应解析问题：app.py 正确从 data 字段获取 content 和 tool_calls
4. 添加操作日志更新：app.py 处理工具调用信息并更新操作日志
🤖 Generated with CodeArts Agent
255 lines
11 KiB
Python
255 lines
11 KiB
Python
import asyncio
|
|
from pathlib import Path
|
|
from textual.app import App, ComposeResult
|
|
from textual.binding import Binding
|
|
|
|
from core import BackendServer, BackendClient
|
|
from .models.message import Message
|
|
|
|
|
|
class GraphMemoryApp(App):
|
|
CSS_PATH = [
|
|
Path(__file__).parent / "styles" / "app.css",
|
|
Path(__file__).parent / "styles" / "messages.css",
|
|
Path(__file__).parent / "styles" / "components.css",
|
|
]
|
|
|
|
BINDINGS = [
|
|
Binding("f1", "show_help", "帮助"),
|
|
Binding("f2", "toggle_sidebar", "侧边栏"),
|
|
Binding("f3", "toggle_tool_details", "工具详情"),
|
|
Binding("f5", "clear_history", "清屏"),
|
|
Binding("f6", "quit", "退出"),
|
|
]
|
|
|
|
def __init__(self, backend_server: BackendServer = None, config_file: str = None, **kwargs):
|
|
super().__init__(**kwargs)
|
|
self._backend_server = backend_server
|
|
self._backend_client = BackendClient(backend_server) if backend_server else None
|
|
self._api_configured = False
|
|
|
|
def compose(self) -> ComposeResult:
|
|
from .widgets.left_panel import LeftPanel
|
|
from .widgets.right_panel import RightPanel
|
|
from .widgets.status_bar import StatusBar
|
|
from .models.config import AppConfig
|
|
|
|
initial_config = AppConfig()
|
|
|
|
if self._backend_client:
|
|
config_result = self._backend_client.get_config()
|
|
config_data = config_result.get("data", {})
|
|
initial_config.api_key = config_data.get("api_key", "")
|
|
initial_config.base_url = config_data.get("base_url", "https://api.deepseek.com")
|
|
initial_config.model = config_data.get("model", "deepseek-chat")
|
|
|
|
limits_result = self._backend_client.get_tool_limits()
|
|
limits_data = limits_result.get("data", {})
|
|
initial_config.persona_query_max = limits_data.get("persona_query_max", 1)
|
|
initial_config.persona_update_max = limits_data.get("persona_update_max", 1)
|
|
initial_config.task_query_max = limits_data.get("task_query_max", 4)
|
|
initial_config.task_update_max = limits_data.get("task_update_max", 2)
|
|
initial_config.memory_query_max = limits_data.get("memory_query_max", 20)
|
|
initial_config.memory_update_max = limits_data.get("memory_update_max", 10)
|
|
|
|
yield LeftPanel()
|
|
yield RightPanel(config=initial_config)
|
|
yield StatusBar()
|
|
|
|
def on_mount(self) -> None:
|
|
from .widgets.status_bar import StatusBar
|
|
status_bar = self.query_one(StatusBar)
|
|
|
|
if not self._backend_server:
|
|
from .widgets.message_history import MessageHistory
|
|
history = self.query_one(MessageHistory)
|
|
error = Message(role="assistant", content="后端未初始化")
|
|
history.add_message(error)
|
|
status_bar.set_api_status(False)
|
|
return
|
|
|
|
status = self._backend_client.get_status()
|
|
data = status.get("data", {})
|
|
self._api_configured = data.get("config", {}).get("api_key", "") != ""
|
|
status_bar.set_api_status(self._api_configured)
|
|
|
|
from .widgets.message_history import MessageHistory
|
|
history = self.query_one(MessageHistory)
|
|
welcome = Message(
|
|
role="assistant",
|
|
content=f"系统就绪\nAPI Key: {'已配置' if self._api_configured else '未配置'}\n\n输入消息开始对话"
|
|
)
|
|
history.add_message(welcome)
|
|
|
|
def on_unmount(self) -> None:
|
|
if self._backend_client:
|
|
self._backend_client.shutdown()
|
|
|
|
def action_show_help(self) -> None:
|
|
self.notify("F1-帮助 F2-侧边栏 F3-工具详情 F5-清屏 F6-退出", title="快捷键", timeout=10)
|
|
|
|
def action_toggle_sidebar(self) -> None:
|
|
from .widgets.right_panel import RightPanel
|
|
sidebar = self.query_one(RightPanel)
|
|
sidebar.toggle()
|
|
|
|
def action_toggle_tool_details(self) -> None:
|
|
from .widgets.message_history import MessageHistory
|
|
history = self.query_one(MessageHistory)
|
|
history.toggle_latest_tool_details()
|
|
|
|
def action_clear_history(self) -> None:
|
|
from .widgets.message_history import MessageHistory
|
|
history = self.query_one(MessageHistory)
|
|
history.clear_messages()
|
|
|
|
def on_input_box_send_message(self, event) -> None:
|
|
if not self._backend_client:
|
|
self.notify("后端未初始化", title="错误", severity="error")
|
|
return
|
|
|
|
if not self._api_configured:
|
|
self.notify("请先配置 API Key (按 F2 打开侧边栏)", title="提示", severity="warning")
|
|
return
|
|
|
|
user_input = event.content
|
|
from .widgets.message_history import MessageHistory
|
|
from .widgets.status_bar import StatusBar
|
|
|
|
history = self.query_one(MessageHistory)
|
|
status_bar = self.query_one(StatusBar)
|
|
|
|
history.add_message(Message(role="user", content=user_input))
|
|
history.add_message(Message(role="assistant", content="⏳ 正在处理..."))
|
|
status_bar.set_processing(True)
|
|
|
|
asyncio.create_task(self._process(user_input))
|
|
|
|
async def _process(self, user_input: str) -> None:
|
|
from .widgets.message_history import MessageHistory
|
|
from .widgets.status_bar import StatusBar
|
|
from .widgets.right_panel import RightPanel
|
|
from .models.log_entry import LogEntry
|
|
from datetime import datetime
|
|
|
|
history = self.query_one(MessageHistory)
|
|
status_bar = self.query_one(StatusBar)
|
|
|
|
result = await asyncio.get_event_loop().run_in_executor(
|
|
None,
|
|
lambda: self._backend_client.process_message(user_input)
|
|
)
|
|
|
|
if result.get("success"):
|
|
# 响应结构: {"success": True, "data": {"content": "...", "tool_calls": [...], ...}, "error": None}
|
|
data = result.get("data", {})
|
|
content = data.get("content", "(无回复)")
|
|
history.update_latest_message(content)
|
|
|
|
# 处理工具调用信息，更新操作日志
|
|
tool_calls = data.get("tool_calls", [])
|
|
if tool_calls:
|
|
try:
|
|
right_panel = self.query_one(RightPanel)
|
|
operation_log = right_panel.get_operation_log()
|
|
|
|
for tool_call in tool_calls:
|
|
entry = LogEntry(
|
|
timestamp=datetime.now(),
|
|
tool_name=tool_call.get("name", "unknown"),
|
|
arguments=tool_call.get("arguments", {}),
|
|
result=str(tool_call.get("result", "")),
|
|
duration=0.0 # 后端没有返回耗时信息
|
|
)
|
|
operation_log.add_log(entry)
|
|
except Exception:
|
|
pass # 忽略操作日志更新失败
|
|
else:
|
|
error = result.get("error", "未知错误")
|
|
history.update_latest_message(f"❌ 错误: {error}")
|
|
|
|
status_bar.set_processing(False)
|
|
|
|
def on_config_section_config_changed(self, event) -> None:
|
|
if not self._backend_client:
|
|
self.notify("后端未初始化，无法保存配置", title="错误", severity="error")
|
|
return
|
|
|
|
config = event.config
|
|
|
|
if event.is_tool_limits:
|
|
asyncio.create_task(self._update_tool_limits_async(config))
|
|
else:
|
|
asyncio.create_task(self._update_config_async(config))
|
|
|
|
async def _update_config_async(self, config) -> None:
|
|
"""异步更新配置"""
|
|
from .widgets.status_bar import StatusBar
|
|
from .widgets.config_section import ConfigSection
|
|
|
|
status_bar = self.query_one(StatusBar)
|
|
|
|
api_key = config.api_key
|
|
base_url = config.base_url
|
|
model = getattr(config, 'model', 'deepseek-chat')
|
|
|
|
try:
|
|
result = await asyncio.get_event_loop().run_in_executor(
|
|
None,
|
|
lambda: self._backend_client.update_config(api_key=api_key, base_url=base_url, model=model)
|
|
)
|
|
|
|
if result.get("success"):
|
|
self._api_configured = bool(api_key)
|
|
status_bar.set_api_status(self._api_configured)
|
|
|
|
# 从后端重新获取配置并刷新 UI
|
|
config_result = await asyncio.get_event_loop().run_in_executor(
|
|
None,
|
|
lambda: self._backend_client.get_config()
|
|
)
|
|
config_data = config_result.get("data", {})
|
|
|
|
# 刷新输入框
|
|
try:
|
|
config_section = self.query_one(ConfigSection)
|
|
config_section.set_config(AppConfig(
|
|
api_key=config_data.get("api_key", ""),
|
|
base_url=config_data.get("base_url", "https://api.deepseek.com"),
|
|
model=config_data.get("model", "deepseek-chat")
|
|
))
|
|
except Exception:
|
|
pass
|
|
|
|
self.notify("✅ 配置已保存并生效", title="配置成功", severity="information")
|
|
else:
|
|
error = result.get("error", "未知错误")
|
|
self.notify(f"❌ 配置失败: {error}", title="配置失败", severity="error")
|
|
except Exception as e:
|
|
self.notify(f"❌ 配置异常: {str(e)}", title="配置失败", severity="error")
|
|
|
|
async def _update_tool_limits_async(self, config) -> None:
|
|
from .widgets.status_bar import StatusBar
|
|
|
|
status_bar = self.query_one(StatusBar)
|
|
|
|
try:
|
|
result = await asyncio.get_event_loop().run_in_executor(
|
|
None,
|
|
lambda: self._backend_client.update_tool_limits(
|
|
persona_query_max=config.persona_query_max,
|
|
persona_update_max=config.persona_update_max,
|
|
task_query_max=config.task_query_max,
|
|
task_update_max=config.task_update_max,
|
|
memory_query_max=config.memory_query_max,
|
|
memory_update_max=config.memory_update_max,
|
|
)
|
|
)
|
|
|
|
if result.get("success"):
|
|
self.notify("✅ 工具限制已保存", title="配置成功", severity="information")
|
|
else:
|
|
error = result.get("error", "未知错误")
|
|
self.notify(f"❌ 配置失败: {error}", title="配置失败", severity="error")
|
|
except Exception as e:
|
|
self.notify(f"❌ 配置异常: {str(e)}", title="配置失败", severity="error") |