fix: script creator multi

Author: VinciGit00

examples/fireworks/script_multi_generator_fireworks.py

@@ -26,18 +26,18 @@
 # ************************************************
 # Create the ScriptCreatorGraph instance and run it
 # ************************************************
-
 urls=[
-    "https://perinim.github.io/",
-    "https://perinim.github.io/cv/"
+    "https://schultzbergagency.com/emil-raste-karlsen/",
+    "https://schultzbergagency.com/johanna-hedberg/",
 ]
 
 # ************************************************
 # Create the ScriptCreatorGraph instance and run it
 # ************************************************
 
 script_creator_graph = ScriptCreatorMultiGraph(
-    prompt="Who is Marco Perini?",
+    prompt="Find information about actors",
+    # also accepts a string with the already downloaded HTML code
     source=urls,
     config=graph_config
 )

examples/mistral/script_multi_generator_mistral.py

@@ -29,16 +29,17 @@
 # ************************************************
 
 urls=[
-    "https://perinim.github.io/",
-    "https://perinim.github.io/cv/"
+    "https://schultzbergagency.com/emil-raste-karlsen/",
+    "https://schultzbergagency.com/johanna-hedberg/",
 ]
 
 # ************************************************
 # Create the ScriptCreatorGraph instance and run it
 # ************************************************
 
 script_creator_graph = ScriptCreatorMultiGraph(
-    prompt="Who is Marco Perini?",
+    prompt="Find information about actors",
+    # also accepts a string with the already downloaded HTML code
     source=urls,
     config=graph_config
 )

examples/nemotron/script_multi_generator_nemotron.py

@@ -29,16 +29,17 @@
 # ************************************************
 
 urls=[
-    "https://perinim.github.io/",
-    "https://perinim.github.io/cv/"
+    "https://schultzbergagency.com/emil-raste-karlsen/",
+    "https://schultzbergagency.com/johanna-hedberg/",
 ]
 
 # ************************************************
 # Create the ScriptCreatorGraph instance and run it
 # ************************************************
 
 script_creator_graph = ScriptCreatorMultiGraph(
-    prompt="Who is Marco Perini?",
+    prompt="Find information about actors",
+    # also accepts a string with the already downloaded HTML code
     source=urls,
     config=graph_config
 )

examples/openai/script_multi_generator_openai.py

@@ -29,16 +29,17 @@
 # ************************************************
 
 urls=[
-    "https://perinim.github.io/",
-    "https://perinim.github.io/cv/"
+    "https://schultzbergagency.com/emil-raste-karlsen/",
+    "https://schultzbergagency.com/johanna-hedberg/",
 ]
 
 # ************************************************
 # Create the ScriptCreatorGraph instance and run it
 # ************************************************
 
 script_creator_graph = ScriptCreatorMultiGraph(
-    prompt="Who is Marco Perini?",
+    prompt="Find information about actors",
+    # also accepts a string with the already downloaded HTML code
     source=urls,
     config=graph_config
 )

scrapegraphai/graphs/json_scraper_multi_graph.py

@@ -45,9 +45,7 @@ def __init__(self, prompt: str, source: List[str],
                  config: dict, schema: Optional[BaseModel] = None):
 
         self.max_results = config.get("max_results", 3)
-
         self.copy_config = safe_deepcopy(config)
-
         self.copy_schema = deepcopy(schema)
 
         super().__init__(prompt, config, source, schema)

scrapegraphai/graphs/markdown_scraper_multi_graph.py

@@ -1,7 +1,7 @@
 """
 MDScraperMultiGraph Module
 """
-from copy import copy, deepcopy
+from copy import deepcopy
 from typing import List, Optional
 from pydantic import BaseModel
 from .base_graph import BaseGraph

scrapegraphai/graphs/script_creator_multi_graph.py

@@ -1,6 +1,7 @@
 """ 
 ScriptCreatorMultiGraph Module
 """
+from copy import deepcopy
 from typing import List, Optional
 from pydantic import BaseModel
 from .base_graph import BaseGraph
@@ -45,7 +46,7 @@ def __init__(self, prompt: str, source: List[str],
         self.max_results = config.get("max_results", 3)
 
         self.copy_config = safe_deepcopy(config)
-
+        self.copy_schema = deepcopy(schema)
         super().__init__(prompt, config, source, schema)
 
     def _create_graph(self) -> BaseGraph:
@@ -55,19 +56,14 @@ def _create_graph(self) -> BaseGraph:
             BaseGraph: A graph instance representing the web scraping and searching workflow.
         """
 
-        script_generator_instance = ScriptCreatorGraph(
-            prompt="",
-            source="",
-            config=self.copy_config,
-            schema=self.schema
-        )
-
         graph_iterator_node = GraphIteratorNode(
             input="user_prompt & urls",
             output=["scripts"],
             node_config={
-                "graph_instance": script_generator_instance,
-            }
+                "graph_instance": ScriptCreatorGraph,
+                "scraper_config": self.copy_config,
+            },
+            schema=self.copy_schema
         )
 
         merge_scripts_node = MergeGeneratedScriptsNode(

scrapegraphai/graphs/smart_scraper_multi_graph.py

@@ -60,13 +60,6 @@ def _create_graph(self) -> BaseGraph:
             BaseGraph: A graph instance representing the web scraping and searching workflow.
         """
 
-        # smart_scraper_instance = SmartScraperGraph(
-        #     prompt="",
-        #     source="",
-        #     config=self.copy_config,
-        #     schema=self.copy_schema
-        # )
-
         graph_iterator_node = GraphIteratorNode(
             input="user_prompt & urls",
             output=["results"],

scrapegraphai/graphs/xml_scraper_multi_graph.py

@@ -45,9 +45,7 @@ def __init__(self, prompt: str, source: List[str],
                  config: dict, schema: Optional[BaseModel] = None):
 
         self.copy_config = safe_deepcopy(config)
-
         self.copy_schema = deepcopy(schema)
-
         super().__init__(prompt, config, source, schema)
 
     def _create_graph(self) -> BaseGraph:
@@ -57,14 +55,6 @@ def _create_graph(self) -> BaseGraph:
         Returns:
             BaseGraph: A graph instance representing the web scraping and searching workflow.
         """
-
-        # smart_scraper_instance = XMLScraperGraph(
-        #     prompt="",
-        #     source="",
-        #     config=self.copy_config,
-        #     schema=self.copy_schema
-        # )
-
         graph_iterator_node = GraphIteratorNode(
             input="user_prompt & jsons",
             output=["results"],