Skip to content

Commit c3f1520

Browse files
committed
Update abstract_graph.py
1 parent b470d97 commit c3f1520

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scrapegraphai/graphs/abstract_graph.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -228,7 +228,7 @@ def handle_model(model_name, provider, token_key, default_token=8192):
228228
llm_params["model"] = "/".join(llm_params["model"].split("/")[1:])
229229
except KeyError as exc:
230230
raise KeyError("Model not supported") from exc
231-
return ChatNVIDIA(llm_params)
231+
return ChatNVIDIA(**llm_config)
232232

233233
# Raise an error if the model did not match any of the previous cases
234234
raise ValueError("Model provided by the configuration not supported")

0 commit comments

Comments
 (0)