25 lines
726 B
Python
25 lines
726 B
Python
import unittest
|
|
|
|
from cli.utils import CUSTOM_OPENROUTER_MODEL, resolve_model_choice
|
|
|
|
|
|
class OpenRouterModelSelectionTests(unittest.TestCase):
|
|
def test_builtin_model_is_returned_unchanged(self):
|
|
self.assertEqual(
|
|
resolve_model_choice("openrouter", "z-ai/glm-4.5-air:free", "Quick-Thinking"),
|
|
"z-ai/glm-4.5-air:free",
|
|
)
|
|
|
|
def test_custom_model_prompt_value_is_trimmed(self):
|
|
chosen = resolve_model_choice(
|
|
"openrouter",
|
|
CUSTOM_OPENROUTER_MODEL,
|
|
"Deep-Thinking",
|
|
prompt_fn=lambda _: " minimax/minimax-m2.1 ",
|
|
)
|
|
self.assertEqual(chosen, "minimax/minimax-m2.1")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|