mirror of
https://github.com/xtekky/gpt4free.git
synced 2025-11-03 02:23:27 +08:00
Improve tests
This commit is contained in:
25
etc/unittest/mocks.py
Normal file
25
etc/unittest/mocks.py
Normal file
@@ -0,0 +1,25 @@
|
||||
from g4f.Provider.base_provider import AbstractProvider, AsyncProvider, AsyncGeneratorProvider
|
||||
|
||||
class ProviderMock(AbstractProvider):
|
||||
working = True
|
||||
|
||||
def create_completion(
|
||||
model, messages, stream, **kwargs
|
||||
):
|
||||
yield "Mock"
|
||||
|
||||
class AsyncProviderMock(AsyncProvider):
|
||||
working = True
|
||||
|
||||
async def create_async(
|
||||
model, messages, **kwargs
|
||||
):
|
||||
return "Mock"
|
||||
|
||||
class AsyncGeneratorProviderMock(AsyncGeneratorProvider):
|
||||
working = True
|
||||
|
||||
async def create_async_generator(
|
||||
model, messages, stream, **kwargs
|
||||
):
|
||||
yield "Mock"
|
||||
Reference in New Issue
Block a user