From 315a2f25955e90ed3f653787762abb5bdee878ff Mon Sep 17 00:00:00 2001 From: Heiner Lohaus Date: Sat, 14 Dec 2024 23:34:13 +0100 Subject: Add streaming and system messages support in Airforce --- etc/unittest/models.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'etc') diff --git a/etc/unittest/models.py b/etc/unittest/models.py index 39fdbb29..cea6e36d 100644 --- a/etc/unittest/models.py +++ b/etc/unittest/models.py @@ -26,4 +26,9 @@ class TestProviderHasModel(unittest.IsolatedAsyncioTestCase): except (MissingRequirementsError, MissingAuthError): return if self.cache[provider.__name__]: - self.assertIn(model, self.cache[provider.__name__], provider.__name__) \ No newline at end of file + self.assertIn(model, self.cache[provider.__name__], provider.__name__) + + async def test_all_providers_working(self): + for model, providers in __models__.values(): + for provider in providers: + self.assertTrue(provider.working, f"{provider.__name__} in {model.name}") \ No newline at end of file -- cgit v1.2.3