mirror of
https://github.com/xtekky/gpt4free.git
synced 2025-10-05 16:26:57 +08:00
Add unitests for the client
Fix: async generator ignored GeneratorExit Fix: ResourceWarning: unclosed event loop
This commit is contained in:
@@ -7,10 +7,10 @@ class ProviderMock(AbstractProvider):
|
||||
model, messages, stream, **kwargs
|
||||
):
|
||||
yield "Mock"
|
||||
|
||||
|
||||
class AsyncProviderMock(AsyncProvider):
|
||||
working = True
|
||||
|
||||
|
||||
async def create_async(
|
||||
model, messages, **kwargs
|
||||
):
|
||||
@@ -18,16 +18,25 @@ class AsyncProviderMock(AsyncProvider):
|
||||
|
||||
class AsyncGeneratorProviderMock(AsyncGeneratorProvider):
|
||||
working = True
|
||||
|
||||
|
||||
async def create_async_generator(
|
||||
model, messages, stream, **kwargs
|
||||
):
|
||||
yield "Mock"
|
||||
|
||||
|
||||
class ModelProviderMock(AbstractProvider):
|
||||
working = True
|
||||
|
||||
def create_completion(
|
||||
model, messages, stream, **kwargs
|
||||
):
|
||||
yield model
|
||||
yield model
|
||||
|
||||
class YieldProviderMock(AsyncGeneratorProvider):
|
||||
working = True
|
||||
|
||||
async def create_async_generator(
|
||||
model, messages, stream, **kwargs
|
||||
):
|
||||
for message in messages:
|
||||
yield message["content"]
|
Reference in New Issue
Block a user