From 80c31fb55bea8a4f1d91e05f6baabe2f2d94bf88 Mon Sep 17 00:00:00 2001 From: Lorenze Jay Date: Wed, 12 Mar 2025 14:56:47 -0700 Subject: [PATCH] Enhance create_llm function to support BaseLLM type - Update the create_llm function to accept both LLM and BaseLLM instances - Ensure compatibility with existing LLM handling logic --- src/crewai/utilities/llm_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/crewai/utilities/llm_utils.py b/src/crewai/utilities/llm_utils.py index 2242d305b..1eb0a4693 100644 --- a/src/crewai/utilities/llm_utils.py +++ b/src/crewai/utilities/llm_utils.py @@ -22,8 +22,8 @@ def create_llm( A BaseLLM instance if successful, or None if something fails. """ - # 1) If llm_value is already a BaseLLM object, return it directly - if isinstance(llm_value, LLM): + # 1) If llm_value is already a BaseLLM or LLM object, return it directly + if isinstance(llm_value, LLM) or isinstance(llm_value, BaseLLM): return llm_value # 2) If llm_value is a string (model name)