diff --git a/scrapegraphai/nodes/generate_answer_from_image_node.py b/scrapegraphai/nodes/generate_answer_from_image_node.py index 7134cabe..9359b2bb 100644 --- a/scrapegraphai/nodes/generate_answer_from_image_node.py +++ b/scrapegraphai/nodes/generate_answer_from_image_node.py @@ -71,10 +71,10 @@ async def execute_async(self, state: dict) -> dict: images = state.get('screenshots', []) analyses = [] - supported_models = ("gpt-4o", "gpt-4o-mini", "gpt-4-turbo") + supported_models = ("gpt-4o", "gpt-4o-mini", "gpt-4-turbo", "gpt-4") if self.node_config["config"]["llm"]["model"].split("/")[-1]not in supported_models: - raise ValueError(f"""Model '{self.node_config['config']['llm']['model']}' + raise ValueError(f"""The model provided is not supported. Supported models are: {', '.join(supported_models)}.""") diff --git a/scrapegraphai/prompts/description_node_prompts.py b/scrapegraphai/prompts/description_node_prompts.py index 86264d0b..944ed24e 100644 --- a/scrapegraphai/prompts/description_node_prompts.py +++ b/scrapegraphai/prompts/description_node_prompts.py @@ -7,4 +7,4 @@ following content from a website. \n Please provide a description summary of maximum of 20 words. \n CONTENT OF THE WEBSITE: {content} -""" \ No newline at end of file +"""