diff --git a/scrapegraphai/graphs/abstract_graph.py b/scrapegraphai/graphs/abstract_graph.py index 555e8211..5d5e407c 100644 --- a/scrapegraphai/graphs/abstract_graph.py +++ b/scrapegraphai/graphs/abstract_graph.py @@ -164,6 +164,10 @@ def handle_model(model_name, provider, token_key, default_token=8192): elif llm_params["model"].startswith("vertexai"): return handle_model(llm_params["model"], "google_vertexai", llm_params["model"]) + elif llm_params["model"].startswith("groq"): + model_name = llm_params["model"].split("/")[-1] + return handle_model(model_name, "groq", model_name) + elif "gpt-" in llm_params["model"]: return handle_model(llm_params["model"], "openai", llm_params["model"])