20 lines
819 B
Python
20 lines
819 B
Python
from constants.llm import DEFAULT_CODEX_MODEL, SUPPORTED_CODEX_MODELS
|
|
from enums.llm_provider import LLMProvider
|
|
from utils import llm_provider
|
|
|
|
|
|
def test_codex_supported_model_is_preserved(monkeypatch):
|
|
monkeypatch.setattr(llm_provider, "get_llm_provider", lambda: LLMProvider.CODEX)
|
|
monkeypatch.setattr(llm_provider, "get_codex_model_env", lambda: "gpt-5.4-mini")
|
|
|
|
assert llm_provider.get_model() == "gpt-5.4-mini"
|
|
|
|
|
|
def test_codex_deprecated_model_falls_back_to_default(monkeypatch):
|
|
monkeypatch.setattr(llm_provider, "get_llm_provider", lambda: LLMProvider.CODEX)
|
|
monkeypatch.setattr(llm_provider, "get_codex_model_env", lambda: "gpt-5.2")
|
|
|
|
assert DEFAULT_CODEX_MODEL == "gpt-5.5"
|
|
assert "gpt-5.2" not in SUPPORTED_CODEX_MODELS
|
|
assert llm_provider.get_model() == DEFAULT_CODEX_MODEL
|