mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-10-31 17:59:27 +00:00
f8e02310bf
* Remove BasePipeline and make a module for RayPipeline * Can load pipelines from yaml, plenty of issues left * Extract graph validation logic into _add_node_to_pipeline_graph & refactor load_from_config and add_node to use it * Fix pipeline tests * Move some tests out of test_pipeline.py and create MockDenseRetriever * myoy and pylint (silencing too-many-public-methods) * Fix issue found in some yaml files and in schema files * Fix paths to YAML and fix some typos in Ray * Fix eval tests * Simplify MockDenseRetriever * Fix Ray test * Accidentally pushed merge coinflict, fixed * Typo in schemas * Typo in _json_schema.py * Slightly reduce noisyness of version validation warnings * Fix version logs tests * Fix version logs tests again * remove seemingly unused file * Add check and test to avoid adding the same node to the pipeline twice * Update Documentation & Code Style * Revert config to pipeline_config * Remo0ve unused import * Complete reverting to pipeline_config * Some more stray config= * Update Documentation & Code Style * Feedback * Move back other_nodes tests into pipeline tests temporarily * Update Documentation & Code Style * Fixing tests * Update Documentation & Code Style * Fixing ray and standard pipeline tests * Rename colliding load() methods in dense retrievers and faiss * Update Documentation & Code Style * Fix mypy on ray.py as well * Add check for no root node * Fix tests to use load_from_directory and load_index * Try to workaround the disabled add_node of RayPipeline * Update Documentation & Code Style * Fix Ray test * Fix FAISS tests * Relax class check in _add_node_to_pipeline_graph * Update Documentation & Code Style * Try to fix mypy in ray.py * unused import * Try another fix for Ray * Fix connector tests * Update Documentation & Code Style * Fix ray * Update Documentation & Code Style * use BaseComponent.load() in pipelines/base.py * another round of feedback * stray BaseComponent.load() * Update Documentation & Code Style * Fix FAISS tests too Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com> Co-authored-by: tstadel <60758086+tstadel@users.noreply.github.com>
122 lines
4.4 KiB
Python
122 lines
4.4 KiB
Python
import json
|
|
from pathlib import Path
|
|
from re import search
|
|
|
|
import pytest
|
|
from haystack.nodes.connector import Crawler
|
|
from haystack.schema import Document
|
|
|
|
|
|
def test_crawler_url_none_exception(tmp_path):
|
|
tmp_dir = tmp_path / "crawled_files"
|
|
with pytest.raises(ValueError):
|
|
Crawler(tmp_dir).crawl()
|
|
|
|
|
|
def test_crawler_depth(tmp_path):
|
|
tmp_dir = tmp_path / "crawled_files"
|
|
_url = ["https://haystack.deepset.ai/overview/get-started"]
|
|
crawler = Crawler(output_dir=tmp_dir)
|
|
doc_path = crawler.crawl(urls=_url, crawler_depth=0)
|
|
assert len(doc_path) == 1
|
|
|
|
_urls = [
|
|
"https://haystack.deepset.ai/overview/v1.2.0/get-started",
|
|
"https://haystack.deepset.ai/overview/v1.1.0/get-started",
|
|
"https://haystack.deepset.ai/overview/v1.0.0/get-started",
|
|
]
|
|
doc_path = crawler.crawl(urls=_urls, crawler_depth=0)
|
|
assert len(doc_path) == 3
|
|
|
|
doc_path = crawler.crawl(urls=_url, crawler_depth=1)
|
|
assert len(doc_path) > 1
|
|
|
|
for json_file in doc_path:
|
|
assert isinstance(json_file, Path)
|
|
with open(json_file.absolute(), "r") as read_file:
|
|
data = json.load(read_file)
|
|
assert "content" in data
|
|
assert "meta" in data
|
|
assert isinstance(data["content"], str)
|
|
assert len(data["content"].split()) > 2
|
|
|
|
|
|
def test_crawler_filter_urls(tmp_path):
|
|
tmp_dir = tmp_path / "crawled_files"
|
|
_url = ["https://haystack.deepset.ai/overview/v1.2.0/"]
|
|
|
|
crawler = Crawler(output_dir=tmp_dir)
|
|
doc_path = crawler.crawl(urls=_url, filter_urls=["haystack\.deepset\.ai\/overview\/v1\.3\.0\/"])
|
|
assert len(doc_path) == 0
|
|
|
|
doc_path = crawler.crawl(urls=_url, filter_urls=["haystack\.deepset\.ai\/overview\/v1\.2\.0\/"])
|
|
assert len(doc_path) > 0
|
|
|
|
doc_path = crawler.crawl(urls=_url, filter_urls=["google\.com"])
|
|
assert len(doc_path) == 0
|
|
|
|
|
|
def test_crawler_content(tmp_path):
|
|
tmp_dir = tmp_path / "crawled_files"
|
|
|
|
partial_content_match: list = [
|
|
{
|
|
"url": "https://haystack.deepset.ai/overview/v1.1.0/intro",
|
|
"partial_content": [
|
|
"Haystack is an open-source framework ",
|
|
"for building search systems that work intelligently ",
|
|
"over large document collections.",
|
|
"Recent advances in NLP have enabled the application of ",
|
|
"question answering, retrieval and summarization ",
|
|
"to real world settings and Haystack is designed to be ",
|
|
"the bridge between research and industry.",
|
|
],
|
|
},
|
|
{
|
|
"url": "https://haystack.deepset.ai/overview/v1.1.0/use-cases",
|
|
"partial_content": [
|
|
"Expect to see results that highlight",
|
|
"the very sentence that contains the answer to your question.",
|
|
"Thanks to the power of Transformer based language models,",
|
|
"results are chosen based on compatibility in meaning",
|
|
"rather than lexical overlap.",
|
|
],
|
|
},
|
|
]
|
|
|
|
crawler = Crawler(output_dir=tmp_dir)
|
|
for _dict in partial_content_match:
|
|
url: str = _dict["url"]
|
|
partial_content: list = _dict["partial_content"]
|
|
|
|
doc_path = crawler.crawl(urls=[url], crawler_depth=0)
|
|
assert len(doc_path) == 1
|
|
|
|
for json_file in doc_path:
|
|
assert isinstance(json_file, Path)
|
|
with open(json_file.absolute(), "r") as read_file:
|
|
content = json.load(read_file)
|
|
assert isinstance(content["content"], str)
|
|
for partial_line in partial_content:
|
|
assert search(partial_line, content["content"])
|
|
assert partial_line in content["content"]
|
|
|
|
|
|
def test_crawler_return_document(tmp_path):
|
|
tmp_dir = tmp_path / "crawled_files"
|
|
_url = ["https://haystack.deepset.ai/docs/v1.0.0/intromd"]
|
|
|
|
crawler = Crawler(output_dir=tmp_dir)
|
|
docs_path = crawler.crawl(urls=_url, crawler_depth=1)
|
|
results, _ = crawler.run(urls=_url, crawler_depth=1, return_documents=True)
|
|
documents = results["documents"]
|
|
|
|
for json_file, document in zip(docs_path, documents):
|
|
assert isinstance(json_file, Path)
|
|
assert isinstance(document, Document)
|
|
|
|
with open(json_file.absolute(), "r") as read_file:
|
|
file_content = json.load(read_file)
|
|
assert file_content["meta"] == document.meta
|
|
assert file_content["content"] == document.content
|