From 70e4a2ee8553aa97fdb6161acfc922fea2a2d18c Mon Sep 17 00:00:00 2001
From: alan blount <alan@zeroasterisk.com>
Date: Tue, 17 Dec 2024 11:19:17 -0500
Subject: [PATCH] Update llm.py (gemini 2.0)

Add setting for Gemini 2.0 context window to llm.py
---
 src/crewai/llm.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/crewai/llm.py b/src/crewai/llm.py
index 058589056..1b0ac9b0a 100644
--- a/src/crewai/llm.py
+++ b/src/crewai/llm.py
@@ -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,