mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-08-28 18:36:36 +00:00
c5edb45c10
* Add SerperDev * Add release note * PR Feedback * Simplify, remove one-liner * Update haystack/preview/components/websearch/serper_dev.py Co-authored-by: ZanSara <sara.zanzottera@deepset.ai> * Update haystack/preview/components/websearch/serper_dev.py Co-authored-by: ZanSara <sara.zanzottera@deepset.ai> * Fix formatting * PR feedback * Fix tests * Function rename * Remove scoring, update tests * PR feedback * Fix return * small adjustments * fix tests * add e2e test * fix release notes * fix tests * fix e2e --------- Co-authored-by: ZanSara <sara.zanzottera@deepset.ai>
16 lines
616 B
Python
16 lines
616 B
Python
import os
|
|
import pytest
|
|
from haystack.preview import Document
|
|
from haystack.preview.components.websearch.serper_dev import SerperDevWebSearch
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
not os.environ.get("SERPERDEV_API_KEY", None),
|
|
reason="Export an env var called SERPERDEV_API_KEY containing the SerperDev API key to run this test.",
|
|
)
|
|
def test_web_search_top_k():
|
|
ws = SerperDevWebSearch(api_key=os.environ.get("SERPERDEV_API_KEY", None), top_k=10)
|
|
results = ws.run(query="Who is the boyfriend of Olivia Wilde?")["documents"]
|
|
assert len(results) == 10
|
|
assert all(isinstance(doc, Document) for doc in results)
|