Fix: Model sometimes not converted #1507

2025-10-23 16:23:33 +08:00 · 2024-01-23 12:17:16 +01:00
parent b6feec9e7d
commit e4f743881c
5 changed files with 45 additions and 10 deletions
--- 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"
+        yield "Mock"
+        
+class ModelProviderMock(AbstractProvider):
+    working = True
+
+    def create_completion(
+        model, messages, stream, **kwargs
+    ):
+        yield model