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

fix: route llm model syntax to litellm

Browse Source 
* fix: route llm model syntax to litellm

* wip: add list of supported models
This commit is contained in:
Greyson LaLonde
2025-11-07 13:34:15 -05:00
committed by GitHub
parent d29867bbb6
commit 1ed307b58c
8 changed files with 802 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
lib/crewai/tests/llms/hooks/test_unsupported_providers.py
View File

@@ -154,7 +154,7 @@ class TestGeminiProviderInterceptor:
# Gemini provider should raise NotImplementedError
with pytest.raises(NotImplementedError) as exc_info:
LLM(
model="gemini/gemini-pro",
model="gemini/gemini-2.5-pro",
interceptor=interceptor,
api_key="test-gemini-key",
)
@@ -169,7 +169,7 @@ class TestGeminiProviderInterceptor:
with pytest.raises(NotImplementedError) as exc_info:
LLM(
model="gemini/gemini-pro",
model="gemini/gemini-2.5-pro",
interceptor=interceptor,
api_key="test-gemini-key",
)
@@ -181,7 +181,7 @@ class TestGeminiProviderInterceptor:
def test_gemini_without_interceptor_works(self) -> None:
"""Test that Gemini LLM works without interceptor."""
llm = LLM(
model="gemini/gemini-pro",
model="gemini/gemini-2.5-pro",
api_key="test-gemini-key",
)
@@ -231,7 +231,7 @@ class TestUnsupportedProviderMessages:
with pytest.raises(NotImplementedError) as exc_info:
LLM(
model="gemini/gemini-pro",
model="gemini/gemini-2.5-pro",
interceptor=interceptor,
api_key="test-gemini-key",
)
@@ -282,7 +282,7 @@ class TestProviderSupportMatrix:
# Gemini - NOT SUPPORTED
with pytest.raises(NotImplementedError):
LLM(
model="gemini/gemini-pro",
model="gemini/gemini-2.5-pro",
interceptor=interceptor,
api_key="test",
)
@@ -315,5 +315,5 @@ class TestProviderSupportMatrix:
assert not hasattr(bedrock_llm, 'interceptor') or bedrock_llm.interceptor is None
# Gemini - doesn't have interceptor attribute
gemini_llm = LLM(model="gemini/gemini-pro", api_key="test")
assert not hasattr(gemini_llm, 'interceptor') or gemini_llm.interceptor is None
gemini_llm = LLM(model="gemini/gemini-2.5-pro", api_key="test")
assert not hasattr(gemini_llm, 'interceptor') or gemini_llm.interceptor is None