Scrapegraph-ai/tests/graphs/search_link_ollama.py
Marco Vinciguerra cec5537f2a
Some checks are pending
/ build (push) Waiting to run
Release / Build (push) Waiting to run
Release / Release (push) Blocked by required conditions
add new tests
Co-Authored-By: Matteo Vedovati <68272450+vedovati-matteo@users.noreply.github.com>
2024-08-11 17:10:55 +02:00

23 lines
541 B
Python

from scrapegraphai.graphs import SearchLinkGraph
from scrapegraphai.utils import prettify_exec_info
def test_smart_scraper_pipeline():
graph_config = {
"llm": {
"model": "ollama/llama3.1",
"temperature": 0,
"format": "json",
},
"verbose": True,
"headless": False
}
smart_scraper_graph = SearchLinkGraph(
source="https://sport.sky.it/nba?gr=www",
config=graph_config
)
result = smart_scraper_graph.run()
assert result is not None