def test_qa_chunk(): kiri = Kiri(local=True) docs = get_chunked_docs() kiri.upload(docs) results = kiri.qa("another?") assert isinstance(results, list) for result in results: assert type(result[0]) == str assert isinstance(result[1], SearchResult)
""" elastic = False if elastic: doc_store = ElasticDocStore("http://localhost:9000", doc_class=ElasticChunkedDocument, index="kiri_default") documents = big_n_docs["elastic"] else: doc_store = InMemoryDocStore(doc_class=ChunkedDocument) documents = big_n_docs["memory"] kiri = Kiri(doc_store, local=True) kiri.upload(documents) print("Hello! I'm a Kiri chatbot.") # Hold previous Q/A pairs for additional context session_qa = [] while True: try: question = input() answers = kiri.qa(question, prev_qa=session_qa) # Only showing the top-rated answer print(answers[0][0]) prev_qa = (question, answers[0][0]) session_qa += prev_qa print() except Exception as e: print("Something broke, try again.")
def test_qa(): kiri = Kiri(local=True) out = kiri.qa("What does Jon like?", text) assert type(out) == str