Partial context updates

.gitignore

@@ -3,6 +3,7 @@
 dist/
 venv/
 build/
+dbs/
 docs/source/apiref
 docs/source/_misc
 docs/source/release_notes.rst

chatsky/__rebuild_pydantic_models__.py

@@ -5,9 +5,13 @@
 from chatsky.core.script import Node
 from chatsky.core.pipeline import Pipeline
 from chatsky.slots.slots import SlotManager
+from chatsky.context_storages import DBContextStorage, ContextInfo
+from chatsky.core.ctx_dict import ContextDict
 from chatsky.core.context import FrameworkData, ServiceState
 from chatsky.core.service import PipelineComponent
 
+ContextInfo.model_rebuild()
+ContextDict.model_rebuild()
 PipelineComponent.model_rebuild()
 Pipeline.model_rebuild()
 Script.model_rebuild()

chatsky/conditions/standard.py

@@ -202,7 +202,7 @@ def __init__(
         super().__init__(flow_labels=flow_labels, labels=labels, last_n_indices=last_n_indices)
 
     async def call(self, ctx: Context) -> bool:
-        labels = list(ctx.labels.values())[-self.last_n_indices :]  # noqa: E203
+        labels = await ctx.labels[-self.last_n_indices :]  # noqa: E203
         for label in labels:
             if label.flow_name in self.flow_labels or label in self.labels:
                 return True

chatsky/context_storages/__init__.py

@@ -1,11 +1,10 @@
 # -*- coding: utf-8 -*-
 
-from .database import DBContextStorage, threadsafe_method, context_storage_factory
-from .json import JSONContextStorage, json_available
-from .pickle import PickleContextStorage, pickle_available
+from .database import DBContextStorage, ContextInfo, context_storage_factory
+from .file import JSONContextStorage, PickleContextStorage, ShelveContextStorage, json_available, pickle_available
 from .sql import SQLContextStorage, postgres_available, mysql_available, sqlite_available, sqlalchemy_available
 from .ydb import YDBContextStorage, ydb_available
 from .redis import RedisContextStorage, redis_available
+from .memory import MemoryContextStorage
 from .mongo import MongoContextStorage, mongo_available
-from .shelve import ShelveContextStorage
 from .protocol import PROTOCOLS, get_protocol_install_suggestion