summaryrefslogtreecommitdiffstats
path: root/etc/unittest/mocks.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--etc/unittest/mocks.py10
1 files changed, 9 insertions, 1 deletions
diff --git a/etc/unittest/mocks.py b/etc/unittest/mocks.py
index a9505997..885bdaee 100644
--- a/etc/unittest/mocks.py
+++ b/etc/unittest/mocks.py
@@ -22,4 +22,12 @@ class AsyncGeneratorProviderMock(AsyncGeneratorProvider):
async def create_async_generator(
model, messages, stream, **kwargs
):
- yield "Mock" \ No newline at end of file
+ yield "Mock"
+
+class ModelProviderMock(AbstractProvider):
+ working = True
+
+ def create_completion(
+ model, messages, stream, **kwargs
+ ):
+ yield model \ No newline at end of file