diff --git a/docker/.env.example b/docker/.env.example index 36e492d3..87ca1727 100644 --- a/docker/.env.example +++ b/docker/.env.example @@ -390,6 +390,9 @@ GID='1000' #------ Exa Search ----------- https://www.exa.ai/ # AGENT_EXA_API_KEY= +#------ Perplexity Search ----------- [https://console.perplexity.ai](https://console.perplexity.ai) +# AGENT_PERPLEXITY_API_KEY= + ########################################### ######## Other Configurations ############ ########################################### diff --git a/frontend/src/pages/Admin/Agents/WebSearchSelection/SearchProviderOptions/index.jsx b/frontend/src/pages/Admin/Agents/WebSearchSelection/SearchProviderOptions/index.jsx index 3a431f7b..1d246bdc 100644 --- a/frontend/src/pages/Admin/Agents/WebSearchSelection/SearchProviderOptions/index.jsx +++ b/frontend/src/pages/Admin/Agents/WebSearchSelection/SearchProviderOptions/index.jsx @@ -384,3 +384,38 @@ export function ExaSearchOptions({ settings }) { ); } + +export function PerplexitySearchOptions({ settings }) { + return ( + <> +

+ You can get an API key{" "} + + from Perplexity. + +

+
+
+ + +
+
+ + ); +} diff --git a/frontend/src/pages/Admin/Agents/WebSearchSelection/icons/perplexity.png b/frontend/src/pages/Admin/Agents/WebSearchSelection/icons/perplexity.png new file mode 100644 index 00000000..f4767169 Binary files /dev/null and b/frontend/src/pages/Admin/Agents/WebSearchSelection/icons/perplexity.png differ diff --git a/frontend/src/pages/Admin/Agents/WebSearchSelection/index.jsx b/frontend/src/pages/Admin/Agents/WebSearchSelection/index.jsx index 3da6035e..ac373eb2 100644 --- a/frontend/src/pages/Admin/Agents/WebSearchSelection/index.jsx +++ b/frontend/src/pages/Admin/Agents/WebSearchSelection/index.jsx @@ -10,6 +10,7 @@ import SearXNGSearchIcon from "./icons/searxng.png"; import TavilySearchIcon from "./icons/tavily.svg"; import DuckDuckGoIcon from "./icons/duckduckgo.png"; import ExaIcon from "./icons/exa.png"; +import PerplexitySearchIcon from "./icons/perplexity.png"; import { CaretUpDown, MagnifyingGlass, @@ -29,6 +30,7 @@ import { TavilySearchOptions, DuckDuckGoOptions, ExaSearchOptions, + PerplexitySearchOptions, } from "./SearchProviderOptions"; const SEARCH_PROVIDERS = [ @@ -109,6 +111,13 @@ const SEARCH_PROVIDERS = [ options: (settings) => , description: "AI-powered search engine optimized for LLM use cases.", }, + { + name: "Perplexity Search", + value: "perplexity-search", + logo: PerplexitySearchIcon, + options: (settings) => , + description: "AI-powered web search using the Perplexity Search API.", + }, ]; export default function AgentWebSearchSelection({ diff --git a/server/.env.example b/server/.env.example index ccc8c87b..bd3d0eef 100644 --- a/server/.env.example +++ b/server/.env.example @@ -388,6 +388,9 @@ TTS_PROVIDER="native" #------ Exa Search ----------- https://www.exa.ai/ # AGENT_EXA_API_KEY= +#------ Perplexity Search ----------- [https://console.perplexity.ai](https://console.perplexity.ai) +# AGENT_PERPLEXITY_API_KEY= + ########################################### ######## Other Configurations ############ ########################################### diff --git a/server/models/systemSettings.js b/server/models/systemSettings.js index 26d44324..2cf4ec11 100644 --- a/server/models/systemSettings.js +++ b/server/models/systemSettings.js @@ -121,6 +121,7 @@ const SystemSettings = { "tavily-search", "duckduckgo-engine", "exa-search", + "perplexity-search", ].includes(update) ) throw new Error("Invalid SERP provider."); @@ -300,6 +301,7 @@ const SystemSettings = { AgentSearXNGApiUrl: process.env.AGENT_SEARXNG_API_URL || null, AgentTavilyApiKey: !!process.env.AGENT_TAVILY_API_KEY || null, AgentExaApiKey: !!process.env.AGENT_EXA_API_KEY || null, + AgentPerplexityApiKey: !!process.env.AGENT_PERPLEXITY_API_KEY || null, // -------------------------------------------------------- // Compliance Settings diff --git a/server/utils/agents/aibitat/plugins/web-browsing.js b/server/utils/agents/aibitat/plugins/web-browsing.js index 336cc72d..9638d17c 100644 --- a/server/utils/agents/aibitat/plugins/web-browsing.js +++ b/server/utils/agents/aibitat/plugins/web-browsing.js @@ -93,6 +93,9 @@ const webBrowsing = { case "exa-search": engine = "_exaSearch"; break; + case "perplexity-search": + engine = "_perplexitySearch"; + break; default: engine = "_duckDuckGoEngine"; } @@ -978,6 +981,84 @@ const webBrowsing = { ); return result; }, + + _perplexitySearch: async function (query) { + if (!process.env.AGENT_PERPLEXITY_API_KEY) { + this.super.introspect( + `${this.caller}: I can't use Perplexity searching because the user has not defined the required API key.\nVisit: [https://console.perplexity.ai](https://console.perplexity.ai) to create the API key.` + ); + return `Search is disabled and no content was found. This functionality is disabled because the user has not set it up yet.`; + } + + this.super.introspect( + `${this.caller}: Using Perplexity to search for "${ + query.length > 100 ? `${query.slice(0, 100)}...` : query + }"` + ); + + const { response, error } = await fetch( + "https://api.perplexity.ai/search", + { + method: "POST", + headers: { + "Content-Type": "application/json", + Authorization: `Bearer ${process.env.AGENT_PERPLEXITY_API_KEY}`, + }, + body: JSON.stringify({ + query: query, + max_results: 5, + max_tokens_per_page: 2048, + }), + } + ) + .then((res) => { + if (res.ok) return res.json(); + throw new Error( + `${res.status} - ${res.statusText}. params: ${JSON.stringify({ + auth: this.middleTruncate( + process.env.AGENT_PERPLEXITY_API_KEY, + 5 + ), + q: query, + })}` + ); + }) + .then((data) => { + return { response: data, error: null }; + }) + .catch((e) => { + this.super.handlerProps.log( + `Perplexity Search Error: ${e.message}` + ); + return { response: null, error: e.message }; + }); + + if (error) + return `There was an error searching for content. ${error}`; + + const data = []; + if (response.results) { + response.results.forEach((result) => { + data.push({ + title: result.title, + link: result.url, + snippet: result.snippet || "", + }); + }); + } + + if (data.length === 0) + return "No information was found online for the search query."; + + this.reportSearchResultsCitations(data); + + const result = JSON.stringify(data); + this.super.introspect( + `${this.caller}: I found ${data.length} results - reviewing the results now. (~${this.countTokens(result)} tokens)` + ); + + return result; + }, }); }, }; diff --git a/server/utils/helpers/updateENV.js b/server/utils/helpers/updateENV.js index cc08afbb..f422db14 100644 --- a/server/utils/helpers/updateENV.js +++ b/server/utils/helpers/updateENV.js @@ -602,6 +602,10 @@ const KEY_MAPPING = { envKey: "AGENT_EXA_API_KEY", checks: [], }, + AgentPerplexityApiKey: { + envKey: "AGENT_PERPLEXITY_API_KEY", + checks: [], + }, // TTS/STT Integration ENVS TextToSpeechProvider: {