"""工作流记忆管理模块

该模块负责管理智能编剧系统工作流的记忆存储和检索。
"""

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)