diff options
Diffstat (limited to 'etc/unittest/integration.py')
-rw-r--r-- | etc/unittest/integration.py | 9 |
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) |