File tree Expand file tree Collapse file tree 1 file changed +7
-5
lines changed Expand file tree Collapse file tree 1 file changed +7
-5
lines changed Original file line number Diff line number Diff line change @@ -150,24 +150,26 @@ def _create_llm(self, llm_config: dict) -> object:
150150 warnings .simplefilter ("ignore" )
151151 return init_chat_model (** llm_params )
152152 else :
153- if llm_params ["model_provider" ] == "deepseek" :
153+ model_provider = llm_params .pop ("model_provider" )
154+
155+ if model_provider == "deepseek" :
154156 return DeepSeek (** llm_params )
155157
156- if llm_params [ " model_provider" ] == "ernie" :
158+ if model_provider == "ernie" :
157159 from langchain_community .chat_models import ErnieBotChat
158160 return ErnieBotChat (** llm_params )
159161
160- elif llm_params [ " model_provider" ] == "oneapi" :
162+ elif model_provider == "oneapi" :
161163 return OneApi (** llm_params )
162164
163- elif llm_params [ " model_provider" ] == "togehterai" :
165+ elif model_provider == "togehterai" :
164166 try :
165167 from langchain_together import ChatTogether
166168 except ImportError :
167169 raise ImportError ("The langchain_together module is not installed. Please install it using `pip install scrapegraphai[other-language-models]`." )
168170 return ChatTogether (** llm_params )
169171
170- elif llm_params [ " model_provider" ] == "nvidia" :
172+ elif model_provider == "nvidia" :
171173 try :
172174 from langchain_nvidia_ai_endpoints import ChatNVIDIA
173175 except ImportError :
You can’t perform that action at this time.
0 commit comments