From d16e47da217ca34b6790d8d85c8e750102942165 Mon Sep 17 00:00:00 2001 From: Daniel Lenton Date: Wed, 6 Nov 2024 15:38:35 +0000 Subject: [PATCH] added test_routing_w_models_skip_providers. --- tests/test_routing/test_routing_syntax.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/test_routing/test_routing_syntax.py b/tests/test_routing/test_routing_syntax.py index d117b2d..6b65e20 100644 --- a/tests/test_routing/test_routing_syntax.py +++ b/tests/test_routing/test_routing_syntax.py @@ -97,5 +97,12 @@ def test_routing_skip_models(): ).generate("Hello.") +def test_routing_w_models_skip_providers(): + unify.Unify( + "router@q:1|i:0.5|models:claude-3-haiku,claude-3-sonnet|" + "skip_providers:aws-bedrock", + ).generate("Hello.") + + if __name__ == "__main__": pass