c56bef871b
CodeQL / Analyze (python) (push) Has been cancelled
Update Platform Components Table / update (push) Has been cancelled
Docker image release / Build base image (push) Has been cancelled
Sync docs with Docusaurus / sync (push) Has been cancelled
Tests / Check if changed (push) Has been cancelled
Tests / format (push) Has been cancelled
Tests / check-imports (push) Has been cancelled
Tests / Unit / macos-latest (push) Has been cancelled
Tests / Unit / ubuntu-latest (push) Has been cancelled
Tests / Unit / windows-latest (push) Has been cancelled
Tests / mypy (push) Has been cancelled
Tests / Integration / ubuntu-latest (push) Has been cancelled
Tests / Integration / macos-latest (push) Has been cancelled
Tests / Integration / windows-latest (push) Has been cancelled
Tests / notify-slack-on-failure (push) Has been cancelled
Tests / Mark tests as completed (push) Has been cancelled
104 lines
3.6 KiB
Plaintext
104 lines
3.6 KiB
Plaintext
---
|
|
title: "BraveWebSearch"
|
|
id: bravewebsearch
|
|
slug: "/bravewebsearch"
|
|
description: "Search engine using the Brave Search API."
|
|
---
|
|
|
|
# BraveWebSearch
|
|
|
|
Search the web using the Brave Search API.
|
|
|
|
<div className="key-value-table">
|
|
|
|
| | |
|
|
| --- | --- |
|
|
| **Most common position in a pipeline** | Before a [`ChatPromptBuilder`](../builders/chatpromptbuilder.mdx) or right at the beginning of an indexing pipeline |
|
|
| **Mandatory init variables** | `api_key`: The Brave Search API key. Can be set with the `BRAVE_API_KEY` env var. |
|
|
| **Mandatory run variables** | `query`: A string with your search query. |
|
|
| **Output variables** | `documents`: A list of Haystack Documents containing search result content and metadata. <br /> <br />`links`: A list of strings of resulting URLs. |
|
|
| **API reference** | [Brave Search API](/reference/integrations-brave) |
|
|
| **GitHub link** | https://github.com/deepset-ai/haystack-core-integrations/blob/main/integrations/brave/src/haystack_integrations/components/websearch/brave/brave_websearch.py |
|
|
|
|
</div>
|
|
|
|
## Overview
|
|
|
|
When you give `BraveWebSearch` a query, it uses the [Brave Search API](https://brave.com/search/api/) to search the web and return relevant content as Haystack `Document` objects. It also returns a list of the source URLs.
|
|
|
|
Brave Search is an independent search engine with its own web index. It is a great fit for RAG pipelines that need reliable, privacy-focused web results without depending on Google or Bing.
|
|
|
|
`BraveWebSearch` requires a Brave Search API key to work. By default, it looks for a `BRAVE_API_KEY` environment variable. Alternatively, you can pass an `api_key` directly during initialization.
|
|
|
|
## Usage
|
|
|
|
### On its own
|
|
|
|
Here is a quick example of how `BraveWebSearch` searches the web based on a query and returns a list of Documents.
|
|
|
|
```python
|
|
from haystack_integrations.components.websearch.brave import BraveWebSearch
|
|
from haystack.utils import Secret
|
|
|
|
web_search = BraveWebSearch(
|
|
api_key=Secret.from_env_var("BRAVE_API_KEY"),
|
|
top_k=5,
|
|
)
|
|
query = "What is Haystack by deepset?"
|
|
|
|
response = web_search.run(query=query)
|
|
|
|
for doc in response["documents"]:
|
|
print(doc.content)
|
|
```
|
|
|
|
### In a pipeline
|
|
|
|
Here is an example of a Retrieval-Augmented Generation (RAG) pipeline that uses `BraveWebSearch` to look up an answer on the web.
|
|
|
|
```python
|
|
from haystack import Pipeline
|
|
from haystack.utils import Secret
|
|
from haystack.components.builders.chat_prompt_builder import ChatPromptBuilder
|
|
from haystack.components.generators.chat import OpenAIChatGenerator
|
|
from haystack_integrations.components.websearch.brave import BraveWebSearch
|
|
from haystack.dataclasses import ChatMessage
|
|
|
|
web_search = BraveWebSearch(
|
|
api_key=Secret.from_env_var("BRAVE_API_KEY"),
|
|
top_k=3,
|
|
)
|
|
|
|
prompt_template = [
|
|
ChatMessage.from_system("You are a helpful assistant."),
|
|
ChatMessage.from_user(
|
|
"Given the information below:\n"
|
|
"{% for document in documents %}{{ document.content }}\n{% endfor %}\n"
|
|
"Answer the following question: {{ query }}.\nAnswer:",
|
|
),
|
|
]
|
|
|
|
prompt_builder = ChatPromptBuilder(
|
|
template=prompt_template,
|
|
required_variables={"query", "documents"},
|
|
)
|
|
|
|
llm = OpenAIChatGenerator(
|
|
api_key=Secret.from_env_var("OPENAI_API_KEY"),
|
|
)
|
|
|
|
pipe = Pipeline()
|
|
pipe.add_component("search", web_search)
|
|
pipe.add_component("prompt_builder", prompt_builder)
|
|
pipe.add_component("llm", llm)
|
|
|
|
pipe.connect("search.documents", "prompt_builder.documents")
|
|
pipe.connect("prompt_builder.prompt", "llm.messages")
|
|
|
|
query = "What is Haystack by deepset?"
|
|
|
|
result = pipe.run(data={"search": {"query": query}, "prompt_builder": {"query": query}})
|
|
|
|
print(result["llm"]["replies"][0].text)
|
|
```
|