82 lines
3.0 KiB
Python
82 lines
3.0 KiB
Python
"""配置数据模型"""
|
||
|
||
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-v4-flash"
|
||
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-v4-flash"),
|
||
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-v4-flash"),
|
||
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)
|