返回提交历史
Modified
etc/unittest/client.py
+2
-2
XFEstudio/gpt4free
Update client.py
7c515fe5
代码差异
1 个文件
+2
-2
@@ -3,7 +3,7 @@ from __future__ import annotations
3
3
import unittest
4
4
5
5
import g4f
6
from g4f.errors import ModelNotFoundError, ResponseError
6
from g4f.errors import ModelNotFoundError, ResponseError, PaymentRequiredError
7
7
from g4f.client import Client, AsyncClient, ChatCompletion, ChatCompletionChunk
8
8
from g4f.client.service import get_model_and_provider
9
9
from g4f.providers.types import BaseProvider
@@ -147,7 +147,7 @@ class TestPassModel(unittest.TestCase):
147
147
client = Client()
148
148
client.chat.completions.create(DEFAULT_MESSAGES, "Hello")
149
149
150
self.assertRaises((ModelNotFoundError, ResponseError), run_exception)
150
self.assertRaises((ModelNotFoundError, ResponseError, PaymentRequiredError), run_exception)
151
151
152
152
def test_best_provider(self):
153
153
not_default_model = "gpt-4o"