Skip to content

Commit da0b744

Browse files
committed
add test
1 parent f42d47e commit da0b744

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed

tests/graphs/search_link_ollama.py

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
1+
from scrapegraphai.graphs import SearchLinkGraph
2+
from scrapegraphai.utils import prettify_exec_info
3+
4+
def test_smart_scraper_pipeline():
5+
graph_config = {
6+
"llm": {
7+
"model": "ollama/llama3",
8+
"temperature": 0,
9+
"format": "json",
10+
},
11+
"embeddings": {
12+
"model": "ollama/nomic-embed-text",
13+
"temperature": 0,
14+
},
15+
"verbose": True,
16+
"headless": False
17+
}
18+
19+
smart_scraper_graph = SearchLinkGraph(
20+
source="https://sport.sky.it/nba?gr=www",
21+
config=graph_config
22+
)
23+
24+
result = smart_scraper_graph.run()
25+
26+
assert result is not None

0 commit comments

Comments
 (0)