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

Fix test failures by updating model ID validation logic

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-25 13:57:48 +00:00
parent 1e7292d0fa
commit e8d61d32db
1 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/crewai/llm.py
View File

@@ -149,14 +149,11 @@ class LLM:
callbacks: List[Any] = [],
**kwargs,
):
# Validate model ID to ensure it's not empty, whitespace, or a numeric value without a provider prefix
if not model or (isinstance(model, str) and model.strip() == ""):
raise ValueError(
f"Invalid model ID: '{model}'. Model ID cannot be empty or whitespace. "
"Please specify a valid model ID with a provider prefix, e.g., 'openai/gpt-4'."
)
if isinstance(model, (int, float)) or (isinstance(model, str) and model.isdigit()):
# Only validate model ID if it's not None and is a numeric value without a provider prefix
if model is not None and (
isinstance(model, (int, float)) or
(isinstance(model, str) and model.strip() and model.strip().isdigit())
):
raise ValueError(
f"Invalid model ID: {model}. Model ID cannot be a numeric value without a provider prefix. "
"Please specify a valid model ID with a provider prefix, e.g., 'openai/gpt-4'."