diff --git a/tests/text_generation/test_chat_gpt.py b/tests/text_generation/test_chat_gpt.py index c396ad7..0f9884f 100644 --- a/tests/text_generation/test_chat_gpt.py +++ b/tests/text_generation/test_chat_gpt.py @@ -1,8 +1,8 @@ import pytest -# import src.text_generation.chat_gpt as chat_gpt +import src.text_generation.chat_gpt as chat_gpt +@pytest.mark.apitest def test_chat_gpt_response(): - # prompt = "Say anything!" - # response = chat_gpt.chat_with_gpt(prompt) - # assert response - assert True \ No newline at end of file + prompt = "Say anything!" + response = chat_gpt.chat_with_gpt(prompt) + assert response \ No newline at end of file