CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-07-01 21:00:48 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix: Azure OpenAI issue

Browse Source
This commit is contained in:
aziz-ullah-khan 2024-08-21 16:18:51 +05:00
parent 1b488715e6
commit a92b9c6970
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
scrapegraphai/graphs/abstract_graph.py
View File

@ -153,6 +153,10 @@ class AbstractGraph(ABC):
raise ValueError(f"Model '{llm_params['model']}' is not supported")
try:
if "azure" in llm_params["model"]:
model_name = llm_params["model"].split("/")[-1]
return handle_model(model_name, "azure_openai", model_name)
if "fireworks" in llm_params["model"]:
model_name = "/".join(llm_params["model"].split("/")[1:])
token_key = llm_params["model"].split("/")[-1]
@ -271,4 +275,4 @@ class AbstractGraph(ABC):
def run(self) -> str:
"""
Abstract method to execute the graph and return the result.
"""
"""