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

fix: generate answer node omni

Browse Source
This commit is contained in:
Marco Vinciguerra 2024-08-07 14:09:02 +02:00
parent 928f7040ab
commit b52e4a390b
3 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
examples/local_models/smart_scraper_ollama.py
View File

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

2
scrapegraphai/nodes/generate_answer_omni_node.py
View File

@ -105,7 +105,7 @@ class GenerateAnswerOmniNode(BaseNode):
template=template_no_chunk_omni_prompt, template=template_no_chunk_omni_prompt,
input_variables=["question"], input_variables=["question"],
partial_variables={ partial_variables={
"context": chunk, "context": doc,
"format_instructions": format_instructions, "format_instructions": format_instructions,
"img_desc": imag_desc, "img_desc": imag_desc,
}, },

2
scrapegraphai/nodes/generate_answer_pdf_node.py
View File

@ -114,7 +114,7 @@ class GenerateAnswerPDFNode(BaseNode):
template=template_no_chunks_pdf_prompt, template=template_no_chunks_pdf_prompt,
input_variables=["question"], input_variables=["question"],
partial_variables={ partial_variables={
"context":doc, "context": doc,
"format_instructions": format_instructions, "format_instructions": format_instructions,
}, },
) )