1. Basics#
The following tutorial shows the basic use of the database connection.
See context_storage_factory function for creating a context storage by path.
In this example JSON file is used as a storage.
[1]:
# installing dependencies
%pip install -q chatsky[json,pickle]
Note: you may need to restart the kernel to use updated packages.
[2]:
import pathlib
from chatsky.context_storages import context_storage_factory
from chatsky import Pipeline
from chatsky.utils.testing.common import (
check_happy_path,
is_interactive_mode,
)
from chatsky.utils.testing.toy_script import TOY_SCRIPT_KWARGS, HAPPY_PATH
pathlib.Path("dbs").mkdir(exist_ok=True)
db = context_storage_factory("json://dbs/file.json")
# db = context_storage_factory("pickle://dbs/file.pkl")
# db = context_storage_factory("shelve://dbs/file")
pipeline = Pipeline(**TOY_SCRIPT_KWARGS, context_storage=db)
if __name__ == "__main__":
check_happy_path(pipeline, HAPPY_PATH, printout=True)
# a function for automatic tutorial running (testing) with HAPPY_PATH
if is_interactive_mode():
pipeline.run()
USER: text='Hi'
BOT : text='Hi, how are you?'
USER: text='i'm fine, how are you?'
BOT : text='Good. What do you want to talk about?'
USER: text='Let's talk about music.'
BOT : text='Sorry, I can not talk about music now.'
USER: text='Ok, goodbye.'
BOT : text='bye'
USER: text='Hi'
BOT : text='Hi, how are you?'