mirror of https://github.com/xtekky/gpt4free
Fix: Model sometimes not converted #1507
parent
b6feec9e7d
commit
e4f743881c
@ -0,0 +1,27 @@
|
||||
import unittest
|
||||
import g4f
|
||||
from g4f import ChatCompletion
|
||||
from .mocks import ModelProviderMock
|
||||
|
||||
DEFAULT_MESSAGES = [{'role': 'user', 'content': 'Hello'}]
|
||||
|
||||
test_model = g4f.models.Model(
|
||||
name = "test/test_model",
|
||||
base_provider = "",
|
||||
best_provider = ModelProviderMock
|
||||
)
|
||||
g4f.models.ModelUtils.convert["test_model"] = test_model
|
||||
|
||||
class TestPassModel(unittest.TestCase):
|
||||
|
||||
def test_model_instance(self):
|
||||
response = ChatCompletion.create(test_model, DEFAULT_MESSAGES)
|
||||
self.assertEqual(test_model.name, response)
|
||||
|
||||
def test_model_name(self):
|
||||
response = ChatCompletion.create("test_model", DEFAULT_MESSAGES)
|
||||
self.assertEqual(test_model.name, response)
|
||||
|
||||
def test_model_pass(self):
|
||||
response = ChatCompletion.create("test/test_model", DEFAULT_MESSAGES, ModelProviderMock)
|
||||
self.assertEqual(test_model.name, response)
|
Loading…
Reference in New Issue