Skip to main content
This notebook shows how to use DuckDB as a vector store. 
! pip install duckdb langchain langchain-community langchain-openai
We want to use OpenAIEmbeddings so we have to get the OpenAI API Key. 
import getpass import os  if "OPENAI_API_KEY" not in os.environ:  os.environ["OPENAI_API_KEY"] = getpass.getpass("OpenAI API Key:") 
from langchain_community.vectorstores import DuckDB from langchain_openai import OpenAIEmbeddings 
from langchain_community.document_loaders import TextLoader from langchain_text_splitters import CharacterTextSplitter  loader = TextLoader("../../how_to/state_of_the_union.txt") documents = loader.load()  documents = CharacterTextSplitter().split_documents(documents) embeddings = OpenAIEmbeddings() 
docsearch = DuckDB.from_documents(documents, embeddings)  query = "What did the president say about Ketanji Brown Jackson" docs = docsearch.similarity_search(query) 
print(docs[0].page_content)
Edit the source of this page on GitHub
⌘I