From ef7a5891dcb1b4ed8a97947f5563fa78af917ecb Mon Sep 17 00:00:00 2001 From: Marco Vinciguerra Date: Sat, 7 Sep 2024 14:55:12 +0200 Subject: [PATCH] fix: screenshot_scraper --- scrapegraphai/nodes/generate_answer_from_image_node.py | 2 +- scrapegraphai/nodes/merge_answers_node.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/scrapegraphai/nodes/generate_answer_from_image_node.py b/scrapegraphai/nodes/generate_answer_from_image_node.py index 4cc93d18..7134cabe 100644 --- a/scrapegraphai/nodes/generate_answer_from_image_node.py +++ b/scrapegraphai/nodes/generate_answer_from_image_node.py @@ -73,7 +73,7 @@ class GenerateAnswerFromImageNode(BaseNode): supported_models = ("gpt-4o", "gpt-4o-mini", "gpt-4-turbo") - if self.node_config["config"]["llm"]["model"] not in supported_models: + if self.node_config["config"]["llm"]["model"].split("/")[-1]not in supported_models: raise ValueError(f"""Model '{self.node_config['config']['llm']['model']}' is not supported. Supported models are: {', '.join(supported_models)}.""") diff --git a/scrapegraphai/nodes/merge_answers_node.py b/scrapegraphai/nodes/merge_answers_node.py index 1e25cccb..ff27d7b3 100644 --- a/scrapegraphai/nodes/merge_answers_node.py +++ b/scrapegraphai/nodes/merge_answers_node.py @@ -76,7 +76,6 @@ class MergeAnswersNode(BaseNode): self.llm_model = self.llm_model.with_structured_output( schema = self.node_config["schema"], method="function_calling") # json schema works only on specific models - # default parser to empty lambda function output_parser = lambda x: x if is_basemodel_subclass(self.node_config["schema"]):