diff --git a/examples/local_models/xml_scraper_ollama.py b/examples/local_models/xml_scraper_ollama.py index f13122f7..cc8c3ad9 100644 --- a/examples/local_models/xml_scraper_ollama.py +++ b/examples/local_models/xml_scraper_ollama.py @@ -27,7 +27,6 @@ graph_config = { "llm": { "model": "ollama/llama3", "temperature": 0, - "format": "json", # Ollama needs the format to be specified explicitly # "model_tokens": 2000, # set context length arbitrarily "base_url": "http://localhost:11434", }, diff --git a/scrapegraphai/nodes/generate_answer_csv_node.py b/scrapegraphai/nodes/generate_answer_csv_node.py index e12c64f9..c12e0688 100644 --- a/scrapegraphai/nodes/generate_answer_csv_node.py +++ b/scrapegraphai/nodes/generate_answer_csv_node.py @@ -59,6 +59,7 @@ class GenerateAnswerCSVNode(BaseNode): """ super().__init__(node_name, "node", input, output, 2, node_config) self.llm_model = node_config["llm_model"] + self.llm_model.format="json" self.verbose = ( False if node_config is None else node_config.get("verbose", False) ) diff --git a/scrapegraphai/nodes/generate_answer_node.py b/scrapegraphai/nodes/generate_answer_node.py index 26a2ed66..44122176 100644 --- a/scrapegraphai/nodes/generate_answer_node.py +++ b/scrapegraphai/nodes/generate_answer_node.py @@ -44,8 +44,8 @@ class GenerateAnswerNode(BaseNode): node_name: str = "GenerateAnswer", ): super().__init__(node_name, "node", input, output, 2, node_config) - self.llm_model = node_config["llm_model"] + self.llm_model.format="json" self.verbose = ( True if node_config is None else node_config.get("verbose", False) ) diff --git a/scrapegraphai/nodes/generate_answer_omni_node.py b/scrapegraphai/nodes/generate_answer_omni_node.py index 2b9281ed..9a0aacc4 100644 --- a/scrapegraphai/nodes/generate_answer_omni_node.py +++ b/scrapegraphai/nodes/generate_answer_omni_node.py @@ -44,6 +44,7 @@ class GenerateAnswerOmniNode(BaseNode): super().__init__(node_name, "node", input, output, 3, node_config) self.llm_model = node_config["llm_model"] + self.llm_model.format="json" self.verbose = ( False if node_config is None else node_config.get("verbose", False) ) diff --git a/scrapegraphai/nodes/generate_answer_pdf_node.py b/scrapegraphai/nodes/generate_answer_pdf_node.py index 3a520745..40ec1889 100644 --- a/scrapegraphai/nodes/generate_answer_pdf_node.py +++ b/scrapegraphai/nodes/generate_answer_pdf_node.py @@ -58,6 +58,7 @@ class GenerateAnswerPDFNode(BaseNode): """ super().__init__(node_name, "node", input, output, 2, node_config) self.llm_model = node_config["llm_model"] + self.llm_model.format="json" self.verbose = ( False if node_config is None else node_config.get("verbose", False) )