80 lines
2.6 KiB
Python
80 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any, Dict, List
|
|
|
|
from core.settings import (
|
|
AGENT_MD_TEMPLATES_FILE,
|
|
BUNDLED_AGENT_MD_TEMPLATES_FILE,
|
|
BUNDLED_TOPIC_PRESETS_TEMPLATES_FILE,
|
|
TOPIC_PRESETS_TEMPLATES_FILE,
|
|
)
|
|
|
|
TEMPLATE_KEYS = ("agents_md", "soul_md", "user_md", "tools_md", "identity_md")
|
|
|
|
|
|
def _load_json_object(path: str, fallback_path: str = "") -> Dict[str, Any]:
|
|
import json
|
|
|
|
for candidate in [path, fallback_path]:
|
|
candidate = str(candidate or "").strip()
|
|
if not candidate:
|
|
continue
|
|
try:
|
|
with open(candidate, "r", encoding="utf-8") as file:
|
|
data = json.load(file)
|
|
if isinstance(data, dict):
|
|
return data
|
|
except Exception:
|
|
continue
|
|
return {}
|
|
|
|
|
|
def _normalize_md_text(value: Any) -> str:
|
|
return str(value or "").replace("\r\n", "\n").strip()
|
|
|
|
|
|
def _write_json_atomic(path: str, payload: Dict[str, Any]) -> None:
|
|
import json
|
|
import os
|
|
|
|
os.makedirs(os.path.dirname(path), exist_ok=True)
|
|
tmp_path = f"{path}.tmp"
|
|
with open(tmp_path, "w", encoding="utf-8") as file:
|
|
json.dump(payload, file, ensure_ascii=False, indent=2)
|
|
os.replace(tmp_path, path)
|
|
|
|
|
|
def get_agent_md_templates() -> Dict[str, str]:
|
|
raw = _load_json_object(str(AGENT_MD_TEMPLATES_FILE), str(BUNDLED_AGENT_MD_TEMPLATES_FILE))
|
|
return {key: _normalize_md_text(raw.get(key)) for key in TEMPLATE_KEYS}
|
|
|
|
|
|
def get_topic_presets() -> Dict[str, Any]:
|
|
raw = _load_json_object(str(TOPIC_PRESETS_TEMPLATES_FILE), str(BUNDLED_TOPIC_PRESETS_TEMPLATES_FILE))
|
|
presets = raw.get("presets")
|
|
if not isinstance(presets, list):
|
|
return {"presets": []}
|
|
return {"presets": [dict(row) for row in presets if isinstance(row, dict)]}
|
|
|
|
|
|
def update_agent_md_templates(raw: Dict[str, Any]) -> Dict[str, str]:
|
|
payload = {key: _normalize_md_text(raw.get(key)) for key in TEMPLATE_KEYS}
|
|
_write_json_atomic(str(AGENT_MD_TEMPLATES_FILE), payload)
|
|
return payload
|
|
|
|
|
|
def update_topic_presets(raw: Dict[str, Any]) -> Dict[str, Any]:
|
|
presets = raw.get("presets") if isinstance(raw, dict) else None
|
|
if presets is None:
|
|
payload: Dict[str, List[Dict[str, Any]]] = {"presets": []}
|
|
elif isinstance(presets, list):
|
|
payload = {"presets": [dict(row) for row in presets if isinstance(row, dict)]}
|
|
else:
|
|
raise ValueError("topic_presets.presets must be an array")
|
|
_write_json_atomic(str(TOPIC_PRESETS_TEMPLATES_FILE), payload)
|
|
return payload
|
|
|
|
|
|
def get_agent_template_value(key: str) -> str:
|
|
return get_agent_md_templates().get(key, "")
|