19 lines
568 B
Python
19 lines
568 B
Python
from langchain.document_loaders import TextLoader
|
|
from langchain.indexes import VectorstoreIndexCreator
|
|
from langchain.chat_models import ChatOpenAI
|
|
from langchain.chains import RetrievalQA
|
|
|
|
# Load YAML doc
|
|
loader = TextLoader("haikal_kb.yaml")
|
|
index = VectorstoreIndexCreator().from_loaders([loader])
|
|
|
|
# Setup QA chain
|
|
qa = RetrievalQA.from_chain_type(
|
|
llm=ChatOpenAI(model="gpt-3.5-turbo", temperature=0),
|
|
retriever=index.vectorstore.as_retriever()
|
|
)
|
|
|
|
# Ask a question
|
|
query = "How do I add a new invoice?"
|
|
response = qa.run(query)
|
|
print("Answer:", response) |