Partial context updates (#93)

# Description

Context storages are updated partially now instead of reading and
writing whole data at once.

# Checklist

- [x] I have covered the code with tests
- [x] I have added comments to my code to help others understand it
- [x] I have updated the documentation to reflect the changes
- [x] I have performed a self-review of the changes
- [x] Consider extending `UpdateScheme` from `BaseModel`
- [x] Decide how we want to use `clear` method.

---------

Co-authored-by: Roman Zlobin <[email protected]>
Co-authored-by: = <=>
Co-authored-by: pseusys <[email protected]>

Author: 3 people

.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.core.context import FrameworkData, ServiceState
+from chatsky.context_storages import DBContextStorage
+from chatsky.core.ctx_dict import ContextDict
+from chatsky.core.ctx_utils import ServiceState, FrameworkData, ContextMainInfo
 from chatsky.core.service import PipelineComponent
 
+ContextMainInfo.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.get(ctx.labels.keys()[-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, 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