test: add simple_chatbot_async.py

luochen1990 · Aug 11, 2024 · 318ce37 · 318ce37
1 parent 26917a2
commit 318ce37
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/test/examples/chat_bot/simple_chatbot_async.py b/test/examples/chat_bot/simple_chatbot_async.py
@@ -0,0 +1,10 @@
+import pytest
+from ai_powered import ChatBot
+from ai_powered.colors import gray, green
+
+@pytest.mark.asyncio
+async def test_simple_chatbot_async():
+    bot = ChatBot()
+    print(green(await bot.chat('hello, please tell me the result of 2^10 + 3^4')))
+    print(green(await bot.chat('and what is above result divided by 2?')))
+    print(gray(f"{bot.conversation}"))