klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Loaders] Improve web page and sitemap loader usability (#961)

Browse Source
This commit is contained in:
Deshraj Yadav
2023-11-16 16:01:43 -08:00
committed by GitHub
parent 28460f725c
commit e0b73e6a5a
3 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/loaders/test_web_page.py
View File

@@ -27,7 +27,7 @@ def test_load_data(web_page_loader):
</body>
</html>
"""
with patch("embedchain.loaders.web_page.requests.get", return_value=mock_response):
with patch("embedchain.loaders.web_page.WebPageLoader._session.get", return_value=mock_response):
result = web_page_loader.load_data(page_url)
content = web_page_loader._get_clean_content(mock_response.content, page_url)