haystack-brave-search 0.1.0__tar.gz

haystack_brave_search-0.1.0/PKG-INFO

@@ -0,0 +1,16 @@
+Metadata-Version: 2.4
+Name: haystack-brave-search
+Version: 0.1.0
+Summary: A Haystack component for web search via the Brave Search API
+Author: Tasis Michalakopoulos
+Author-email: Tasis Michalakopoulos <tasis@gaia-labs.eu>
+License-Expression: MIT
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Programming Language :: Python :: 3
+Requires-Dist: brave-search-python-client>=0.4.27
+Requires-Dist: haystack-ai>=2.25.2
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+

haystack_brave_search-0.1.0/pyproject.toml

@@ -0,0 +1,35 @@
+[project]
+name = "haystack-brave-search"
+version = "0.1.0"
+description = "A Haystack component for web search via the Brave Search API"
+readme = "README.md"
+authors = [
+    { name = "Tasis Michalakopoulos", email = "tasis@gaia-labs.eu" }
+]
+license = "MIT"
+requires-python = ">=3.12"
+dependencies = [
+    "brave-search-python-client>=0.4.27",
+    "haystack-ai>=2.25.2",
+]
+classifiers = [
+  "Development Status :: 3 - Alpha",
+  "Intended Audience :: Developers",
+  "Topic :: Scientific/Engineering :: Artificial Intelligence",
+  "Programming Language :: Python :: 3",
+]
+
+[project.scripts]
+haystack-brave-search = "haystack_brave_search:main"
+
+[build-system]
+requires = ["uv_build>=0.9.26,<0.10.0"]
+build-backend = "uv_build"
+
+[dependency-groups]
+dev = [
+    "psutil>=7.2.2",
+    "pytest>=9.0.2",
+    "python-dotenv>=1.2.2",
+    "ruff>=0.15.5",
+]

haystack_brave_search-0.1.0/src/haystack_brave_search/__init__.py

@@ -0,0 +1,3 @@
+from haystack_brave_search.brave_web_search import BraveWebSearch
+
+__all__ = ["BraveWebSearch"]

haystack_brave_search-0.1.0/src/haystack_brave_search/brave_web_search.py

@@ -0,0 +1,33 @@
+import asyncio
+from typing import List, Optional
+from haystack import component, default_from_dict, default_to_dict
+from haystack.dataclasses import Document
+from brave_search_python_client import BraveSearch, WebSearchRequest
+
+@component
+class BraveWebSearch:
+    def __init__(self, api_key: str, top_k: int = 5, country: Optional[str] = None):
+        self.api_key = api_key
+        self.top_k = top_k
+        self.country = country
+
+    @component.output_types(documents=List[Document])
+    def run(self, query: str):
+        return asyncio.run(self._async_run(query))
+
+    async def _async_run(self, query: str):
+        bs = BraveSearch(api_key=self.api_key)
+        req = WebSearchRequest(q=query, count=self.top_k, country=self.country)
+        response = await bs.web(req)
+        results = response.web.results if response.web else []
+        return {"documents": [
+            Document(content=r.description or "", meta={"title": r.title, "url": str(r.url)})
+            for r in results
+        ]}
+
+    def to_dict(self):
+        return default_to_dict(self, api_key=self.api_key, top_k=self.top_k, country=self.country)
+
+    @classmethod
+    def from_dict(cls, data):
+        return default_from_dict(cls, data)