a292e69b22
主要修改： 1. 服务端新增 Thread 模型，用于存储对话列表信息，但不保存实际的历史记录 2. langgraph 配置了 InMemorySaver，添加 thread_id 参数，消息对话保存在内存中（服务重启后丢失） 3. 添加获取历史记录的 API 接口，从内存中获取 4. 添加 Graph 的单例模式，get_runnable_agent 导致的重复创建 Agent 实例的情况。 5. 优化 Agent 管理页面，更容易配置，更容易调试 6. 添加独立页面的侧边栏。
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
import os
|
|
import pathlib
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
|
|
from server.models import Base
|
|
from server.models.user_model import User
|
|
from server.models.thread_model import Thread
|
|
|
|
class DBManager:
|
|
"""数据库管理器"""
|
|
|
|
def __init__(self):
|
|
self.db_path = os.path.join("saves", "data", "server.db")
|
|
self.ensure_db_dir()
|
|
|
|
# 创建SQLAlchemy引擎
|
|
self.engine = create_engine(f"sqlite:///{self.db_path}")
|
|
|
|
# 创建会话工厂
|
|
self.Session = sessionmaker(bind=self.engine)
|
|
|
|
# 确保表存在
|
|
self.create_tables()
|
|
|
|
def ensure_db_dir(self):
|
|
"""确保数据库目录存在"""
|
|
db_dir = os.path.dirname(self.db_path)
|
|
pathlib.Path(db_dir).mkdir(parents=True, exist_ok=True)
|
|
|
|
def create_tables(self):
|
|
"""创建数据库表"""
|
|
Base.metadata.create_all(self.engine)
|
|
|
|
def check_first_run(self):
|
|
"""检查是否首次运行"""
|
|
session = self.get_session()
|
|
try:
|
|
# 检查是否有任何用户存在
|
|
return session.query(User).count() == 0
|
|
finally:
|
|
session.close()
|
|
|
|
def get_session(self):
|
|
"""获取数据库会话"""
|
|
return self.Session()
|
|
|
|
# 创建全局数据库管理器实例
|
|
db_manager = DBManager() |