e3768e2707
- Added a new optional parameter `enable_graph_memory_update` to the simulation API, allowing real-time updates of agent activities to the Zep knowledge graph. - Introduced `ZepGraphMemoryUpdater` and `ZepGraphMemoryManager` classes to handle the background processing of activity updates, ensuring efficient API calls and data management. - Updated the README.md to include detailed instructions on the new graph memory update functionality and its configuration. - Enhanced the simulation runner to manage the lifecycle of the graph memory updater, including starting and stopping the updater based on user configuration. - Improved logging to track the status of graph memory updates, providing better insights into the simulation process.
59 lines
1.4 KiB
Python
59 lines
1.4 KiB
Python
"""
|
|
业务服务模块
|
|
"""
|
|
|
|
from .ontology_generator import OntologyGenerator
|
|
from .graph_builder import GraphBuilderService
|
|
from .text_processor import TextProcessor
|
|
from .zep_entity_reader import ZepEntityReader, EntityNode, FilteredEntities
|
|
from .oasis_profile_generator import OasisProfileGenerator, OasisAgentProfile
|
|
from .simulation_manager import SimulationManager, SimulationState, SimulationStatus
|
|
from .simulation_config_generator import (
|
|
SimulationConfigGenerator,
|
|
SimulationParameters,
|
|
AgentActivityConfig,
|
|
TimeSimulationConfig,
|
|
EventConfig,
|
|
PlatformConfig
|
|
)
|
|
from .simulation_runner import (
|
|
SimulationRunner,
|
|
SimulationRunState,
|
|
RunnerStatus,
|
|
AgentAction,
|
|
RoundSummary
|
|
)
|
|
from .zep_graph_memory_updater import (
|
|
ZepGraphMemoryUpdater,
|
|
ZepGraphMemoryManager,
|
|
AgentActivity
|
|
)
|
|
|
|
__all__ = [
|
|
'OntologyGenerator',
|
|
'GraphBuilderService',
|
|
'TextProcessor',
|
|
'ZepEntityReader',
|
|
'EntityNode',
|
|
'FilteredEntities',
|
|
'OasisProfileGenerator',
|
|
'OasisAgentProfile',
|
|
'SimulationManager',
|
|
'SimulationState',
|
|
'SimulationStatus',
|
|
'SimulationConfigGenerator',
|
|
'SimulationParameters',
|
|
'AgentActivityConfig',
|
|
'TimeSimulationConfig',
|
|
'EventConfig',
|
|
'PlatformConfig',
|
|
'SimulationRunner',
|
|
'SimulationRunState',
|
|
'RunnerStatus',
|
|
'AgentAction',
|
|
'RoundSummary',
|
|
'ZepGraphMemoryUpdater',
|
|
'ZepGraphMemoryManager',
|
|
'AgentActivity',
|
|
]
|
|
|