unique_web_search 2026.24.0.dev2__tar.gz → 2026.24.0.dev3__tar.gz

{unique_web_search-2026.24.0.dev2 → unique_web_search-2026.24.0.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-web-search
-Version: 2026.24.0.dev2
+Version: 2026.24.0.dev3
 Summary: 
 Author: Andreas Hauri, Gustav Hartz, Rami Azouz
 Author-email: Andreas Hauri <andreas@unique.ch>, Gustav Hartz <gustav.hartz.ext@unique.ch>, Rami Azouz <rami.ext@unique.ch>
@@ -27,6 +27,7 @@ Requires-Dist: httpx>=0.28.1
 Requires-Dist: certifi>=2025.11.12
 Requires-Dist: jinja2>=3.1.6
 Requires-Dist: playwright>=1.58.0
+Requires-Dist: perplexityai>=0.34.0
 Requires-Python: >=3.12, <4.0
 Description-Content-Type: text/markdown
 

{unique_web_search-2026.24.0.dev2 → unique_web_search-2026.24.0.dev3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_web_search"
-version = "2026.24.0.dev2"
+version = "2026.24.0.dev3"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12,<4.0"
@@ -33,6 +33,7 @@ dependencies = [
     "certifi>=2025.11.12",
     "jinja2>=3.1.6",
     "playwright>=1.58.0",
+    "perplexityai>=0.34.0",
 ]
 
 [build-system]

{unique_web_search-2026.24.0.dev2 → unique_web_search-2026.24.0.dev3}/src/unique_web_search/client_settings.py

@@ -200,3 +200,38 @@ def get_brave_search_settings() -> BraveSearchSettings:
     if _brave_search_settings is None:
         _brave_search_settings = BraveSearchSettings.from_env_settings()
     return _brave_search_settings
+
+
+class PerplexitySearchSettings(BaseModel):
+    api_key: str | None = None
+
+    @property
+    def is_configured(self) -> bool:
+        return self.api_key is not None
+
+    @classmethod
+    def from_env_settings(cls):
+        missing_settings = []
+
+        if env_settings.perplexity_api_key is None:
+            missing_settings.append("API Key")
+
+        if missing_settings:
+            _LOGGER.warning(
+                "Perplexity Search API missing required settings: "
+                f"{', '.join(missing_settings)}"
+            )
+        else:
+            _LOGGER.info("Perplexity Search API is properly configured")
+
+        return cls(api_key=env_settings.perplexity_api_key)
+
+
+_perplexity_search_settings: PerplexitySearchSettings | None = None
+
+
+def get_perplexity_search_settings() -> PerplexitySearchSettings:
+    global _perplexity_search_settings
+    if _perplexity_search_settings is None:
+        _perplexity_search_settings = PerplexitySearchSettings.from_env_settings()
+    return _perplexity_search_settings

{unique_web_search-2026.24.0.dev2 → unique_web_search-2026.24.0.dev3}/src/unique_web_search/services/search_engine/__init__.py

@@ -35,6 +35,10 @@ from unique_web_search.services.search_engine.jina import (
     JinaConfig,
     JinaSearch,
 )
+from unique_web_search.services.search_engine.perplexity import (
+    PerplexitySearch,
+    PerplexitySearchConfig,
+)
 from unique_web_search.services.search_engine.tavily import (
     TavilyConfig,
     TavilySearch,
@@ -51,6 +55,7 @@ SearchEngineTypes = (
     | JinaSearch
     | TavilySearch
     | BraveSearch
+    | PerplexitySearch
     | FireCrawlSearch
     | VertexAI
 )
@@ -61,6 +66,7 @@ SearchEngineConfigTypes = (
     | JinaConfig
     | TavilyConfig
     | BraveSearchConfig
+    | PerplexitySearchConfig
     | FireCrawlConfig
     | VertexAIConfig
 )
@@ -71,6 +77,7 @@ ENGINE_NAME_TO_CONFIG = {
     "tavily": TavilyConfig,
     "bing": BingSearchConfig,
     "brave": BraveSearchConfig,
+    "perplexity": PerplexitySearchConfig,
     "firecrawl": FireCrawlConfig,
     "vertexai": VertexAIConfig,
     "custom_api": CustomAPIConfig,
@@ -94,6 +101,8 @@ def get_search_engine_service(
             return BingSearch(search_engine_config, language_model_service)
         case SearchEngineType.BRAVE:
             return BraveSearch(search_engine_config)
+        case SearchEngineType.PERPLEXITY:
+            return PerplexitySearch(search_engine_config)
         case SearchEngineType.VERTEXAI:
             return VertexAI(search_engine_config, language_model_service)
         case SearchEngineType.CUSTOM_API:
@@ -140,6 +149,8 @@ __all__ = [
     "BingSearch",
     "BraveSearch",
     "BraveSearchConfig",
+    "PerplexitySearch",
+    "PerplexitySearchConfig",
     "get_search_engine_service",
     "BaseSearchEngineConfig",
     "SearchEngine",

{unique_web_search-2026.24.0.dev2 → unique_web_search-2026.24.0.dev3}/src/unique_web_search/services/search_engine/base.py

@@ -21,6 +21,7 @@ class SearchEngineType(StrEnum):
     FIRECRAWL = "Firecrawl"
     TAVILY = "Tavily"
     BRAVE = "Brave"
+    PERPLEXITY = "Perplexity"
     BING = "Bing"
     DUCKDUCKGO = "DuckDuckGo"
     VERTEXAI = "VertexAI"
@@ -33,6 +34,7 @@ _SEARCH_ENGINE_MODE_MAP: dict[SearchEngineType, SearchEngineMode] = {
     SearchEngineType.FIRECRAWL: SearchEngineMode.STANDARD,
     SearchEngineType.TAVILY: SearchEngineMode.STANDARD,
     SearchEngineType.BRAVE: SearchEngineMode.STANDARD,
+    SearchEngineType.PERPLEXITY: SearchEngineMode.STANDARD,
     SearchEngineType.DUCKDUCKGO: SearchEngineMode.STANDARD,
     SearchEngineType.BING: SearchEngineMode.AGENT,
     SearchEngineType.VERTEXAI: SearchEngineMode.AGENT,
@@ -62,6 +64,7 @@ _SearchEngineExposedName = {
     SearchEngineType.FIRECRAWL: "Firecrawl Search",
     SearchEngineType.TAVILY: "Tavily Search",
     SearchEngineType.BRAVE: "Brave Search",
+    SearchEngineType.PERPLEXITY: "Perplexity Search",
     SearchEngineType.BING: "Grounding with Bing",
     SearchEngineType.DUCKDUCKGO: "DuckDuckGo Search",
     SearchEngineType.VERTEXAI: "Grounding with VertexAI",

unique_web_search-2026.24.0.dev3/src/unique_web_search/services/search_engine/perplexity.py

@@ -0,0 +1,121 @@
+import logging
+from typing import Annotated, Literal, TypeVar
+
+from perplexity import AsyncPerplexity, Omit
+from perplexity.types.search_create_response import Result
+from pydantic import Field
+from unique_toolkit._common.pydantic_helpers import DeactivatedNone
+
+from unique_web_search.client_settings import get_perplexity_search_settings
+from unique_web_search.services.client.proxy_config import async_client
+from unique_web_search.services.search_engine import (
+    BaseSearchEngineConfig,
+    SearchEngine,
+    SearchEngineType,
+)
+from unique_web_search.services.search_engine.base import get_search_engine_model_config
+from unique_web_search.services.search_engine.schema import WebSearchResult
+
+_LOGGER = logging.getLogger(__name__)
+
+# Perplexity Search API allows 1–20 results per request.
+MAX_RESULTS_PER_REQUEST = 20
+
+
+class PerplexitySearchConfig(BaseSearchEngineConfig[SearchEngineType.PERPLEXITY]):
+    model_config = get_search_engine_model_config(SearchEngineType.PERPLEXITY)
+    search_engine_name: Literal[SearchEngineType.PERPLEXITY] = (
+        SearchEngineType.PERPLEXITY
+    )
+    country: Annotated[str, Field(title="Active")] | DeactivatedNone = Field(
+        default=None,
+        description="The country to use for the search. It's a two-letter country code.",
+    )
+    max_tokens: Annotated[int, Field(title="Active")] | DeactivatedNone = Field(
+        default=None,
+        description="The maximum number of tokens to use for the context. It's a positive integer.",
+    )
+    search_mode: (
+        Annotated[Literal["web", "academic", "sec"], Field(title="Active")]
+        | DeactivatedNone
+    ) = Field(
+        default=None,
+        description="The search mode to use for the search. It's a string.",
+    )
+    search_recency_filter: (
+        Annotated[
+            Literal["hour", "day", "week", "month", "year"], Field(title="Active")
+        ]
+        | DeactivatedNone
+    ) = Field(
+        default=None,
+        description="The recency filter to use for the search. It's a string.",
+    )
+    search_type: (
+        Annotated[Literal["web", "people"], Field(title="Active")] | DeactivatedNone
+    ) = Field(
+        default=None,
+        description="The search type to use for the search. It's a string.",
+    )
+
+    requires_scraping: bool = False
+
+
+class PerplexitySearch(SearchEngine[PerplexitySearchConfig]):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.is_configured = get_perplexity_search_settings().is_configured
+
+    @property
+    def requires_scraping(self) -> bool:
+        return self.config.requires_scraping
+
+    async def search(self, query: str, **kwargs) -> list[WebSearchResult]:
+        settings = get_perplexity_search_settings()
+        assert settings.is_configured
+
+        max_results = min(self.config.fetch_size, MAX_RESULTS_PER_REQUEST)
+
+        async with async_client() as http_client:
+            perplexity_client = AsyncPerplexity(
+                api_key=settings.api_key, http_client=http_client
+            )
+            response = await perplexity_client.search.create(
+                query=query,
+                max_results=max_results,
+                country=_return_omit_if_none(self.config.country),
+                max_tokens=_return_omit_if_none(self.config.max_tokens),
+                search_mode=_return_omit_if_none(self.config.search_mode),
+                search_recency_filter=_return_omit_if_none(
+                    self.config.search_recency_filter
+                ),
+                search_type=_return_omit_if_none(self.config.search_type),
+            )
+
+        return self._to_web_search_results(response.results)
+
+    def _to_web_search_results(
+        self, results: list[Result] | None
+    ) -> list[WebSearchResult]:
+        if not results:
+            _LOGGER.warning("No search results found in Perplexity search response")
+            return []
+
+        return [
+            WebSearchResult(
+                url=result.url,
+                title=result.title,
+                snippet=result.snippet or "No Snippet Found",
+            )
+            for result in results
+        ]
+
+
+T = TypeVar("T")
+
+
+def _return_omit_if_none(value: T | None) -> T | Omit:
+    if value is None:
+        return Omit()
+    else:
+        return value

{unique_web_search-2026.24.0.dev2 → unique_web_search-2026.24.0.dev3}/src/unique_web_search/settings.py

@@ -60,6 +60,9 @@ class Base(BaseSettings):
     brave_search_api_key: str | None = None
     brave_search_api_endpoint: str | None = None
 
+    # Perplexity API settings
+    perplexity_api_key: str | None = None
+
     # VertexAI API settings
     vertexai_service_account_credentials: str | None = None
     vertexai_service_account_scopes: list[str] | None = None