llama-index

1
from llama_index.legacy.storage.chat_store.base import BaseChatStore
2
from llama_index.legacy.storage.chat_store.simple_chat_store import SimpleChatStore
3

4
RECOGNIZED_CHAT_STORES = {
5
    SimpleChatStore.class_name(): SimpleChatStore,
6
}
7

8

9
def load_chat_store(data: dict) -> BaseChatStore:
10
    """Load a chat store from a dict."""
11
    chat_store_name = data.get("class_name", None)
12
    if chat_store_name is None:
13
        raise ValueError("ChatStore loading requires a class_name")
14

15
    if chat_store_name not in RECOGNIZED_CHAT_STORES:
16
        raise ValueError(f"Invalid ChatStore name: {chat_store_name}")
17

18
    return RECOGNIZED_CHAT_STORES[chat_store_name].from_dict(data)
19