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

fix: abstract_graph moel token bug

Browse Source
This commit is contained in:
Marco Vinciguerra 2024-07-21 18:37:15 +02:00
parent 208ab267ce
commit ce6be37fbc
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scrapegraphai/graphs/abstract_graph.py
View File

@ -148,6 +148,10 @@ class AbstractGraph(ABC):
# If model instance is passed directly instead of the model details # If model instance is passed directly instead of the model details
if "model_instance" in llm_params: if "model_instance" in llm_params:
try:
self.model_token = llm_params["model_tokens"]
except KeyError as exc:
raise KeyError("model_tokens not specified") from exc
return llm_params["model_instance"] return llm_params["model_instance"]
# Instantiate the language model based on the model name # Instantiate the language model based on the model name