Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-04-02 02:17:43 +00:00
parent a3b5413f16
commit 5d4395b0b0
3 changed files with 16 additions and 0 deletions

View File

@@ -0,0 +1,14 @@
import pytest
from crewai.llm import LLM
def test_get_custom_llm_provider_gemini_2_5():
"""Test that the Gemini 2.5 model is correctly identified as a Gemini provider."""
llm = LLM(model="gemini/gemini-2.5-pro-exp-03-25")
assert llm._get_custom_llm_provider() == "gemini"
def test_gemini_2_5_context_window_size():
"""Test that the Gemini 2.5 model has the correct context window size."""
llm = LLM(model="gemini-2.5-pro-exp-03-25")
from crewai.llm import CONTEXT_WINDOW_USAGE_RATIO
expected_size = int(2097152 * CONTEXT_WINDOW_USAGE_RATIO)
assert llm.get_context_window_size() == expected_size