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

Enhance create_llm function to support BaseLLM type

Browse Source 
- Update the create_llm function to accept both LLM and BaseLLM instances
- Ensure compatibility with existing LLM handling logic
This commit is contained in:
Lorenze Jay
2025-03-12 14:56:47 -07:00
parent 902c330113
commit 80c31fb55b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/crewai/utilities/llm_utils.py
View File

@@ -22,8 +22,8 @@ def create_llm(
A BaseLLM instance if successful, or None if something fails. A BaseLLM instance if successful, or None if something fails.
""" """
# 1) If llm_value is already a BaseLLM object, return it directly # 1) If llm_value is already a BaseLLM or LLM object, return it directly
if isinstance(llm_value, LLM): if isinstance(llm_value, LLM) or isinstance(llm_value, BaseLLM):
return llm_value return llm_value
# 2) If llm_value is a string (model name) # 2) If llm_value is a string (model name)