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

feat: Add gemini-3-pro-preview (#3950)

Browse Source 
* Add gemini-3-pro-preview

Also refactors the tool support check for better forward compatibility.

* Add cassette for Gemini 3 Pro

---------

Co-authored-by: Greyson LaLonde <greyson.r.lalonde@gmail.com>
This commit is contained in:
Mark McDonald
2025-11-25 03:49:29 +08:00
committed by GitHub
parent b759654e7d
commit a978267fa2
7 changed files with 78 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
lib/crewai/src/crewai/llms/providers/gemini/completion.py
View File

@@ -1,5 +1,6 @@
import logging
import os
import re
from typing import Any, cast
from pydantic import BaseModel
@@ -100,9 +101,8 @@ class GeminiCompletion(BaseLLM):
self.stop_sequences = stop_sequences or []
# Model-specific settings
self.is_gemini_2 = "gemini-2" in model.lower()
self.is_gemini_1_5 = "gemini-1.5" in model.lower()
self.supports_tools = self.is_gemini_1_5 or self.is_gemini_2
version_match = re.search(r"gemini-(\d+(?:\.\d+)?)", model.lower())
self.supports_tools = bool(version_match and float(version_match.group(1)) >= 1.5)
@property
def stop(self) -> list[str]:
@@ -559,6 +559,7 @@ class GeminiCompletion(BaseLLM):
)
context_windows = {
"gemini-3-pro-preview": 1048576, # 1M tokens
"gemini-2.0-flash": 1048576, # 1M tokens
"gemini-2.0-flash-thinking": 32768,
"gemini-2.0-flash-lite": 1048576,