andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-15 02:58:30 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fixes #2511

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

1
src/crewai/cli/constants.py
View File

@@ -123,6 +123,7 @@ MODELS = {
"claude-3-haiku-20240307",
],
"gemini": [
"gemini/gemini-2.5-pro-exp-03-25",
"gemini/gemini-1.5-flash",
"gemini/gemini-1.5-pro",
"gemini/gemini-gemma-2-9b-it",

1
src/crewai/llm.py
View File

@@ -81,6 +81,7 @@ LLM_CONTEXT_WINDOW_SIZES = {
"o1-mini": 128000,
"o3-mini": 200000, # Based on official o3-mini specifications
# gemini
"gemini-2.5-pro-exp-03-25": 2097152,
"gemini-2.0-flash": 1048576,
"gemini-1.5-pro": 2097152,
"gemini-1.5-flash": 1048576,

14
tests/test_gemini_2_5_support.py Normal file
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