mirror of
https://github.com/xtekky/gpt4free.git
synced 2024-11-09 01:10:34 +00:00
73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
|
import sys
|
||
|
|
import pathlib
|
||
|
|
import unittest
|
||
|
|
from unittest.mock import MagicMock
|
||
|
|
|
||
|
|
sys.path.append(str(pathlib.Path(__file__).parent.parent.parent))
|
||
|
|
|
||
|
|
import g4f
|
||
|
|
from g4f import ChatCompletion, get_last_provider
|
||
|
|
from g4f.gui.server.backend import Backend_Api, get_error_message
|
||
|
|
from g4f.base_provider import BaseProvider
|
||
|
|
|
||
|
|
g4f.debug.logging = False
|
||
|
|
|
||
|
|
class MockProvider(BaseProvider):
|
||
|
|
working = True
|
||
|
|
|
||
|
|
def create_completion(
|
||
|
|
model, messages, stream, **kwargs
|
||
|
|
):
|
||
|
|
yield "Mock"
|
||
|
|
|
||
|
|
async def create_async(
|
||
|
|
model, messages, **kwargs
|
||
|
|
):
|
||
|
|
return "Mock"
|
||
|
|
|
||
|
|
class TestBackendApi(unittest.TestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.app = MagicMock()
|
||
|
|
self.api = Backend_Api(self.app)
|
||
|
|
|
||
|
|
def test_version(self):
|
||
|
|
response = self.api.get_version()
|
||
|
|
self.assertIn("version", response)
|
||
|
|
self.assertIn("latest_version", response)
|
||
|
|
|
||
|
|
class TestChatCompletion(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_create(self):
|
||
|
|
messages = [{'role': 'user', 'content': 'Hello'}]
|
||
|
|
result = ChatCompletion.create(g4f.models.default, messages)
|
||
|
|
self.assertTrue("Hello" in result or "Good" in result)
|
||
|
|
|
||
|
|
def test_get_last_provider(self):
|
||
|
|
messages = [{'role': 'user', 'content': 'Hello'}]
|
||
|
|
ChatCompletion.create(g4f.models.default, messages, MockProvider)
|
||
|
|
self.assertEqual(get_last_provider(), MockProvider)
|
||
|
|
|
||
|
|
def test_bing_provider(self):
|
||
|
|
messages = [{'role': 'user', 'content': 'Hello'}]
|
||
|
|
provider = g4f.Provider.Bing
|
||
|
|
result = ChatCompletion.create(g4f.models.default, messages, provider)
|
||
|
|
self.assertTrue("Bing" in result)
|
||
|
|
|
||
|
|
class TestChatCompletionAsync(unittest.IsolatedAsyncioTestCase):
|
||
|
|
|
||
|
|
async def test_async(self):
|
||
|
|
messages = [{'role': 'user', 'content': 'Hello'}]
|
||
|
|
result = await ChatCompletion.create_async(g4f.models.default, messages, MockProvider)
|
||
|
|
self.assertTrue("Mock" in result)
|
||
|
|
|
||
|
|
class TestUtilityFunctions(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_get_error_message(self):
|
||
|
|
g4f.debug.last_provider = g4f.Provider.Bing
|
||
|
|
exception = Exception("Message")
|
||
|
|
result = get_error_message(exception)
|
||
|
|
self.assertEqual("Bing: Exception: Message", result)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|