Fix: GPT test

SverreNystad · Sep 18, 2023 · 5f34c03 · 5f34c03
1 parent 9c58aa2
commit 5f34c03
Showing 1 changed file with 5 additions and 5 deletions.
diff --git 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
+    prompt = "Say anything!"
+    response = chat_gpt.chat_with_gpt(prompt)
+    assert response