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:
150
150
warnings .simplefilter ("ignore" )
151
151
return init_chat_model (** llm_params )
152
152
else :
153
- if llm_params ["model_provider" ] == "deepseek" :
153
+ model_provider = llm_params .pop ("model_provider" )
154
+
155
+ if model_provider == "deepseek" :
154
156
return DeepSeek (** llm_params )
155
157
156
- if llm_params [ " model_provider" ] == "ernie" :
158
+ if model_provider == "ernie" :
157
159
from langchain_community .chat_models import ErnieBotChat
158
160
return ErnieBotChat (** llm_params )
159
161
160
- elif llm_params [ " model_provider" ] == "oneapi" :
162
+ elif model_provider == "oneapi" :
161
163
return OneApi (** llm_params )
162
164
163
- elif llm_params [ " model_provider" ] == "togehterai" :
165
+ elif model_provider == "togehterai" :
164
166
try :
165
167
from langchain_together import ChatTogether
166
168
except ImportError :
167
169
raise ImportError ("The langchain_together module is not installed. Please install it using `pip install scrapegraphai[other-language-models]`." )
168
170
return ChatTogether (** llm_params )
169
171
170
- elif llm_params [ " model_provider" ] == "nvidia" :
172
+ elif model_provider == "nvidia" :
171
173
try :
172
174
from langchain_nvidia_ai_endpoints import ChatNVIDIA
173
175
except ImportError :
You can’t perform that action at this time.
0 commit comments