Add support for OpenAI's o4-mini model

Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-05-05 05:18:40 +00:00
parent 17474a3a0c
commit 018fbd044d
4 changed files with 59 additions and 9 deletions

View File

@@ -337,9 +337,13 @@ def test_o3_mini_reasoning_effort_medium():
def test_context_window_validation():
"""Test that context window validation works correctly."""
# Test valid window size
# Test valid window size for o3-mini
llm = LLM(model="o3-mini")
assert llm.get_context_window_size() == int(200000 * CONTEXT_WINDOW_USAGE_RATIO)
# Test valid window size for o4-mini
llm = LLM(model="o4-mini")
assert llm.get_context_window_size() == int(200000 * CONTEXT_WINDOW_USAGE_RATIO)
# Test invalid window size
with pytest.raises(ValueError) as excinfo: