improve QA unit tests
Browse files
test.py
CHANGED
@@ -119,7 +119,7 @@ class TestQAChain(unittest.TestCase):
|
|
119 |
result = qa_chain.call_chain(inputs, None)
|
120 |
end3 = timer()
|
121 |
print(f"Inference completed in {end3 - end2:.3f}s")
|
122 |
-
|
123 |
|
124 |
def test_openai(self):
|
125 |
self.run_test_case("openai", self.question)
|
|
|
119 |
result = qa_chain.call_chain(inputs, None)
|
120 |
end3 = timer()
|
121 |
print(f"Inference completed in {end3 - end2:.3f}s")
|
122 |
+
print_llm_response(result)
|
123 |
|
124 |
def test_openai(self):
|
125 |
self.run_test_case("openai", self.question)
|