gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

tests: fix test_add - add missing required parameter

Browse source
This commit is contained in:
Daulet Amirkhanov 2025-10-16 18:27:03 +01:00
parent 572c8ebce7
commit c0d450b165
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
cognee/tests/integration/web_url_crawler/test_add.py
View file

@ -18,11 +18,19 @@ async def test_add_succesfully_adds_url_when_preferred_loader_specified():
await cognee.prune.prune_data()
await cognee.prune.prune_system(metadata=True)
extraction_rules = {
"title": {"selector": "title", "attr": "text"},
"headings": {"selector": "h1, h2, h3", "attr": "text", "all": True},
"links": {"selector": "a", "attr": "href", "all": True},
"paragraphs": {"selector": "p", "attr": "text", "all": True},
}
loaders_config = {
"web_url_loader": {
"soup_config": {
"max_depth": 1,
"follow_links": False,
"extraction_rules": extraction_rules,
}
}
}