perplexityai 0.7.1__py3-none-any.whl → 0.8.0__py3-none-any.whl

perplexity/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "perplexity"
-__version__ = "0.7.1"  # x-release-please-version
+__version__ = "0.8.0"  # x-release-please-version

perplexity/resources/async_/chat/completions.py

@@ -2,10 +2,12 @@
 
 from __future__ import annotations
 
+from typing import Optional
+
 import httpx
 
 from ...._types import NOT_GIVEN, Body, Query, Headers, NotGiven
-from ...._utils import maybe_transform, async_maybe_transform
+from ...._utils import maybe_transform, strip_not_given, async_maybe_transform
 from ...._compat import cached_property
 from ...._resource import SyncAPIResource, AsyncAPIResource
 from ...._response import (
@@ -15,7 +17,7 @@ from ...._response import (
     async_to_streamed_response_wrapper,
 )
 from ...._base_client import make_request_options
-from ....types.async_.chat import completion_list_params, completion_create_params
+from ....types.async_.chat import completion_get_params, completion_create_params
 from ....types.async_.chat.completion_get_response import CompletionGetResponse
 from ....types.async_.chat.completion_list_response import CompletionListResponse
 from ....types.async_.chat.completion_create_response import CompletionCreateResponse
@@ -47,6 +49,7 @@ class CompletionsResource(SyncAPIResource):
         self,
         *,
         request: completion_create_params.Request,
+        idempotency_key: Optional[str] | 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,
@@ -55,7 +58,10 @@ class CompletionsResource(SyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> CompletionCreateResponse:
         """
-        Creates an asynchronous chat completion job
+        FastAPI wrapper around async chat completions
+
+        This endpoint creates an asynchronous chat completion job and returns a job ID
+        that can be used to poll for results.
 
         Args:
           extra_headers: Send extra headers
@@ -68,7 +74,13 @@ class CompletionsResource(SyncAPIResource):
         """
         return self._post(
             "/async/chat/completions",
-            body=maybe_transform({"request": request}, completion_create_params.CompletionCreateParams),
+            body=maybe_transform(
+                {
+                    "request": request,
+                    "idempotency_key": idempotency_key,
+                },
+                completion_create_params.CompletionCreateParams,
+            ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -78,8 +90,6 @@ class CompletionsResource(SyncAPIResource):
     def list(
         self,
         *,
-        limit: int | NotGiven = NOT_GIVEN,
-        next_token: str | 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,
@@ -87,44 +97,25 @@ class CompletionsResource(SyncAPIResource):
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> CompletionListResponse:
-        """
-        Lists all asynchronous chat completion requests for the authenticated user
-
-        Args:
-          limit: Maximum number of requests to return
-
-          next_token: Token for fetching the next page of results
-
-          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
-        """
+        """list all async chat completion requests for a given user."""
         return self._get(
             "/async/chat/completions",
             options=make_request_options(
-                extra_headers=extra_headers,
-                extra_query=extra_query,
-                extra_body=extra_body,
-                timeout=timeout,
-                query=maybe_transform(
-                    {
-                        "limit": limit,
-                        "next_token": next_token,
-                    },
-                    completion_list_params.CompletionListParams,
-                ),
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
             cast_to=CompletionListResponse,
         )
 
     def get(
         self,
-        request_id: str,
+        api_request: str,
         *,
+        local_mode: bool | NotGiven = NOT_GIVEN,
+        x_client_env: str | NotGiven = NOT_GIVEN,
+        x_client_name: str | NotGiven = NOT_GIVEN,
+        x_request_time: str | NotGiven = NOT_GIVEN,
+        x_usage_tier: str | NotGiven = NOT_GIVEN,
+        x_user_id: str | 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,
@@ -133,7 +124,7 @@ class CompletionsResource(SyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> CompletionGetResponse:
         """
-        Retrieves the status and result of a specific asynchronous chat completion job
+        get the response for a given async chat completion request.
 
         Args:
           extra_headers: Send extra headers
@@ -144,12 +135,28 @@ class CompletionsResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        if not request_id:
-            raise ValueError(f"Expected a non-empty value for `request_id` but received {request_id!r}")
+        if not api_request:
+            raise ValueError(f"Expected a non-empty value for `api_request` but received {api_request!r}")
+        extra_headers = {
+            **strip_not_given(
+                {
+                    "x-client-env": x_client_env,
+                    "x-client-name": x_client_name,
+                    "x-request-time": x_request_time,
+                    "x-usage-tier": x_usage_tier,
+                    "x-user-id": x_user_id,
+                }
+            ),
+            **(extra_headers or {}),
+        }
         return self._get(
-            f"/async/chat/completions/{request_id}",
+            f"/async/chat/completions/{api_request}",
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform({"local_mode": local_mode}, completion_get_params.CompletionGetParams),
             ),
             cast_to=CompletionGetResponse,
         )
@@ -179,6 +186,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
         self,
         *,
         request: completion_create_params.Request,
+        idempotency_key: Optional[str] | 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,
@@ -187,7 +195,10 @@ class AsyncCompletionsResource(AsyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> CompletionCreateResponse:
         """
-        Creates an asynchronous chat completion job
+        FastAPI wrapper around async chat completions
+
+        This endpoint creates an asynchronous chat completion job and returns a job ID
+        that can be used to poll for results.
 
         Args:
           extra_headers: Send extra headers
@@ -200,7 +211,13 @@ class AsyncCompletionsResource(AsyncAPIResource):
         """
         return await self._post(
             "/async/chat/completions",
-            body=await async_maybe_transform({"request": request}, completion_create_params.CompletionCreateParams),
+            body=await async_maybe_transform(
+                {
+                    "request": request,
+                    "idempotency_key": idempotency_key,
+                },
+                completion_create_params.CompletionCreateParams,
+            ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -210,8 +227,6 @@ class AsyncCompletionsResource(AsyncAPIResource):
     async def list(
         self,
         *,
-        limit: int | NotGiven = NOT_GIVEN,
-        next_token: str | 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,
@@ -219,44 +234,25 @@ class AsyncCompletionsResource(AsyncAPIResource):
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> CompletionListResponse:
-        """
-        Lists all asynchronous chat completion requests for the authenticated user
-
-        Args:
-          limit: Maximum number of requests to return
-
-          next_token: Token for fetching the next page of results
-
-          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
-        """
+        """list all async chat completion requests for a given user."""
         return await self._get(
             "/async/chat/completions",
             options=make_request_options(
-                extra_headers=extra_headers,
-                extra_query=extra_query,
-                extra_body=extra_body,
-                timeout=timeout,
-                query=await async_maybe_transform(
-                    {
-                        "limit": limit,
-                        "next_token": next_token,
-                    },
-                    completion_list_params.CompletionListParams,
-                ),
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
             cast_to=CompletionListResponse,
         )
 
     async def get(
         self,
-        request_id: str,
+        api_request: str,
         *,
+        local_mode: bool | NotGiven = NOT_GIVEN,
+        x_client_env: str | NotGiven = NOT_GIVEN,
+        x_client_name: str | NotGiven = NOT_GIVEN,
+        x_request_time: str | NotGiven = NOT_GIVEN,
+        x_usage_tier: str | NotGiven = NOT_GIVEN,
+        x_user_id: str | 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,
@@ -265,7 +261,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> CompletionGetResponse:
         """
-        Retrieves the status and result of a specific asynchronous chat completion job
+        get the response for a given async chat completion request.
 
         Args:
           extra_headers: Send extra headers
@@ -276,12 +272,30 @@ class AsyncCompletionsResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        if not request_id:
-            raise ValueError(f"Expected a non-empty value for `request_id` but received {request_id!r}")
+        if not api_request:
+            raise ValueError(f"Expected a non-empty value for `api_request` but received {api_request!r}")
+        extra_headers = {
+            **strip_not_given(
+                {
+                    "x-client-env": x_client_env,
+                    "x-client-name": x_client_name,
+                    "x-request-time": x_request_time,
+                    "x-usage-tier": x_usage_tier,
+                    "x-user-id": x_user_id,
+                }
+            ),
+            **(extra_headers or {}),
+        }
         return await self._get(
-            f"/async/chat/completions/{request_id}",
+            f"/async/chat/completions/{api_request}",
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=await async_maybe_transform(
+                    {"local_mode": local_mode}, completion_get_params.CompletionGetParams
+                ),
             ),
             cast_to=CompletionGetResponse,
         )

perplexity/resources/chat/completions.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Iterable, Optional
+from typing import Dict, Union, Iterable, Optional
 from typing_extensions import Literal
 
 import httpx
@@ -19,8 +19,8 @@ from ..._response import (
 )
 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
+from ...types.shared_params.chat_message_input import ChatMessageInput
 
 __all__ = ["CompletionsResource", "AsyncCompletionsResource"]
 
@@ -48,20 +48,60 @@ class CompletionsResource(SyncAPIResource):
     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,
+        messages: Iterable[ChatMessageInput],
+        model: str,
+        _debug_pro_search: bool | NotGiven = NOT_GIVEN,
+        _inputs: Optional[Iterable[int]] | NotGiven = NOT_GIVEN,
+        _is_browser_agent: Optional[bool] | NotGiven = NOT_GIVEN,
+        _prompt_token_length: Optional[int] | NotGiven = NOT_GIVEN,
+        best_of: Optional[int] | NotGiven = NOT_GIVEN,
+        country: Optional[str] | NotGiven = NOT_GIVEN,
+        cum_logprobs: Optional[bool] | NotGiven = NOT_GIVEN,
+        debug_params: Optional[completion_create_params.DebugParams] | NotGiven = NOT_GIVEN,
+        disable_search: Optional[bool] | NotGiven = NOT_GIVEN,
+        diverse_first_token: Optional[bool] | NotGiven = NOT_GIVEN,
+        enable_search_classifier: Optional[bool] | NotGiven = NOT_GIVEN,
+        file_workspace_id: Optional[str] | NotGiven = NOT_GIVEN,
+        frequency_penalty: Optional[float] | NotGiven = NOT_GIVEN,
+        has_image_url: bool | NotGiven = NOT_GIVEN,
+        image_domain_filter: Optional[SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
+        image_format_filter: Optional[SequenceNotStr[str]] | 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,
+        latitude: Optional[float] | NotGiven = NOT_GIVEN,
+        logprobs: Optional[bool] | NotGiven = NOT_GIVEN,
+        longitude: Optional[float] | NotGiven = NOT_GIVEN,
+        max_tokens: Optional[int] | NotGiven = NOT_GIVEN,
+        n: Optional[int] | NotGiven = NOT_GIVEN,
+        num_images: int | NotGiven = NOT_GIVEN,
+        num_search_results: int | NotGiven = NOT_GIVEN,
+        parallel_tool_calls: Optional[bool] | NotGiven = NOT_GIVEN,
+        presence_penalty: Optional[float] | NotGiven = NOT_GIVEN,
+        ranking_model: Optional[str] | NotGiven = NOT_GIVEN,
+        reasoning_effort: Optional[Literal["minimal", "low", "medium", "high"]] | NotGiven = NOT_GIVEN,
+        response_format: Optional[completion_create_params.ResponseFormat] | NotGiven = NOT_GIVEN,
+        response_metadata: Optional[Dict[str, object]] | NotGiven = NOT_GIVEN,
+        return_images: Optional[bool] | NotGiven = NOT_GIVEN,
+        return_related_questions: Optional[bool] | NotGiven = NOT_GIVEN,
+        return_videos: Optional[bool] | 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,
         search_domain_filter: Optional[SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
+        search_internal_properties: Optional[Dict[str, object]] | 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,
+        search_tenant: Optional[str] | NotGiven = NOT_GIVEN,
+        stop: Union[str, SequenceNotStr[str], None] | NotGiven = NOT_GIVEN,
+        stream: Optional[bool] | NotGiven = NOT_GIVEN,
+        temperature: Optional[float] | NotGiven = NOT_GIVEN,
+        tool_choice: Optional[Literal["none", "auto", "required"]] | NotGiven = NOT_GIVEN,
+        tools: Optional[Iterable[completion_create_params.Tool]] | NotGiven = NOT_GIVEN,
+        top_k: Optional[int] | NotGiven = NOT_GIVEN,
+        top_logprobs: Optional[int] | NotGiven = NOT_GIVEN,
+        top_p: Optional[float] | NotGiven = NOT_GIVEN,
+        updated_after_timestamp: Optional[int] | NotGiven = NOT_GIVEN,
+        updated_before_timestamp: Optional[int] | 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.
@@ -71,40 +111,9 @@ class CompletionsResource(SyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> CompletionCreateResponse:
         """
-        Generates a model's response for the given chat conversation with integrated
-        search capabilities
+        FastAPI wrapper around chat completions
 
         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
@@ -119,18 +128,58 @@ class CompletionsResource(SyncAPIResource):
                 {
                     "messages": messages,
                     "model": model,
+                    "_debug_pro_search": _debug_pro_search,
+                    "_inputs": _inputs,
+                    "_is_browser_agent": _is_browser_agent,
+                    "_prompt_token_length": _prompt_token_length,
+                    "best_of": best_of,
+                    "country": country,
+                    "cum_logprobs": cum_logprobs,
+                    "debug_params": debug_params,
                     "disable_search": disable_search,
+                    "diverse_first_token": diverse_first_token,
                     "enable_search_classifier": enable_search_classifier,
+                    "file_workspace_id": file_workspace_id,
+                    "frequency_penalty": frequency_penalty,
+                    "has_image_url": has_image_url,
+                    "image_domain_filter": image_domain_filter,
+                    "image_format_filter": image_format_filter,
                     "last_updated_after_filter": last_updated_after_filter,
                     "last_updated_before_filter": last_updated_before_filter,
+                    "latitude": latitude,
+                    "logprobs": logprobs,
+                    "longitude": longitude,
+                    "max_tokens": max_tokens,
+                    "n": n,
+                    "num_images": num_images,
+                    "num_search_results": num_search_results,
+                    "parallel_tool_calls": parallel_tool_calls,
+                    "presence_penalty": presence_penalty,
+                    "ranking_model": ranking_model,
                     "reasoning_effort": reasoning_effort,
+                    "response_format": response_format,
+                    "response_metadata": response_metadata,
                     "return_images": return_images,
                     "return_related_questions": return_related_questions,
+                    "return_videos": return_videos,
+                    "safe_search": safe_search,
                     "search_after_date_filter": search_after_date_filter,
                     "search_before_date_filter": search_before_date_filter,
                     "search_domain_filter": search_domain_filter,
+                    "search_internal_properties": search_internal_properties,
                     "search_mode": search_mode,
                     "search_recency_filter": search_recency_filter,
+                    "search_tenant": search_tenant,
+                    "stop": stop,
+                    "stream": stream,
+                    "temperature": temperature,
+                    "tool_choice": tool_choice,
+                    "tools": tools,
+                    "top_k": top_k,
+                    "top_logprobs": top_logprobs,
+                    "top_p": top_p,
+                    "updated_after_timestamp": updated_after_timestamp,
+                    "updated_before_timestamp": updated_before_timestamp,
                     "web_search_options": web_search_options,
                 },
                 completion_create_params.CompletionCreateParams,
@@ -165,20 +214,60 @@ class AsyncCompletionsResource(AsyncAPIResource):
     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,
+        messages: Iterable[ChatMessageInput],
+        model: str,
+        _debug_pro_search: bool | NotGiven = NOT_GIVEN,
+        _inputs: Optional[Iterable[int]] | NotGiven = NOT_GIVEN,
+        _is_browser_agent: Optional[bool] | NotGiven = NOT_GIVEN,
+        _prompt_token_length: Optional[int] | NotGiven = NOT_GIVEN,
+        best_of: Optional[int] | NotGiven = NOT_GIVEN,
+        country: Optional[str] | NotGiven = NOT_GIVEN,
+        cum_logprobs: Optional[bool] | NotGiven = NOT_GIVEN,
+        debug_params: Optional[completion_create_params.DebugParams] | NotGiven = NOT_GIVEN,
+        disable_search: Optional[bool] | NotGiven = NOT_GIVEN,
+        diverse_first_token: Optional[bool] | NotGiven = NOT_GIVEN,
+        enable_search_classifier: Optional[bool] | NotGiven = NOT_GIVEN,
+        file_workspace_id: Optional[str] | NotGiven = NOT_GIVEN,
+        frequency_penalty: Optional[float] | NotGiven = NOT_GIVEN,
+        has_image_url: bool | NotGiven = NOT_GIVEN,
+        image_domain_filter: Optional[SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
+        image_format_filter: Optional[SequenceNotStr[str]] | 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,
+        latitude: Optional[float] | NotGiven = NOT_GIVEN,
+        logprobs: Optional[bool] | NotGiven = NOT_GIVEN,
+        longitude: Optional[float] | NotGiven = NOT_GIVEN,
+        max_tokens: Optional[int] | NotGiven = NOT_GIVEN,
+        n: Optional[int] | NotGiven = NOT_GIVEN,
+        num_images: int | NotGiven = NOT_GIVEN,
+        num_search_results: int | NotGiven = NOT_GIVEN,
+        parallel_tool_calls: Optional[bool] | NotGiven = NOT_GIVEN,
+        presence_penalty: Optional[float] | NotGiven = NOT_GIVEN,
+        ranking_model: Optional[str] | NotGiven = NOT_GIVEN,
+        reasoning_effort: Optional[Literal["minimal", "low", "medium", "high"]] | NotGiven = NOT_GIVEN,
+        response_format: Optional[completion_create_params.ResponseFormat] | NotGiven = NOT_GIVEN,
+        response_metadata: Optional[Dict[str, object]] | NotGiven = NOT_GIVEN,
+        return_images: Optional[bool] | NotGiven = NOT_GIVEN,
+        return_related_questions: Optional[bool] | NotGiven = NOT_GIVEN,
+        return_videos: Optional[bool] | 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,
         search_domain_filter: Optional[SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
+        search_internal_properties: Optional[Dict[str, object]] | 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,
+        search_tenant: Optional[str] | NotGiven = NOT_GIVEN,
+        stop: Union[str, SequenceNotStr[str], None] | NotGiven = NOT_GIVEN,
+        stream: Optional[bool] | NotGiven = NOT_GIVEN,
+        temperature: Optional[float] | NotGiven = NOT_GIVEN,
+        tool_choice: Optional[Literal["none", "auto", "required"]] | NotGiven = NOT_GIVEN,
+        tools: Optional[Iterable[completion_create_params.Tool]] | NotGiven = NOT_GIVEN,
+        top_k: Optional[int] | NotGiven = NOT_GIVEN,
+        top_logprobs: Optional[int] | NotGiven = NOT_GIVEN,
+        top_p: Optional[float] | NotGiven = NOT_GIVEN,
+        updated_after_timestamp: Optional[int] | NotGiven = NOT_GIVEN,
+        updated_before_timestamp: Optional[int] | 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.
@@ -188,40 +277,9 @@ class AsyncCompletionsResource(AsyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> CompletionCreateResponse:
         """
-        Generates a model's response for the given chat conversation with integrated
-        search capabilities
+        FastAPI wrapper around chat completions
 
         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
@@ -236,18 +294,58 @@ class AsyncCompletionsResource(AsyncAPIResource):
                 {
                     "messages": messages,
                     "model": model,
+                    "_debug_pro_search": _debug_pro_search,
+                    "_inputs": _inputs,
+                    "_is_browser_agent": _is_browser_agent,
+                    "_prompt_token_length": _prompt_token_length,
+                    "best_of": best_of,
+                    "country": country,
+                    "cum_logprobs": cum_logprobs,
+                    "debug_params": debug_params,
                     "disable_search": disable_search,
+                    "diverse_first_token": diverse_first_token,
                     "enable_search_classifier": enable_search_classifier,
+                    "file_workspace_id": file_workspace_id,
+                    "frequency_penalty": frequency_penalty,
+                    "has_image_url": has_image_url,
+                    "image_domain_filter": image_domain_filter,
+                    "image_format_filter": image_format_filter,
                     "last_updated_after_filter": last_updated_after_filter,
                     "last_updated_before_filter": last_updated_before_filter,
+                    "latitude": latitude,
+                    "logprobs": logprobs,
+                    "longitude": longitude,
+                    "max_tokens": max_tokens,
+                    "n": n,
+                    "num_images": num_images,
+                    "num_search_results": num_search_results,
+                    "parallel_tool_calls": parallel_tool_calls,
+                    "presence_penalty": presence_penalty,
+                    "ranking_model": ranking_model,
                     "reasoning_effort": reasoning_effort,
+                    "response_format": response_format,
+                    "response_metadata": response_metadata,
                     "return_images": return_images,
                     "return_related_questions": return_related_questions,
+                    "return_videos": return_videos,
+                    "safe_search": safe_search,
                     "search_after_date_filter": search_after_date_filter,
                     "search_before_date_filter": search_before_date_filter,
                     "search_domain_filter": search_domain_filter,
+                    "search_internal_properties": search_internal_properties,
                     "search_mode": search_mode,
                     "search_recency_filter": search_recency_filter,
+                    "search_tenant": search_tenant,
+                    "stop": stop,
+                    "stream": stream,
+                    "temperature": temperature,
+                    "tool_choice": tool_choice,
+                    "tools": tools,
+                    "top_k": top_k,
+                    "top_logprobs": top_logprobs,
+                    "top_p": top_p,
+                    "updated_after_timestamp": updated_after_timestamp,
+                    "updated_before_timestamp": updated_before_timestamp,
                     "web_search_options": web_search_options,
                 },
                 completion_create_params.CompletionCreateParams,