mirror of https://github.com/xtekky/gpt4free.git
25 lines
605 B
Python
25 lines
605 B
Python
|
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"
|