print out llm response

predibase · Nov 8, 2024 · 79c8174 · 79c8174
1 parent 349913e
commit 79c8174
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/integration-tests-2/test_base_mistral.py b/integration-tests-2/test_base_mistral.py
@@ -16,5 +16,6 @@ def test_base_mistral():
         json={"inputs": test_prompt, "parameters": {"max_new_tokens": 10}},
     )
     response.raise_for_status()
+    print("RESPONSE FROM LLM: ", response.json())
     assert len(response.json()["generated_text"]) > 0
     runner.stop_container()