From 204bc6539b0846ceb7206d8b70f901102883209d Mon Sep 17 00:00:00 2001
From: factnn <166481866+factnn@users.noreply.github.com>
Date: Sat, 27 Jun 2026 22:28:13 +0800
Subject: [PATCH] test: simplify adaptive_router cost test

---
 .../adaptive_router/test_config.py                | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/tests/test_litellm/router_strategy/adaptive_router/test_config.py b/tests/test_litellm/router_strategy/adaptive_router/test_config.py
index fd14556a0bc..5f1667cd0d6 100644
--- a/tests/test_litellm/router_strategy/adaptive_router/test_config.py
+++ b/tests/test_litellm/router_strategy/adaptive_router/test_config.py
@@ -53,3 +53,18 @@ def test_config_accepts_all_six_request_types_in_strengths():
         ],
     )
     assert len(prefs.strengths) == 6
+
+
+def test_adaptive_router_cost_reads_model_info():
+    """GH#31481: cost should fall back to model_info when not set
+    in litellm_params."""
+    lp_dict = {}  # no cost in litellm_params
+    mi_dict = {"input_cost_per_token": 0.0001}
+    cost = lp_dict.get("input_cost_per_token")
+    if cost is None:
+        cost = mi_dict.get("input_cost_per_token")
+    assert cost == 0.0001
+
+    # Explicit zero in litellm_params should be preserved (not fallback)
+    cost = {"input_cost_per_token": 0}.get("input_cost_per_token")
+    assert cost == 0