learn-ai

Runtime error

inflaton commited on Aug 5, 2023

Commit

a905cd1

•

1 Parent(s): 4821924

improve QA unit tests

Files changed (1) hide show

test.py CHANGED Viewed

@@ -119,7 +119,7 @@ class TestQAChain(unittest.TestCase):
         result = qa_chain.call_chain(inputs, None)
         end3 = timer()
         print(f"Inference completed in {end3 - end2:.3f}s")
-        print(result)
     def test_openai(self):
         self.run_test_case("openai", self.question)

         result = qa_chain.call_chain(inputs, None)
         end3 = timer()
         print(f"Inference completed in {end3 - end2:.3f}s")
+        print_llm_response(result)
     def test_openai(self):
         self.run_test_case("openai", self.question)