Skip to content

Commit 74dfc69

Browse files
committed
fix(DeepSeek): proper model initialization
1 parent 13efd4e commit 74dfc69

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

scrapegraphai/graphs/abstract_graph.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff 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:

0 commit comments

Comments
 (0)