andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-05 09:12:39 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Update llm.py (gemini 2.0)

Browse Source 
Add setting for Gemini 2.0 context window to llm.py
This commit is contained in:
alan blount
2024-12-17 11:19:17 -05:00
committed by GitHub
parent c3ffeee26b
commit 70e4a2ee85
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/crewai/llm.py
View File

@@ -44,6 +44,7 @@ LLM_CONTEXT_WINDOW_SIZES = {
"o1-preview": 128000,
"o1-mini": 128000,
# gemini
"gemini-2.0-flash": 1048576,
"gemini-1.5-pro": 2097152,
"gemini-1.5-flash": 1048576,
"gemini-1.5-flash-8b": 1048576,