perplexityai 0.5.0__py3-none-any.whl → 0.7.0__py3-none-any.whl

perplexity/resources/chat/completions.py

@@ -0,0 +1,295 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Iterable, Optional
+from typing_extensions import Literal
+
+import httpx
+
+from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven, SequenceNotStr
+from ..._utils import maybe_transform, async_maybe_transform
+from ..._compat import cached_property
+from ..._resource import SyncAPIResource, AsyncAPIResource
+from ..._response import (
+    to_raw_response_wrapper,
+    to_streamed_response_wrapper,
+    async_to_raw_response_wrapper,
+    async_to_streamed_response_wrapper,
+)
+from ...types.chat import completion_create_params
+from ..._base_client import make_request_options
+from ...types.shared_params.chat_message import ChatMessage
+from ...types.chat.completion_create_response import CompletionCreateResponse
+
+__all__ = ["CompletionsResource", "AsyncCompletionsResource"]
+
+
+class CompletionsResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> CompletionsResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/ppl-ai/perplexity-py#accessing-raw-response-data-eg-headers
+        """
+        return CompletionsResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> CompletionsResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/ppl-ai/perplexity-py#with_streaming_response
+        """
+        return CompletionsResourceWithStreamingResponse(self)
+
+    def create(
+        self,
+        *,
+        messages: Iterable[ChatMessage],
+        model: Literal["sonar", "sonar-pro", "sonar-deep-research", "sonar-reasoning", "sonar-reasoning-pro"],
+        disable_search: bool | NotGiven = NOT_GIVEN,
+        enable_search_classifier: bool | NotGiven = NOT_GIVEN,
+        last_updated_after_filter: Optional[str] | NotGiven = NOT_GIVEN,
+        last_updated_before_filter: Optional[str] | NotGiven = NOT_GIVEN,
+        reasoning_effort: Optional[Literal["low", "medium", "high"]] | NotGiven = NOT_GIVEN,
+        return_images: bool | NotGiven = NOT_GIVEN,
+        return_related_questions: bool | NotGiven = NOT_GIVEN,
+        search_after_date_filter: Optional[str] | NotGiven = NOT_GIVEN,
+        search_before_date_filter: Optional[str] | NotGiven = NOT_GIVEN,
+        search_domain_filter: Optional[SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
+        search_mode: Optional[Literal["web", "academic", "sec"]] | NotGiven = NOT_GIVEN,
+        search_recency_filter: Optional[Literal["hour", "day", "week", "month", "year"]] | NotGiven = NOT_GIVEN,
+        web_search_options: completion_create_params.WebSearchOptions | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> CompletionCreateResponse:
+        """
+        Generates a model's response for the given chat conversation with integrated
+        search capabilities
+
+        Args:
+          messages: A list of messages comprising the conversation so far
+
+          model: The name of the model that will complete your prompt
+
+          disable_search: Disables web search completely - model uses only training data
+
+          enable_search_classifier: Enables classifier that decides if web search is needed
+
+          last_updated_after_filter: Only include content last updated after this date (YYYY-MM-DD)
+
+          last_updated_before_filter: Only include content last updated before this date (YYYY-MM-DD)
+
+          reasoning_effort: Controls computational effort for sonar-deep-research model. Higher effort =
+              more thorough but more tokens
+
+          return_images: Whether to include images in search results
+
+          return_related_questions: Whether to return related questions
+
+          search_after_date_filter: Only include content published after this date (YYYY-MM-DD)
+
+          search_before_date_filter: Only include content published before this date (YYYY-MM-DD)
+
+          search_domain_filter: List of domains to limit search results to. Use '-' prefix to exclude domains
+
+          search_mode: Type of search: 'web' for general, 'academic' for scholarly, 'sec' for SEC
+              filings
+
+          search_recency_filter: Filter results by how recently they were published
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._post(
+            "/chat/completions",
+            body=maybe_transform(
+                {
+                    "messages": messages,
+                    "model": model,
+                    "disable_search": disable_search,
+                    "enable_search_classifier": enable_search_classifier,
+                    "last_updated_after_filter": last_updated_after_filter,
+                    "last_updated_before_filter": last_updated_before_filter,
+                    "reasoning_effort": reasoning_effort,
+                    "return_images": return_images,
+                    "return_related_questions": return_related_questions,
+                    "search_after_date_filter": search_after_date_filter,
+                    "search_before_date_filter": search_before_date_filter,
+                    "search_domain_filter": search_domain_filter,
+                    "search_mode": search_mode,
+                    "search_recency_filter": search_recency_filter,
+                    "web_search_options": web_search_options,
+                },
+                completion_create_params.CompletionCreateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=CompletionCreateResponse,
+        )
+
+
+class AsyncCompletionsResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncCompletionsResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/ppl-ai/perplexity-py#accessing-raw-response-data-eg-headers
+        """
+        return AsyncCompletionsResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncCompletionsResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/ppl-ai/perplexity-py#with_streaming_response
+        """
+        return AsyncCompletionsResourceWithStreamingResponse(self)
+
+    async def create(
+        self,
+        *,
+        messages: Iterable[ChatMessage],
+        model: Literal["sonar", "sonar-pro", "sonar-deep-research", "sonar-reasoning", "sonar-reasoning-pro"],
+        disable_search: bool | NotGiven = NOT_GIVEN,
+        enable_search_classifier: bool | NotGiven = NOT_GIVEN,
+        last_updated_after_filter: Optional[str] | NotGiven = NOT_GIVEN,
+        last_updated_before_filter: Optional[str] | NotGiven = NOT_GIVEN,
+        reasoning_effort: Optional[Literal["low", "medium", "high"]] | NotGiven = NOT_GIVEN,
+        return_images: bool | NotGiven = NOT_GIVEN,
+        return_related_questions: bool | NotGiven = NOT_GIVEN,
+        search_after_date_filter: Optional[str] | NotGiven = NOT_GIVEN,
+        search_before_date_filter: Optional[str] | NotGiven = NOT_GIVEN,
+        search_domain_filter: Optional[SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
+        search_mode: Optional[Literal["web", "academic", "sec"]] | NotGiven = NOT_GIVEN,
+        search_recency_filter: Optional[Literal["hour", "day", "week", "month", "year"]] | NotGiven = NOT_GIVEN,
+        web_search_options: completion_create_params.WebSearchOptions | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> CompletionCreateResponse:
+        """
+        Generates a model's response for the given chat conversation with integrated
+        search capabilities
+
+        Args:
+          messages: A list of messages comprising the conversation so far
+
+          model: The name of the model that will complete your prompt
+
+          disable_search: Disables web search completely - model uses only training data
+
+          enable_search_classifier: Enables classifier that decides if web search is needed
+
+          last_updated_after_filter: Only include content last updated after this date (YYYY-MM-DD)
+
+          last_updated_before_filter: Only include content last updated before this date (YYYY-MM-DD)
+
+          reasoning_effort: Controls computational effort for sonar-deep-research model. Higher effort =
+              more thorough but more tokens
+
+          return_images: Whether to include images in search results
+
+          return_related_questions: Whether to return related questions
+
+          search_after_date_filter: Only include content published after this date (YYYY-MM-DD)
+
+          search_before_date_filter: Only include content published before this date (YYYY-MM-DD)
+
+          search_domain_filter: List of domains to limit search results to. Use '-' prefix to exclude domains
+
+          search_mode: Type of search: 'web' for general, 'academic' for scholarly, 'sec' for SEC
+              filings
+
+          search_recency_filter: Filter results by how recently they were published
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return await self._post(
+            "/chat/completions",
+            body=await async_maybe_transform(
+                {
+                    "messages": messages,
+                    "model": model,
+                    "disable_search": disable_search,
+                    "enable_search_classifier": enable_search_classifier,
+                    "last_updated_after_filter": last_updated_after_filter,
+                    "last_updated_before_filter": last_updated_before_filter,
+                    "reasoning_effort": reasoning_effort,
+                    "return_images": return_images,
+                    "return_related_questions": return_related_questions,
+                    "search_after_date_filter": search_after_date_filter,
+                    "search_before_date_filter": search_before_date_filter,
+                    "search_domain_filter": search_domain_filter,
+                    "search_mode": search_mode,
+                    "search_recency_filter": search_recency_filter,
+                    "web_search_options": web_search_options,
+                },
+                completion_create_params.CompletionCreateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=CompletionCreateResponse,
+        )
+
+
+class CompletionsResourceWithRawResponse:
+    def __init__(self, completions: CompletionsResource) -> None:
+        self._completions = completions
+
+        self.create = to_raw_response_wrapper(
+            completions.create,
+        )
+
+
+class AsyncCompletionsResourceWithRawResponse:
+    def __init__(self, completions: AsyncCompletionsResource) -> None:
+        self._completions = completions
+
+        self.create = async_to_raw_response_wrapper(
+            completions.create,
+        )
+
+
+class CompletionsResourceWithStreamingResponse:
+    def __init__(self, completions: CompletionsResource) -> None:
+        self._completions = completions
+
+        self.create = to_streamed_response_wrapper(
+            completions.create,
+        )
+
+
+class AsyncCompletionsResourceWithStreamingResponse:
+    def __init__(self, completions: AsyncCompletionsResource) -> None:
+        self._completions = completions
+
+        self.create = async_to_streamed_response_wrapper(
+            completions.create,
+        )

perplexity/resources/content.py

@@ -0,0 +1,167 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+import httpx
+
+from ..types import content_create_params
+from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven, SequenceNotStr
+from .._utils import maybe_transform, async_maybe_transform
+from .._compat import cached_property
+from .._resource import SyncAPIResource, AsyncAPIResource
+from .._response import (
+    to_raw_response_wrapper,
+    to_streamed_response_wrapper,
+    async_to_raw_response_wrapper,
+    async_to_streamed_response_wrapper,
+)
+from .._base_client import make_request_options
+from ..types.content_create_response import ContentCreateResponse
+
+__all__ = ["ContentResource", "AsyncContentResource"]
+
+
+class ContentResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> ContentResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/ppl-ai/perplexity-py#accessing-raw-response-data-eg-headers
+        """
+        return ContentResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> ContentResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/ppl-ai/perplexity-py#with_streaming_response
+        """
+        return ContentResourceWithStreamingResponse(self)
+
+    def create(
+        self,
+        *,
+        urls: SequenceNotStr[str],
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> ContentCreateResponse:
+        """
+        Retrieve the full content from specified URLs
+
+        Args:
+          urls: List of URLs to retrieve content from
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._post(
+            "/content",
+            body=maybe_transform({"urls": urls}, content_create_params.ContentCreateParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=ContentCreateResponse,
+        )
+
+
+class AsyncContentResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncContentResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/ppl-ai/perplexity-py#accessing-raw-response-data-eg-headers
+        """
+        return AsyncContentResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncContentResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/ppl-ai/perplexity-py#with_streaming_response
+        """
+        return AsyncContentResourceWithStreamingResponse(self)
+
+    async def create(
+        self,
+        *,
+        urls: SequenceNotStr[str],
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> ContentCreateResponse:
+        """
+        Retrieve the full content from specified URLs
+
+        Args:
+          urls: List of URLs to retrieve content from
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return await self._post(
+            "/content",
+            body=await async_maybe_transform({"urls": urls}, content_create_params.ContentCreateParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=ContentCreateResponse,
+        )
+
+
+class ContentResourceWithRawResponse:
+    def __init__(self, content: ContentResource) -> None:
+        self._content = content
+
+        self.create = to_raw_response_wrapper(
+            content.create,
+        )
+
+
+class AsyncContentResourceWithRawResponse:
+    def __init__(self, content: AsyncContentResource) -> None:
+        self._content = content
+
+        self.create = async_to_raw_response_wrapper(
+            content.create,
+        )
+
+
+class ContentResourceWithStreamingResponse:
+    def __init__(self, content: ContentResource) -> None:
+        self._content = content
+
+        self.create = to_streamed_response_wrapper(
+            content.create,
+        )
+
+
+class AsyncContentResourceWithStreamingResponse:
+    def __init__(self, content: AsyncContentResource) -> None:
+        self._content = content
+
+        self.create = async_to_streamed_response_wrapper(
+            content.create,
+        )

perplexity/resources/search.py

@@ -53,6 +53,7 @@ class SearchResource(SyncAPIResource):
         last_updated_before_filter: Optional[str] | NotGiven = NOT_GIVEN,
         max_results: int | NotGiven = NOT_GIVEN,
         max_tokens: int | NotGiven = NOT_GIVEN,
+        max_tokens_per_page: int | NotGiven = NOT_GIVEN,
         safe_search: Optional[bool] | NotGiven = NOT_GIVEN,
         search_after_date_filter: Optional[str] | NotGiven = NOT_GIVEN,
         search_before_date_filter: Optional[str] | NotGiven = NOT_GIVEN,
@@ -67,9 +68,38 @@ class SearchResource(SyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> SearchCreateResponse:
         """
-        Search
+        Perform a search query with various filtering and customization options
 
         Args:
+          query: Search query string or array of query strings to search for
+
+          country: Country code to bias search results towards (e.g., 'US', 'GB', 'CA')
+
+          last_updated_after_filter: Only include results last updated after this date (ISO 8601 format: YYYY-MM-DD)
+
+          last_updated_before_filter: Only include results last updated before this date (ISO 8601 format: YYYY-MM-DD)
+
+          max_results: Maximum number of search results to return
+
+          max_tokens: Maximum number of tokens to return across all results
+
+          max_tokens_per_page: Maximum number of tokens to return per individual search result
+
+          safe_search: Enable safe search filtering to exclude adult content
+
+          search_after_date_filter: Only include results published after this date (ISO 8601 format: YYYY-MM-DD)
+
+          search_before_date_filter: Only include results published before this date (ISO 8601 format: YYYY-MM-DD)
+
+          search_domain_filter: List of domains to restrict search results to (e.g., ['example.com',
+              'another.com'])
+
+          search_mode: Type of search to perform: 'web' for general web search, 'academic' for
+              scholarly articles, 'sec' for SEC filings
+
+          search_recency_filter: Filter results by how recently they were published (hour, day, week, month, or
+              year)
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -88,6 +118,7 @@ class SearchResource(SyncAPIResource):
                     "last_updated_before_filter": last_updated_before_filter,
                     "max_results": max_results,
                     "max_tokens": max_tokens,
+                    "max_tokens_per_page": max_tokens_per_page,
                     "safe_search": safe_search,
                     "search_after_date_filter": search_after_date_filter,
                     "search_before_date_filter": search_before_date_filter,
@@ -133,6 +164,7 @@ class AsyncSearchResource(AsyncAPIResource):
         last_updated_before_filter: Optional[str] | NotGiven = NOT_GIVEN,
         max_results: int | NotGiven = NOT_GIVEN,
         max_tokens: int | NotGiven = NOT_GIVEN,
+        max_tokens_per_page: int | NotGiven = NOT_GIVEN,
         safe_search: Optional[bool] | NotGiven = NOT_GIVEN,
         search_after_date_filter: Optional[str] | NotGiven = NOT_GIVEN,
         search_before_date_filter: Optional[str] | NotGiven = NOT_GIVEN,
@@ -147,9 +179,38 @@ class AsyncSearchResource(AsyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> SearchCreateResponse:
         """
-        Search
+        Perform a search query with various filtering and customization options
 
         Args:
+          query: Search query string or array of query strings to search for
+
+          country: Country code to bias search results towards (e.g., 'US', 'GB', 'CA')
+
+          last_updated_after_filter: Only include results last updated after this date (ISO 8601 format: YYYY-MM-DD)
+
+          last_updated_before_filter: Only include results last updated before this date (ISO 8601 format: YYYY-MM-DD)
+
+          max_results: Maximum number of search results to return
+
+          max_tokens: Maximum number of tokens to return across all results
+
+          max_tokens_per_page: Maximum number of tokens to return per individual search result
+
+          safe_search: Enable safe search filtering to exclude adult content
+
+          search_after_date_filter: Only include results published after this date (ISO 8601 format: YYYY-MM-DD)
+
+          search_before_date_filter: Only include results published before this date (ISO 8601 format: YYYY-MM-DD)
+
+          search_domain_filter: List of domains to restrict search results to (e.g., ['example.com',
+              'another.com'])
+
+          search_mode: Type of search to perform: 'web' for general web search, 'academic' for
+              scholarly articles, 'sec' for SEC filings
+
+          search_recency_filter: Filter results by how recently they were published (hour, day, week, month, or
+              year)
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -168,6 +229,7 @@ class AsyncSearchResource(AsyncAPIResource):
                     "last_updated_before_filter": last_updated_before_filter,
                     "max_results": max_results,
                     "max_tokens": max_tokens,
+                    "max_tokens_per_page": max_tokens_per_page,
                     "safe_search": safe_search,
                     "search_after_date_filter": search_after_date_filter,
                     "search_before_date_filter": search_before_date_filter,

perplexity/types/__init__.py

@@ -2,5 +2,13 @@
 
 from __future__ import annotations
 
+from .shared import (
+    UsageInfo as UsageInfo,
+    ChatChoice as ChatChoice,
+    ChatMessage as ChatMessage,
+    SearchResult as SearchResult,
+)
 from .search_create_params import SearchCreateParams as SearchCreateParams
+from .content_create_params import ContentCreateParams as ContentCreateParams
 from .search_create_response import SearchCreateResponse as SearchCreateResponse
+from .content_create_response import ContentCreateResponse as ContentCreateResponse

perplexity/types/async_/__init__.py

@@ -0,0 +1,3 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations

perplexity/types/async_/chat/__init__.py

@@ -0,0 +1,9 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from .completion_list_params import CompletionListParams as CompletionListParams
+from .completion_get_response import CompletionGetResponse as CompletionGetResponse
+from .completion_create_params import CompletionCreateParams as CompletionCreateParams
+from .completion_list_response import CompletionListResponse as CompletionListResponse
+from .completion_create_response import CompletionCreateResponse as CompletionCreateResponse