1 files changed, 10 insertions, 7 deletions

diff --git a/testing/test_main.py b/testing/test_main.py
index 612ec695..7c28f1d2 100644
--- a/testing/test_main.py
+++ b/testing/test_main.py
@@ -1,24 +1,27 @@
-from openai_rev import openai_rev, Provider, quora, forefront
+import gpt4free
+from gpt4free import Provider, quora, forefront
 
 # usage You
-response = openai_rev.Completion.create(Provider.You, prompt='Write a poem on Lionel Messi')
+response = gpt4free.Completion.create(Provider.You, prompt='Write a poem on Lionel Messi')
 print(response)
 
 # usage Poe
 token = quora.Account.create(logging=False)
-response = openai_rev.Completion.create(
-    Provider.Poe, prompt='Write a poem on Lionel Messi', token=token, model='ChatGPT'
-)
+response = gpt4free.Completion.create(Provider.Poe, prompt='Write a poem on Lionel Messi', token=token, model='ChatGPT')
 print(response)
 
 # usage forefront
 token = forefront.Account.create(logging=False)
-response = openai_rev.Completion.create(
+response = gpt4free.Completion.create(
     Provider.ForeFront, prompt='Write a poem on Lionel Messi', model='gpt-4', token=token
 )
 print(response)
 print(f'END')
 
 # usage theb
-response = openai_rev.Completion.create(Provider.Theb, prompt='Write a poem on Lionel Messi')
+response = gpt4free.Completion.create(Provider.Theb, prompt='Write a poem on Lionel Messi')
+print(response)
+
+# usage cocalc
+response = gpt4free.Completion.create(Provider.CoCalc, prompt='Write a poem on Lionel Messi', cookie_input='')
 print(response)