Skip to content

Commit cef2fdb

Browse files
Update abstract_graph.py
Co-Authored-By: Matteo Vedovati <[email protected]>
1 parent c77231c commit cef2fdb

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

scrapegraphai/graphs/abstract_graph.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -148,11 +148,7 @@ def handle_model(model_name, provider, token_key, default_token=8192):
148148
warnings.simplefilter("ignore")
149149
return init_chat_model(**llm_params)
150150

151-
if "azure" in llm_params["model"]:
152-
model_name = llm_params["model"].split("/")[-1]
153-
return handle_model(model_name, "azure_openai", model_name)
154-
155-
elif "fireworks" in llm_params["model"]:
151+
if "fireworks" in llm_params["model"]:
156152
model_name = "/".join(llm_params["model"].split("/")[1:])
157153
token_key = llm_params["model"].split("/")[-1]
158154
return handle_model(model_name, "fireworks", token_key)

0 commit comments

Comments
 (0)