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