Skip to content

Commit e69a8ca

Browse files
committed
fix: abstract graph
1 parent 869bbd7 commit e69a8ca

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

scrapegraphai/graphs/abstract_graph.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -164,6 +164,10 @@ def handle_model(model_name, provider, token_key, default_token=8192):
164164
elif llm_params["model"].startswith("vertexai"):
165165
return handle_model(llm_params["model"], "google_vertexai", llm_params["model"])
166166

167+
elif llm_params["model"].startswith("groq"):
168+
model_name = llm_params["model"].split("/")[-1]
169+
return handle_model(model_name, "groq", model_name)
170+
167171
elif "gpt-" in llm_params["model"]:
168172
return handle_model(llm_params["model"], "openai", llm_params["model"])
169173

0 commit comments

Comments
 (0)