From cef2fdb420253e11c98f800764f352dec55b6159 Mon Sep 17 00:00:00 2001 From: Marco Vinciguerra Date: Sun, 11 Aug 2024 19:19:17 +0200 Subject: [PATCH] Update abstract_graph.py Co-Authored-By: Matteo Vedovati <68272450+vedovati-matteo@users.noreply.github.com> --- scrapegraphai/graphs/abstract_graph.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/scrapegraphai/graphs/abstract_graph.py b/scrapegraphai/graphs/abstract_graph.py index 41e9c9b9..b0da6a53 100644 --- a/scrapegraphai/graphs/abstract_graph.py +++ b/scrapegraphai/graphs/abstract_graph.py @@ -148,11 +148,7 @@ class AbstractGraph(ABC): warnings.simplefilter("ignore") return init_chat_model(**llm_params) - if "azure" in llm_params["model"]: - model_name = llm_params["model"].split("/")[-1] - return handle_model(model_name, "azure_openai", model_name) - - elif "fireworks" in llm_params["model"]: + if "fireworks" in llm_params["model"]: model_name = "/".join(llm_params["model"].split("/")[1:]) token_key = llm_params["model"].split("/")[-1] return handle_model(model_name, "fireworks", token_key)