Skip to content

Commit df70b4f

Browse files
committed
Update abstract_graph.py
1 parent cf73883 commit df70b4f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

scrapegraphai/graphs/abstract_graph.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -125,19 +125,19 @@ def _create_llm(self, llm_config: dict) -> object:
125125
self.model_token = llm_params["model_tokens"]
126126
except KeyError as exc:
127127
raise KeyError("model_tokens not specified") from exc
128-
return llm_params["model_instance"]
128+
return llm_params["model_instance"]
129129

130130
known_providers = {"openai", "azure_openai", "google_genai", "google_vertexai",
131131
"ollama", "oneapi", "nvidia", "groq", "anthropic" "bedrock", "mistralai",
132132
"hugging_face", "deepseek", "ernie", "fireworks"}
133-
133+
134134
split_model_provider = llm_params["model"].split("/")
135135
llm_params["model_provider"] = split_model_provider[0]
136136
llm_params["model"] = split_model_provider[1:]
137137

138138
if llm_params["model_provider"] not in known_providers:
139139
raise ValueError(f"Provider {llm_params['model_provider']} is not supported. If possible, try to use a model instance instead.")
140-
140+
141141
try:
142142
self.model_token = models_tokens[llm_params["model_provider"]].get(llm_params["model"][0])
143143
except KeyError:

0 commit comments

Comments
 (0)