Skip to content

Commit 51c55eb

Browse files
committed
feat: add model integration gpt4
1 parent 4f1ed93 commit 51c55eb

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

scrapegraphai/nodes/generate_answer_from_image_node.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,10 +71,10 @@ async def execute_async(self, state: dict) -> dict:
7171
images = state.get('screenshots', [])
7272
analyses = []
7373

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

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

scrapegraphai/prompts/description_node_prompts.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,4 +7,4 @@
77
following content from a website. \n
88
Please provide a description summary of maximum of 20 words. \n
99
CONTENT OF THE WEBSITE: {content}
10-
"""
10+
"""

0 commit comments

Comments
 (0)