fix: generate answer node omni

Author: VinciGit00

examples/local_models/smart_scraper_ollama.py

@@ -22,10 +22,9 @@
 # ************************************************
 # Create the SmartScraperGraph instance and run it
 # ************************************************
-
 smart_scraper_graph = SmartScraperGraph(
-    prompt="List me all the titles of the website",
-    source="https://sport.sky.it/nba?gr=www",
+    prompt="Find some information about what does the company do, the name and a contact email.",
+    source="https://scrapegraphai.com/",
     config=graph_config
 )
 

scrapegraphai/nodes/generate_answer_omni_node.py

@@ -105,7 +105,7 @@ def execute(self, state: dict) -> dict:
                 template=template_no_chunk_omni_prompt,
                 input_variables=["question"],
                 partial_variables={
-                    "context": chunk,
+                    "context": doc,
                     "format_instructions": format_instructions,
                     "img_desc": imag_desc,
                 },

scrapegraphai/nodes/generate_answer_pdf_node.py

@@ -56,7 +56,7 @@ def __init__(
         self.llm_model = node_config["llm_model"]
         if isinstance(node_config["llm_model"], ChatOllama):
             self.llm_model.format="json"
-   
+
         self.verbose = (
             False if node_config is None else node_config.get("verbose", False)
         )
@@ -114,7 +114,7 @@ def execute(self, state):
                 template=template_no_chunks_pdf_prompt,
                 input_variables=["question"],
                 partial_variables={
-                    "context":doc,
+                    "context": doc,
                     "format_instructions": format_instructions,
                 },
             )