Source code for memoryscope.core.memoryscope_context

from concurrent.futures import ThreadPoolExecutor
from dataclasses import dataclass, field

from memoryscope.enumeration.language_enum import LanguageEnum
from memoryscope.core.utils.singleton import singleton

@singleton
@dataclass
class MemoryscopeContext(object):
    """
    The context class archives all configs utilized by store, monitor, services and workers.
    """

    language: LanguageEnum = LanguageEnum.EN

    thread_pool: ThreadPoolExecutor | None = None

    memory_store = None

    monitor = None

    memory_chat_dict: dict = field(default_factory=lambda: {}, metadata={"help": "name -> memory_chat"})

    memory_service_dict: dict = field(default_factory=lambda: {}, metadata={"help": "name -> memory_service"})

    model_dict: dict = field(default_factory=lambda: {}, metadata={"help": "name -> model"})

    worker_conf_dict: dict = field(default_factory=lambda: {}, metadata={"help": "name -> worker_conf"})

    meta_data: dict = field(default_factory=lambda: {})

    memory_scope_uuid: str = ""

    print_workflow_dynamic: bool = False

    log_elasticsearch_dynamic: bool = False




def get_memoryscope_uuid():
    ms_context = MemoryscopeContext()
    if ms_context.memory_scope_uuid:
        return ms_context.memory_scope_uuid
    else:
        # raise RuntimeError("MemoryscopeContext is not initialized yet. Please initialize it first.")
        return "memory_scope_uuid_not_registered"




def get_memoryscope_context():
    ms_context = MemoryscopeContext()
    if ms_context.memory_scope_uuid:
        return ms_context
    else:
        # raise RuntimeError("MemoryscopeContext is not initialized yet. Please initialize it first.")
        return "memory_scope_uuid_not_registered"