summaryrefslogtreecommitdiffstats
path: root/etc/unittest/integration.py
diff options
context:
space:
mode:
Diffstat (limited to 'etc/unittest/integration.py')
-rw-r--r--etc/unittest/integration.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/etc/unittest/integration.py b/etc/unittest/integration.py
index 808a8d1d..36f09c0e 100644
--- a/etc/unittest/integration.py
+++ b/etc/unittest/integration.py
@@ -1,6 +1,12 @@
import unittest
import json
+try:
+ import nest_asyncio
+ has_nest_asyncio = True
+except:
+ has_nest_asyncio = False
+
from g4f.client import Client, ChatCompletion
from g4f.Provider import Bing, OpenaiChat
@@ -8,6 +14,9 @@ DEFAULT_MESSAGES = [{"role": "system", "content": 'Response in json, Example: {"
{"role": "user", "content": "Say success true in json"}]
class TestProviderIntegration(unittest.TestCase):
+ def setUp(self):
+ if not has_nest_asyncio:
+ self.skipTest("nest_asyncio is not installed")
def test_bing(self):
client = Client(provider=Bing)