scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-09-06 08:16:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

[dev] modify g context attr

Browse source
This commit is contained in:
hs 2024-06-27 12:29:35 +08:00
parent 5bcb6301ac
commit 6dbf44904a
1 changed files with 18 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
memory_scope/chat_v2/global_context.py
View file

@ -9,18 +9,26 @@ from memory_scope.memory.service.base_memory_service import BaseMemoryService
from memory_scope.models.base_model import BaseModel
from memory_scope.storage.base_monitor import BaseMonitor
from memory_scope.storage.base_vector_store import BaseVectorStore
from memory_scope.memory.worker.base_worker import BaseWorker
class GlobalContext(pydantic.BaseModel):
global_config: Dict[str, Any] = pydantic.Field({}, description="global config")
worker_config: Dict[str, Any] = pydantic.Field({}, description="worker config")
class GlobalContext(object):
def __init__(self):
self.global_configs: Dict[str, Any] = {}
memory_service_dict: Dict[str, BaseMemoryService] = pydantic.Field({}, description="memory_service dict")
model_dict: Dict[str, BaseModel] = pydantic.Field({}, description="model dict")
memory_chat_dict: Dict[str, BaseMemoryChat] = pydantic.Field({}, description="memory_chat dict")
self.worker_config: Dict[str, Dict[str, BaseWorker]] = {}
vector_store: BaseVectorStore | None = pydantic.Field(None, description="global vector_store")
monitor: BaseMonitor | None = pydantic.Field(None, description="global monitor")
thread_pool: ThreadPoolExecutor | None = pydantic.Field(None, description="global thread_pool")
language: LanguageEnum = pydantic.Field(LanguageEnum.CN, description="language: cn / en")
self.model_dict: Dict[str, BaseModel] = {}
self.memory_chat_dict: Dict[str, BaseMemoryChat] = {}
self.vector_store: BaseVectorStore | None = None
self.monitor: BaseMonitor | None = None
self.thread_pool: ThreadPoolExecutor | None = None
self.language: LanguageEnum = LanguageEnum.EN
G_CONTEXT = GlobalContext()