InMemoryByteStore 통합

무언가를 아주 잠깐만 기억해 두고 싶을 때가 있어요. 프로세스가 살아 있는 동안만 데이터를 보관하면 충분하고, 끝나면 사라져도 괜찮은 상황이죠. 데모를 만들거나 간단히 테스트할 때 특히 그렇구요. 이럴 때 쓰는 게 InMemoryByteStore예요.

출처: 공식문서 - InMemoryByteStore integration

InMemoryByteStore는 ByteStore를 비영속적으로 구현한 저장소예요. 모든 데이터를 Python 딕셔너리에 저장하죠. 그래서 Python 프로세스가 끝나면 함께 사라져요. 자세한 기능과 설정은 API 레퍼런스에서 확인할 수 있어요.

준비하기

InMemoryByteStore는 langchain-core 패키지에 들어 있어요.

pip install -qU langchain-core

바이트 저장소를 아주 간단하게 만들 수 있어요.

from langchain_core.stores import InMemoryByteStore

kv_store = InMemoryByteStore()

생성자에 인자를 전달할 필요가 없어요. 그냥 비어 있는 딕셔너리 하나 만들어 주는 거죠.

mset 메서드로 키에 값을 저장해요.

kv_store.mset(
    [
        ["key1", b"value1"],
        ["key2", b"value2"],
    ]
)

kv_store.mget(
    [
        "key1",
        "key2",
    ]
)

[b'value1', b'value2']

저장한 값이 그대로 돌아오는 걸 확인할 수 있어요.

삭제는 mdelete 메서드를 쓰면 돼요.

kv_store.mdelete(
    [
        "key1",
        "key2",
    ]
)

kv_store.mget(
    [
        "key1",
        "key2",
    ]
)

[None, None]

삭제한 키를 조회하면 None이 돌아와요.

InMemoryByteStore의 모든 기능과 설정은 API 레퍼런스에서 확인할 수 있어요.