From e4f743881c915ba250b859d01ef78c7b90a96bf5 Mon Sep 17 00:00:00 2001 From: Heiner Lohaus Date: Tue, 23 Jan 2024 12:17:16 +0100 Subject: Fix: Model sometimes not converted #1507 --- etc/unittest/mocks.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) (limited to 'etc/unittest/mocks.py') 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 -- cgit v1.2.3