diff --git a/ai.py b/ai.py
index f5eebf0..609e378 100644
--- a/ai.py
+++ b/ai.py
@@ -35,3 +35,4 @@ def generate_ai_text(forecast):
     )
 
     print(response.choices[0].message.content)
+    return response.choices[0].message.content
diff --git a/tests/ai_test.py b/tests/ai_test.py
new file mode 100644
index 0000000..e69de29