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, 1 insertions, 8 deletions
diff --git a/etc/unittest/integration.py b/etc/unittest/integration.py
index d8fc41d0..af7494e2 100644
--- a/etc/unittest/integration.py
+++ b/etc/unittest/integration.py
@@ -8,7 +8,7 @@ except ImportError:
has_nest_asyncio = False
from g4f.client import Client, ChatCompletion
-from g4f.Provider import Bing, OpenaiChat, DuckDuckGo
+from g4f.Provider import Bing, OpenaiChat
DEFAULT_MESSAGES = [{"role": "system", "content": 'Response in json, Example: {"success": false}'},
{"role": "user", "content": "Say success true in json"}]
@@ -25,13 +25,6 @@ class TestProviderIntegration(unittest.TestCase):
self.assertIsInstance(response, ChatCompletion)
self.assertIn("success", json.loads(response.choices[0].message.content))
- def test_duckduckgo(self):
- self.skipTest("Not working")
- client = Client(provider=DuckDuckGo)
- response = client.chat.completions.create(DEFAULT_MESSAGES, "", response_format={"type": "json_object"})
- self.assertIsInstance(response, ChatCompletion)
- self.assertIn("success", json.loads(response.choices[0].message.content))
-
def test_openai(self):
self.skipTest("not working in this network")
client = Client(provider=OpenaiChat)