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)