refactor: 统一 Packet 通信协议 + 后端配置管理 + UI 清理

- 合并 server.py 到 core/__init__.py,使用统一 Packet 协议
- 后端管理配置持久化 (~/.trulymem/config.json)
- 前端移除 ConfigService,通过 BackendClient 与后端通信
- 删除 UI 中冗余的 AI 推理逻辑 (chat_service, tool_service, message_handler)
- 删除 core/tools 重复文件 (tool_executor, tool_limiter)
- 提示词管理器支持用户自定义 (~/.trulyemem/system_prompt.md)
- 启动入口优化配置路径逻辑
- 更新测试覆盖 (42 tests)
- 更新文档
This commit is contained in:
root
2026-04-14 11:49:39 +08:00
parent 2a76fc6477
commit c742a30e1b
19 changed files with 1020 additions and 2349 deletions

View File

@ -1,61 +1,75 @@
import threading
import time
from typing import Any, Dict
from .server import BackendServer, MessageType
from .server import BackendServer, Packet, PacketType
class BackendClient:
def __init__(self, server: BackendServer):
self._server = server
self._request_counter = 0
self._counter = 0
self._lock = threading.Lock()
def process_message(self, user_input: str, timeout: float = 30.0) -> Dict[str, Any]:
def _next_id(self) -> str:
with self._lock:
self._request_counter += 1
return self._server.process_message(user_input, timeout)
self._counter += 1
return f"{time.time()}_{self._counter}"
def execute_tool(self, tool_name: str, arguments: Dict[str, Any], timeout: float = 10.0) -> str:
with self._lock:
self._request_counter += 1
return self._server.execute_tool(tool_name, arguments, timeout)
def send(self, message: str) -> Dict:
return self.process_message(message)
def update_config(self, api_key: str, base_url: str = "https://api.deepseek.com") -> Dict[str, Any]:
return self._send_request(MessageType.SET_CONFIG, {"api_key": api_key, "base_url": base_url})
def process_message(self, user_input: str) -> Dict:
return self._server.process_message(user_input)
def get_config(self) -> Dict[str, str]:
result = self._send_request(MessageType.GET_CONFIG, {})
return result.get("data", {"api_key": "", "base_url": "https://api.deepseek.com"})
def get_message_history(self) -> list:
result = self._send_request(MessageType.GET_HISTORY, {})
return result.get("data", {}).get("history", [])
def save_message_history(self, messages: list) -> None:
self._send_request(MessageType.SAVE_HISTORY, {"messages": messages})
def _send_request(self, message_type: MessageType, payload: Dict[str, Any]) -> Dict[str, Any]:
import queue
request_id = f"{id(self)}"
response_queue = queue.Queue()
from .server import BackendRequest
request = BackendRequest(
request_id=request_id,
message_type=message_type,
payload=payload,
response_queue=response_queue
def update_config(self, api_key: str, base_url: str = "https://api.deepseek.com", model: str = "deepseek-chat") -> Dict:
packet = Packet(
id=self._next_id(),
type=PacketType.SET_CONFIG,
body={"api_key": api_key, "base_url": base_url, "model": model}
)
self._server._request_queue.put(request)
try:
response = response_queue.get(timeout=5.0)
if not response.success:
raise Exception(response.error)
return {"success": True, "data": response.data}
except queue.Empty:
return {"success": False, "error": "timeout"}
return self._server.send(packet).body
def execute_tool(self, name: str, arguments: Dict) -> Dict:
packet = Packet(
id=self._next_id(),
type=PacketType.EXECUTE_TOOL,
body={"tool_name": name, "arguments": arguments}
)
return self._server.send(packet).body
def get_status(self) -> Dict:
packet = Packet(
id=self._next_id(),
type=PacketType.GET_STATUS,
body={}
)
return self._server.send(packet).body
def get_config(self) -> Dict:
packet = Packet(
id=self._next_id(),
type=PacketType.GET_CONFIG,
body={}
)
return self._server.send(packet).body
def save_history(self, messages: list) -> None:
packet = Packet(
id=self._next_id(),
type=PacketType.SAVE_HISTORY,
body={"messages": messages}
)
self._server.send(packet)
def get_history(self) -> list:
packet = Packet(
id=self._next_id(),
type=PacketType.GET_HISTORY,
body={}
)
return self._server.send(packet).body.get("history", [])
def shutdown(self) -> None:
self._server.shutdown()