mirror of
https://gitcode.com/JianFeeeee/TrulyMEM-TrueHumanMEM.git
synced 2026-10-04 00:14:26 +00:00
chore: 删除 TUI/多用户/CLI 代码,准备鸿蒙工程
This commit is contained in:
@ -1 +0,0 @@
|
||||
"""Data Models for Graph Memory TUI"""
|
||||
@ -1,81 +0,0 @@
|
||||
"""配置数据模型"""
|
||||
|
||||
import json
|
||||
import os
|
||||
from dataclasses import dataclass, asdict
|
||||
from pathlib import Path
|
||||
from typing import Optional
|
||||
|
||||
|
||||
@dataclass
|
||||
class AppConfig:
|
||||
"""应用配置"""
|
||||
api_key: str = ""
|
||||
model: str = "deepseek-chat"
|
||||
base_url: str = "https://api.deepseek.com"
|
||||
persona_update_max: int = 1
|
||||
task_update_max: int = 5
|
||||
memory_query_max: int = 20
|
||||
memory_update_max: int = 10
|
||||
web_username: str = ""
|
||||
web_password: str = ""
|
||||
enable_web: bool = False
|
||||
web_port: int = 4096
|
||||
enable_tui: bool = True
|
||||
|
||||
@classmethod
|
||||
def from_env(cls, username: str = "") -> "AppConfig":
|
||||
"""
|
||||
从环境变量加载配置。
|
||||
如果指定了 username,尝试从用户的配置文件加载。
|
||||
"""
|
||||
# 如果指定了用户名,尝试从用户的配置文件加载
|
||||
if username:
|
||||
from pathlib import Path
|
||||
user_config_path = Path.home() / ".trulymem" / username / "config.json"
|
||||
if user_config_path.exists():
|
||||
return cls.from_file(user_config_path)
|
||||
|
||||
return cls(
|
||||
api_key=os.getenv("DEEPSEEK_API_KEY", ""),
|
||||
model=os.getenv("MODEL_NAME", "deepseek-chat"),
|
||||
base_url=os.getenv("DEEPSEEK_BASE_URL", "https://api.deepseek.com"),
|
||||
persona_update_max=int(os.getenv("PERSONA_UPDATE_MAX", 1)),
|
||||
task_update_max=int(os.getenv("TASK_UPDATE_MAX", 5)),
|
||||
memory_query_max=int(os.getenv("MEMORY_QUERY_MAX", 20)),
|
||||
memory_update_max=int(os.getenv("MEMORY_UPDATE_MAX", 10)),
|
||||
web_username=os.getenv("WEB_USERNAME", ""),
|
||||
web_password=os.getenv("WEB_PASSWORD", ""),
|
||||
enable_web=os.getenv("ENABLE_WEB", "false").lower() == "true",
|
||||
web_port=int(os.getenv("WEB_PORT", 4096)),
|
||||
enable_tui=os.getenv("ENABLE_TUI", "true").lower() == "true",
|
||||
)
|
||||
|
||||
@classmethod
|
||||
def from_file(cls, path: Path) -> "AppConfig":
|
||||
if not path.exists():
|
||||
return cls()
|
||||
|
||||
with open(path, "r", encoding="utf-8") as f:
|
||||
data = json.load(f)
|
||||
|
||||
return cls(
|
||||
api_key=data.get("api_key", ""),
|
||||
model=data.get("model", "deepseek-chat"),
|
||||
base_url=data.get("base_url", "https://api.deepseek.com"),
|
||||
persona_update_max=data.get("persona_update_max", 1),
|
||||
task_update_max=data.get("task_update_max", 5),
|
||||
memory_query_max=data.get("memory_query_max", 20),
|
||||
memory_update_max=data.get("memory_update_max", 10),
|
||||
web_username=data.get("web_username", ""),
|
||||
web_password=data.get("web_password", ""),
|
||||
enable_web=data.get("enable_web", False),
|
||||
web_port=data.get("web_port", 4096),
|
||||
enable_tui=data.get("enable_tui", True),
|
||||
)
|
||||
|
||||
def save(self, path: Path) -> None:
|
||||
path.parent.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
with open(path, "w", encoding="utf-8") as f:
|
||||
json.dump(asdict(self), f, indent=2, ensure_ascii=False)
|
||||
@ -1,30 +0,0 @@
|
||||
"""日志条目数据模型"""
|
||||
|
||||
from dataclasses import dataclass
|
||||
from datetime import datetime
|
||||
from typing import Any, Dict
|
||||
|
||||
|
||||
@dataclass
|
||||
class LogEntry:
|
||||
"""日志条目"""
|
||||
timestamp: datetime
|
||||
tool_name: str
|
||||
arguments: Dict[str, Any]
|
||||
result: str
|
||||
duration: float
|
||||
|
||||
@property
|
||||
def args_summary(self) -> str:
|
||||
"""参数摘要(截断到50字符)"""
|
||||
args_str = str(self.arguments)
|
||||
if len(args_str) > 50:
|
||||
return args_str[:50] + "..."
|
||||
return args_str
|
||||
|
||||
@property
|
||||
def result_summary(self) -> str:
|
||||
"""结果摘要(截断到100字符)"""
|
||||
if len(self.result) > 100:
|
||||
return self.result[:100] + "..."
|
||||
return self.result
|
||||
@ -1,33 +0,0 @@
|
||||
"""消息数据模型"""
|
||||
|
||||
from dataclasses import dataclass, field
|
||||
from datetime import datetime
|
||||
from typing import Dict, List, Literal, Optional, Any
|
||||
|
||||
|
||||
@dataclass
|
||||
class ToolCall:
|
||||
"""工具调用"""
|
||||
id: str
|
||||
name: str
|
||||
arguments: Dict[str, Any]
|
||||
|
||||
|
||||
@dataclass
|
||||
class ToolResult:
|
||||
"""工具执行结果"""
|
||||
tool_call_id: str
|
||||
name: str
|
||||
arguments: Dict[str, Any]
|
||||
result: str
|
||||
success: bool
|
||||
|
||||
|
||||
@dataclass
|
||||
class Message:
|
||||
"""消息"""
|
||||
role: Literal["user", "assistant", "system"]
|
||||
content: str
|
||||
timestamp: datetime = field(default_factory=datetime.now)
|
||||
tool_calls: Optional[List[ToolCall]] = None
|
||||
tool_results: Optional[List[ToolResult]] = None
|
||||
Reference in New Issue
Block a user