Skip to content

Commit 73db48d

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
chore: Remove smart-router-001 system test
PiperOrigin-RevId: 743257063
1 parent 79ca86a commit 73db48d

File tree

1 file changed

+0
-16
lines changed

1 file changed

+0
-16
lines changed

tests/system/vertexai/test_generative_models.py

Lines changed: 0 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,6 @@
4040
GEMINI_VISION_MODEL_NAME = "gemini-1.0-pro-vision"
4141
GEMINI_15_MODEL_NAME = "gemini-1.5-pro-preview-0409"
4242
GEMINI_15_PRO_MODEL_NAME = "gemini-1.5-pro-001"
43-
SMART_ROUTER_NAME = "smart-router-001"
4443
GEMINI_15_PRO_2_MODEL_NAME = "gemini-1.5-pro-002"
4544

4645
STAGING_API_ENDPOINT = "STAGING_ENDPOINT"
@@ -601,21 +600,6 @@ def test_generate_content_function_calling(self, api_endpoint_env_name):
601600

602601
assert summary
603602

604-
def test_generate_content_model_router(self, api_endpoint_env_name):
605-
model = generative_models.GenerativeModel(SMART_ROUTER_NAME)
606-
response = model.generate_content(
607-
contents="Why is sky blue?",
608-
generation_config=generative_models.GenerationConfig(
609-
temperature=0,
610-
routing_config=generative_models.GenerationConfig.RoutingConfig(
611-
routing_config=generative_models.GenerationConfig.RoutingConfig.AutoRoutingMode(
612-
model_routing_preference=generative_models.GenerationConfig.RoutingConfig.AutoRoutingMode.ModelRoutingPreference.BALANCED,
613-
),
614-
),
615-
),
616-
)
617-
assert response.text
618-
619603
def test_chat_automatic_function_calling(self, api_endpoint_env_name):
620604
get_current_weather_func = generative_models.FunctionDeclaration.from_func(
621605
get_current_weather

0 commit comments

Comments
 (0)