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