Skip to content

Commit ef7a589

Browse files
committed
fix: screenshot_scraper
1 parent 9e9c775 commit ef7a589

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

scrapegraphai/nodes/generate_answer_from_image_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ async def execute_async(self, state: dict) -> dict:
7373

7474
supported_models = ("gpt-4o", "gpt-4o-mini", "gpt-4-turbo")
7575

76-
if self.node_config["config"]["llm"]["model"] not in supported_models:
76+
if self.node_config["config"]["llm"]["model"].split("/")[-1]not in supported_models:
7777
raise ValueError(f"""Model '{self.node_config['config']['llm']['model']}'
7878
is not supported. Supported models are:
7979
{', '.join(supported_models)}.""")

scrapegraphai/nodes/merge_answers_node.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,6 @@ def execute(self, state: dict) -> dict:
7676
self.llm_model = self.llm_model.with_structured_output(
7777
schema = self.node_config["schema"],
7878
method="function_calling") # json schema works only on specific models
79-
8079
# default parser to empty lambda function
8180
output_parser = lambda x: x
8281
if is_basemodel_subclass(self.node_config["schema"]):

0 commit comments

Comments
 (0)