36 lines
934 B
Python
36 lines
934 B
Python
from typing import Dict
|
|
|
|
from src.common.logger import get_logger
|
|
from .relationship_builder import RelationshipBuilder
|
|
|
|
logger = get_logger("relationship_builder_manager")
|
|
|
|
|
|
class RelationshipBuilderManager:
|
|
"""关系构建器管理器
|
|
|
|
简单的关系构建器存储和获取管理
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.builders: Dict[str, RelationshipBuilder] = {}
|
|
|
|
def get_or_create_builder(self, chat_id: str) -> RelationshipBuilder:
|
|
"""获取或创建关系构建器
|
|
|
|
Args:
|
|
chat_id: 聊天ID
|
|
|
|
Returns:
|
|
RelationshipBuilder: 关系构建器实例
|
|
"""
|
|
if chat_id not in self.builders:
|
|
self.builders[chat_id] = RelationshipBuilder(chat_id)
|
|
logger.debug(f"创建聊天 {chat_id} 的关系构建器")
|
|
|
|
return self.builders[chat_id]
|
|
|
|
|
|
# 全局管理器实例
|
|
relationship_builder_manager = RelationshipBuilderManager()
|