Xiaoli Dev
e1eee6a0c2
本次更新由 mimo 自动化执行，与开发者无关。 新增 xiaoli/ 包: - __main__.py: 入口点 (python -m xiaoli) - app.py: 主应用薄编排层（~400行 vs 原3860行） - config.py: 配置管理 - engines.py: AI 引擎管理 - sandbox.py: 代码安全沙箱 - prompt.py: 系统提示词构建 - conversation/: 对话解析器 + 历史管理 - display/: CLI/TUI 双模式输出 改进: - 支持 argparse (--tui, --engine, --version) - 插件命令注册表 - 更新 README - 更新 .gitignore 原 ai_cli.py 保留向后兼容
87 lines
3.0 KiB
Python
87 lines
3.0 KiB
Python
"""对话历史管理"""
|
|
import os
|
|
import json
|
|
import time
|
|
import logging
|
|
from typing import Optional
|
|
from colorama import Fore, Style
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class History:
|
|
"""对话历史管理器"""
|
|
|
|
def __init__(self, history_dir: str, max_items: int = 999999):
|
|
self.dir = history_dir
|
|
self.max = max_items
|
|
self.messages = []
|
|
|
|
os.makedirs(self.dir, exist_ok=True)
|
|
|
|
def add(self, role: str, content: str):
|
|
"""添加消息"""
|
|
self.messages.append({"role": role, "content": content})
|
|
if len(self.messages) > self.max:
|
|
self.messages = self.messages[-self.max:]
|
|
|
|
def get(self) -> list:
|
|
"""获取全部历史"""
|
|
return self.messages
|
|
|
|
def clear(self):
|
|
"""清空历史"""
|
|
self.messages = []
|
|
|
|
def set_all(self, messages: list):
|
|
"""设置全部历史"""
|
|
self.messages = messages
|
|
|
|
def save(self, name: str) -> bool:
|
|
"""保存到文件"""
|
|
try:
|
|
data = {
|
|
"timestamp": time.strftime("%Y-%m-%d %H:%M:%S"),
|
|
"conversation": self.messages.copy()
|
|
}
|
|
path = os.path.join(self.dir, f"{name}.json")
|
|
with open(path, 'w', encoding='utf-8') as f:
|
|
json.dump(data, f, ensure_ascii=False, indent=2, default=str)
|
|
print(f"{Fore.GREEN}聊天记录已保存: {name}{Style.RESET_ALL}")
|
|
return True
|
|
except Exception as e:
|
|
print(f"{Fore.RED}保存失败: {e}{Style.RESET_ALL}")
|
|
return False
|
|
|
|
def load(self, name: str) -> bool:
|
|
"""从文件加载"""
|
|
try:
|
|
path = os.path.join(self.dir, f"{name}.json")
|
|
if not os.path.exists(path):
|
|
print(f"{Fore.RED}记录 '{name}' 不存在{Style.RESET_ALL}")
|
|
return False
|
|
with open(path, 'r', encoding='utf-8') as f:
|
|
data = json.load(f)
|
|
self.messages = data.get("conversation", [])
|
|
print(f"{Fore.GREEN}已加载: {name} ({len(self.messages)} 条){Style.RESET_ALL}")
|
|
return True
|
|
except Exception as e:
|
|
print(f"{Fore.RED}加载失败: {e}{Style.RESET_ALL}")
|
|
return False
|
|
|
|
def list_saved(self):
|
|
"""列出所有已保存的记录"""
|
|
try:
|
|
files = [f for f in os.listdir(self.dir) if f.endswith('.json')]
|
|
if not files:
|
|
print(f"{Fore.YELLOW}没有已保存的记录{Style.RESET_ALL}")
|
|
return
|
|
print(f"{Fore.GREEN}已保存的记录:{Style.RESET_ALL}")
|
|
for i, f in enumerate(files, 1):
|
|
path = os.path.join(self.dir, f)
|
|
mtime = time.strftime("%Y-%m-%d %H:%M:%S",
|
|
time.localtime(os.path.getmtime(path)))
|
|
print(f"{Fore.GREEN}{i}. {f[:-5]} ({mtime}){Style.RESET_ALL}")
|
|
except Exception as e:
|
|
print(f"{Fore.RED}列出记录失败: {e}{Style.RESET_ALL}")
|