31 lines
845 B
Python
31 lines
845 B
Python
"""工作流记忆管理模块
|
|
|
|
该模块负责管理智能编剧系统工作流的记忆存储和检索。
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
from typing import Dict, Any, List, Optional
|
|
from datetime import datetime
|
|
import json
|
|
from database import client # type: ignore
|
|
from langgraph.checkpoint.mongodb import MongoDBSaver
|
|
|
|
# 添加项目根目录到路径
|
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(__file__)))))
|
|
from agentgraph.utils.logger import get_logger
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
DB_NAME = "langgraph_memory_db"
|
|
|
|
class WorkflowMemory:
|
|
"""工作流记忆管理类
|
|
|
|
负责管理工作流执行过程中的状态存储、检索和历史记录。
|
|
"""
|
|
|
|
def __init__(self):
|
|
"""初始化工作流记忆管理器"""
|
|
self.memory = MongoDBSaver(client, db_name=DB_NAME)
|