CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-07-01 21:00:48 +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', []) images = state.get('screenshots', [])
analyses = [] 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: 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: is not supported. Supported models are:
{', '.join(supported_models)}.""") {', '.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 following content from a website. \n
Please provide a description summary of maximum of 20 words. \n Please provide a description summary of maximum of 20 words. \n
CONTENT OF THE WEBSITE: {content} CONTENT OF THE WEBSITE: {content}
""" """