fix: bug on generate answer node

Author: VinciGit00

examples/smart_scraper_graph/ollama/smart_scraper_ollama.py

@@ -11,7 +11,7 @@
 
 graph_config = {
     "llm": {
-        "model": "ollama/llama3.2:3b",
+        "model": "ollama/llama3.2",
         "temperature": 0,
         # "base_url": "http://localhost:11434", # set ollama URL arbitrarily
         "model_tokens": 4096,
@@ -24,7 +24,7 @@
 # Create the SmartScraperGraph instance and run it
 # ************************************************
 smart_scraper_graph = SmartScraperGraph(
-    prompt="Find some information about what does the company do and the list of founders.",
+    prompt="Find some information about the founders.",
     source="https://scrapegraphai.com/",
     config=graph_config,
 )

scrapegraphai/nodes/generate_answer_node.py

@@ -180,9 +180,8 @@ def execute(self, state: dict) -> dict:
         if len(doc) == 1:
             prompt = PromptTemplate(
                 template=template_no_chunks_prompt,
-                input_variables=["question"],
+                input_variables=["content", "question"],
                 partial_variables={
-                    "context": doc,
                     "format_instructions": format_instructions,
                 },
             )
@@ -192,7 +191,7 @@ def execute(self, state: dict) -> dict:
 
             try:
                 answer = self.invoke_with_timeout(
-                    chain, {"question": user_prompt}, self.timeout
+                    chain, {"content": doc, "question": user_prompt}, self.timeout
                 )
             except (Timeout, json.JSONDecodeError) as e:
                 error_msg = (
@@ -216,7 +215,7 @@ def execute(self, state: dict) -> dict:
                 template=template_chunks_prompt,
                 input_variables=["question"],
                 partial_variables={
-                    "context": chunk,
+                    "content": chunk,
                     "chunk_id": i + 1,
                     "format_instructions": format_instructions,
                 },
@@ -242,7 +241,7 @@ def execute(self, state: dict) -> dict:
 
         merge_prompt = PromptTemplate(
             template=template_merge_prompt,
-            input_variables=["context", "question"],
+            input_variables=["content", "question"],
             partial_variables={"format_instructions": format_instructions},
         )
 
@@ -252,7 +251,7 @@ def execute(self, state: dict) -> dict:
         try:
             answer = self.invoke_with_timeout(
                 merge_chain,
-                {"context": batch_results, "question": user_prompt},
+                {"content": batch_results, "question": user_prompt},
                 self.timeout,
             )
         except (Timeout, json.JSONDecodeError) as e:

scrapegraphai/prompts/generate_answer_node_prompts.py

@@ -13,7 +13,7 @@
 and things that will invalidate the dictionary. \n
 Do not start the response with ```json because it will invalidate the postprocessing. \n
 OUTPUT INSTRUCTIONS: {format_instructions}\n
-Content of {chunk_id}: {context}. \n
+Content of {chunk_id}: {content}. \n
 """
 
 TEMPLATE_NO_CHUNKS_MD = """
@@ -27,7 +27,7 @@
 Do not start the response with ```json because it will invalidate the postprocessing. \n
 OUTPUT INSTRUCTIONS: {format_instructions}\n
 USER QUESTION: {question}\n
-WEBSITE CONTENT:  {context}\n
+WEBSITE CONTENT:  {content}\n
 """
 
 TEMPLATE_MERGE_MD = """
@@ -42,7 +42,7 @@
 Do not start the response with ```json because it will invalidate the postprocessing. \n
 OUTPUT INSTRUCTIONS: {format_instructions}\n
 USER QUESTION: {question}\n
-WEBSITE CONTENT: {context}\n
+WEBSITE CONTENT: {content}\n
 """
 
 TEMPLATE_CHUNKS = """
@@ -56,7 +56,7 @@
 and things that will invalidate the dictionary. \n
 Do not start the response with ```json because it will invalidate the postprocessing. \n
 OUTPUT INSTRUCTIONS: {format_instructions}\n
-Content of {chunk_id}: {context}. \n
+Content of {chunk_id}: {content}. \n
 """
 
 TEMPLATE_NO_CHUNKS = """
@@ -70,7 +70,7 @@
 Do not start the response with ```json because it will invalidate the postprocessing. \n
 OUTPUT INSTRUCTIONS: {format_instructions}\n
 USER QUESTION: {question}\n
-WEBSITE CONTENT:  {context}\n
+WEBSITE CONTENT:  {content}\n
 """
 
 TEMPLATE_MERGE = """
@@ -84,7 +84,7 @@
 Do not start the response with ```json because it will invalidate the postprocessing. \n
 OUTPUT INSTRUCTIONS: {format_instructions}\n
 USER QUESTION: {question}\n
-WEBSITE CONTENT: {context}\n
+WEBSITE CONTENT: {content}\n
 """
 
 REGEN_ADDITIONAL_INFO = """