CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-07-01 21:00:48 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(DeepSeek): proper model initialization

Browse Source
This commit is contained in:
Federico Aguzzi 2024-09-02 12:28:22 +02:00
parent 13efd4e3a4
commit 74dfc693f6
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
scrapegraphai/graphs/abstract_graph.py
View File

@ -150,24 +150,26 @@ class AbstractGraph(ABC):
warnings.simplefilter("ignore") warnings.simplefilter("ignore")
return init_chat_model(**llm_params) return init_chat_model(**llm_params)
else: else:
if llm_params["model_provider"] == "deepseek": model_provider = llm_params.pop("model_provider")
if model_provider == "deepseek":
return DeepSeek(**llm_params) return DeepSeek(**llm_params)
if llm_params["model_provider"] == "ernie": if model_provider == "ernie":
from langchain_community.chat_models import ErnieBotChat from langchain_community.chat_models import ErnieBotChat
return ErnieBotChat(**llm_params) return ErnieBotChat(**llm_params)
elif llm_params["model_provider"] == "oneapi": elif model_provider == "oneapi":
return OneApi(**llm_params) return OneApi(**llm_params)
elif llm_params["model_provider"] == "togehterai": elif model_provider == "togehterai":
try: try:
from langchain_together import ChatTogether from langchain_together import ChatTogether
except ImportError: except ImportError:
raise ImportError("The langchain_together module is not installed. Please install it using `pip install scrapegraphai[other-language-models]`.") raise ImportError("The langchain_together module is not installed. Please install it using `pip install scrapegraphai[other-language-models]`.")
return ChatTogether(**llm_params) return ChatTogether(**llm_params)
elif llm_params["model_provider"] == "nvidia": elif model_provider == "nvidia":
try: try:
from langchain_nvidia_ai_endpoints import ChatNVIDIA from langchain_nvidia_ai_endpoints import ChatNVIDIA
except ImportError: except ImportError: