# ./src/agent/store.pyimport contextlibfrom langchain.embeddings import init_embeddingsfrom langgraph.store.base import IndexConfigfrom langgraph.store.sqlite import AsyncSqliteStoreembeddings = init_embeddings("openai:text-embedding-3-small")@contextlib.asynccontextmanagerasync def generate_store(): """Yield a BaseStore, open for the duration of the server.""" async with AsyncSqliteStore.from_conn_string( "./custom_store.sql", index=IndexConfig( dims=1536, embed=embeddings, fields=["$"], ), ) as store: await store.setup() yield store