CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-07-06 21:11:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: add forcing format as json

Browse Source
This commit is contained in:
Marco Vinciguerra 2024-06-02 12:24:54 +02:00
parent 38d138e36f
commit 5cfc10178a
5 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
examples/local_models/xml_scraper_ollama.py
View File

@ -27,7 +27,6 @@ graph_config = {
"llm": { "llm": {
"model": "ollama/llama3", "model": "ollama/llama3",
"temperature": 0, "temperature": 0,
"format": "json", # Ollama needs the format to be specified explicitly
# "model_tokens": 2000, # set context length arbitrarily # "model_tokens": 2000, # set context length arbitrarily
"base_url": "http://localhost:11434", "base_url": "http://localhost:11434",
}, },

1
scrapegraphai/nodes/generate_answer_csv_node.py
View File

@ -59,6 +59,7 @@ class GenerateAnswerCSVNode(BaseNode):
""" """
super().__init__(node_name, "node", input, output, 2, node_config) super().__init__(node_name, "node", input, output, 2, node_config)
self.llm_model = node_config["llm_model"] self.llm_model = node_config["llm_model"]
self.llm_model.format="json"
self.verbose = ( self.verbose = (
False if node_config is None else node_config.get("verbose", False) False if node_config is None else node_config.get("verbose", False)
) )

2
scrapegraphai/nodes/generate_answer_node.py
View File

@ -44,8 +44,8 @@ class GenerateAnswerNode(BaseNode):
node_name: str = "GenerateAnswer", node_name: str = "GenerateAnswer",
): ):
super().__init__(node_name, "node", input, output, 2, node_config) super().__init__(node_name, "node", input, output, 2, node_config)
self.llm_model = node_config["llm_model"] self.llm_model = node_config["llm_model"]
self.llm_model.format="json"
self.verbose = ( self.verbose = (
True if node_config is None else node_config.get("verbose", False) True if node_config is None else node_config.get("verbose", False)
) )

1
scrapegraphai/nodes/generate_answer_omni_node.py
View File

@ -44,6 +44,7 @@ class GenerateAnswerOmniNode(BaseNode):
super().__init__(node_name, "node", input, output, 3, node_config) super().__init__(node_name, "node", input, output, 3, node_config)
self.llm_model = node_config["llm_model"] self.llm_model = node_config["llm_model"]
self.llm_model.format="json"
self.verbose = ( self.verbose = (
False if node_config is None else node_config.get("verbose", False) False if node_config is None else node_config.get("verbose", False)
) )

1
scrapegraphai/nodes/generate_answer_pdf_node.py
View File

@ -58,6 +58,7 @@ class GenerateAnswerPDFNode(BaseNode):
""" """
super().__init__(node_name, "node", input, output, 2, node_config) super().__init__(node_name, "node", input, output, 2, node_config)
self.llm_model = node_config["llm_model"] self.llm_model = node_config["llm_model"]
self.llm_model.format="json"
self.verbose = ( self.verbose = (
False if node_config is None else node_config.get("verbose", False) False if node_config is None else node_config.get("verbose", False)
) )