mirror of
https://gitcode.com/JianFeeeee/TrulyMEM-TrueHumanMEM.git
synced 2026-09-20 17:08:18 +00:00
c742a30e1b
- 合并 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) - 更新文档
75 lines
2.2 KiB
Python
75 lines
2.2 KiB
Python
import threading
|
|
import time
|
|
from typing import Any, Dict
|
|
|
|
from .server import BackendServer, Packet, PacketType
|
|
|
|
|
|
class BackendClient:
|
|
|
|
def __init__(self, server: BackendServer):
|
|
self._server = server
|
|
self._counter = 0
|
|
self._lock = threading.Lock()
|
|
|
|
def _next_id(self) -> str:
|
|
with self._lock:
|
|
self._counter += 1
|
|
return f"{time.time()}_{self._counter}"
|
|
|
|
def send(self, message: str) -> Dict:
|
|
return self.process_message(message)
|
|
|
|
def process_message(self, user_input: str) -> Dict:
|
|
return self._server.process_message(user_input)
|
|
|
|
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}
|
|
)
|
|
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() |