feat: add model integration gpt4

U-C4N · Oct 24, 2024 · 51c55eb · 51c55eb
1 parent 4f1ed93
commit 51c55eb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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}
-"""
+"""