From d1e0a196b5a88ce8af5eff35381ab1833a10081b Mon Sep 17 00:00:00 2001 From: Joshua Vial Date: Wed, 23 Aug 2023 21:32:00 +1200 Subject: [PATCH] Add test for creating models from OpenRouter API. --- tests/test_models.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/test_models.py b/tests/test_models.py index 39482dee3..1b826b29b 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -23,6 +23,12 @@ class TestModels(unittest.TestCase): model = Model.create("gpt-4-32k-2123") self.assertEqual(model.max_context_tokens, 32 * 1024) + def test_openrouter_models(self): + import openai + openai.api_base = 'https://openrouter.ai/api/v1' + model = Model.create("gpt-3.5-turbo") + self.assertEqual(model.name, 'openai/gpt-3.5-turbo') + if __name__ == "__main__": unittest.main()