! pip install duckdb langchain langchain-community langchain-openai

import getpass
import os

if"OPENAI_API_KEY"notin os.environ:
    os.environ["OPENAI_API_KEY"]= getpass.getpass("OpenAI API Key:")

from langchain_community.vectorstoresimport DuckDB
from langchain_openaiimport OpenAIEmbeddings

from langchain_community.document_loadersimport TextLoader
from langchain_text_splittersimport 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)