|
"""Ask a question to the notion database.""" |
|
import faiss |
|
from langchain import OpenAI |
|
from langchain.chains import VectorDBQAWithSourcesChain |
|
import pickle |
|
import argparse |
|
|
|
parser = argparse.ArgumentParser(description='Ask a question about the paper') |
|
parser.add_argument('question', type=str, help='The question to ask about the paper') |
|
args = parser.parse_args() |
|
|
|
|
|
index = faiss.read_index("docs.index") |
|
|
|
with open("faiss_store.pkl", "rb") as f: |
|
store = pickle.load(f) |
|
|
|
store.index = index |
|
chain = VectorDBQAWithSourcesChain.from_llm(llm=OpenAI(temperature=0), vectorstore=store) |
|
result = chain({"question": args.question}) |
|
print(f"Answer: {result['answer']}") |
|
sources = result["sources"].split(", ") |
|
sources = [s.title() for s in sources] |
|
|
|
import code |
|
code.interact(local=locals()) |
|
|
