2024-01-20 20:20:23 -05:00
|
|
|
from g4f.Provider.base_provider import AbstractProvider, AsyncProvider, AsyncGeneratorProvider
|
|
|
|
|
|
|
|
class ProviderMock(AbstractProvider):
|
|
|
|
working = True
|
|
|
|
|
|
|
|
def create_completion(
|
|
|
|
model, messages, stream, **kwargs
|
|
|
|
):
|
|
|
|
yield "Mock"
|
2024-02-14 03:21:57 -05:00
|
|
|
|
2024-01-20 20:20:23 -05:00
|
|
|
class AsyncProviderMock(AsyncProvider):
|
|
|
|
working = True
|
2024-02-14 03:21:57 -05:00
|
|
|
|
2024-01-20 20:20:23 -05:00
|
|
|
async def create_async(
|
|
|
|
model, messages, **kwargs
|
|
|
|
):
|
|
|
|
return "Mock"
|
|
|
|
|
|
|
|
class AsyncGeneratorProviderMock(AsyncGeneratorProvider):
|
|
|
|
working = True
|
2024-02-14 03:21:57 -05:00
|
|
|
|
2024-01-20 20:20:23 -05:00
|
|
|
async def create_async_generator(
|
|
|
|
model, messages, stream, **kwargs
|
|
|
|
):
|
2024-01-23 06:17:16 -05:00
|
|
|
yield "Mock"
|
2024-02-14 03:21:57 -05:00
|
|
|
|
2024-01-23 06:17:16 -05:00
|
|
|
class ModelProviderMock(AbstractProvider):
|
|
|
|
working = True
|
|
|
|
|
|
|
|
def create_completion(
|
|
|
|
model, messages, stream, **kwargs
|
|
|
|
):
|
2024-02-14 03:21:57 -05:00
|
|
|
yield model
|
|
|
|
|
|
|
|
class YieldProviderMock(AsyncGeneratorProvider):
|
|
|
|
working = True
|
|
|
|
|
|
|
|
async def create_async_generator(
|
|
|
|
model, messages, stream, **kwargs
|
|
|
|
):
|
|
|
|
for message in messages:
|
|
|
|
yield message["content"]
|