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

feat: add model integration gpt4

Browse Source
This commit is contained in:
Marco Vinciguerra 2024-10-24 09:10:51 +02:00
parent 4f1ed939e6
commit 51c55eb3a2
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scrapegraphai/nodes/generate_answer_from_image_node.py
View File

@ -71,10 +71,10 @@ class GenerateAnswerFromImageNode(BaseNode):
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)}.""")

2
scrapegraphai/prompts/description_node_prompts.py
View File

@ -7,4 +7,4 @@ You are a scraper and you have just scraped the
following content from a website. \n
Please provide a description summary of maximum of 20 words. \n
CONTENT OF THE WEBSITE: {content}
"""
"""