feat(api): add betas back in for search

stainless-app[bot] · stainless-app[bot] · commit 23493c6ae666 · 2026-03-09T21:49:30.000Z
diff --git a/.stats.yml b/.stats.yml
@@ -1,4 +1,4 @@
 configured_endpoints: 22
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/parallel-web%2Fparallel-sdk-492cb97b032ae6d87675f06806a66303d679c21e3b112cbeb36ef78063a4598c.yml
-openapi_spec_hash: dc8819b83a755680658c15fb30e30678
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/parallel-web%2Fparallel-sdk-970b780e86490322cc3c7e2b57f140ca6766a3d9f6e0d3402837ebaf7c2183fc.yml
+openapi_spec_hash: 34f784ce2dec796048e6780924bae08f
 config_hash: a398d153133d8884bed4e5256a0ae818
diff --git a/src/parallel/resources/beta/beta.py b/src/parallel/resources/beta/beta.py
@@ -192,6 +192,7 @@ def search(
         processor: Optional[Literal["base", "pro"]] | Omit = omit,
         search_queries: Optional[SequenceNotStr[str]] | Omit = omit,
         source_policy: Optional[SourcePolicy] | Omit = omit,
+        betas: List[ParallelBetaParam] | Omit = omit,
         # 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,
@@ -234,6 +235,8 @@ def search(
 
               This policy governs which sources are allowed/disallowed in results.
 
+          betas: Optional header to specify the beta version(s) to enable.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -242,6 +245,16 @@ def search(
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
+        extra_headers = {
+            **strip_not_given(
+                {
+                    "parallel-beta": ",".join(chain((str(e) for e in betas), ["search-extract-2025-10-10"]))
+                    if is_given(betas)
+                    else not_given
+                }
+            ),
+            **(extra_headers or {}),
+        }
         extra_headers = {"parallel-beta": "search-extract-2025-10-10", **(extra_headers or {})}
         return self._post(
             "/v1beta/search",
@@ -404,6 +417,7 @@ async def search(
         processor: Optional[Literal["base", "pro"]] | Omit = omit,
         search_queries: Optional[SequenceNotStr[str]] | Omit = omit,
         source_policy: Optional[SourcePolicy] | Omit = omit,
+        betas: List[ParallelBetaParam] | Omit = omit,
         # 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,
@@ -446,6 +460,8 @@ async def search(
 
               This policy governs which sources are allowed/disallowed in results.
 
+          betas: Optional header to specify the beta version(s) to enable.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -454,6 +470,16 @@ async def search(
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
+        extra_headers = {
+            **strip_not_given(
+                {
+                    "parallel-beta": ",".join(chain((str(e) for e in betas), ["search-extract-2025-10-10"]))
+                    if is_given(betas)
+                    else not_given
+                }
+            ),
+            **(extra_headers or {}),
+        }
         extra_headers = {"parallel-beta": "search-extract-2025-10-10", **(extra_headers or {})}
         return await self._post(
             "/v1beta/search",
diff --git a/src/parallel/types/beta/beta_search_params.py b/src/parallel/types/beta/beta_search_params.py
@@ -2,11 +2,13 @@
 
 from __future__ import annotations
 
-from typing import Optional
-from typing_extensions import Literal, TypedDict
+from typing import List, Optional
+from typing_extensions import Literal, Annotated, TypedDict
 
 from ..._types import SequenceNotStr
+from ..._utils import PropertyInfo
 from .fetch_policy_param import FetchPolicyParam
+from .parallel_beta_param import ParallelBetaParam
 from .excerpt_settings_param import ExcerptSettingsParam
 from ..shared_params.source_policy import SourcePolicy
 
@@ -59,3 +61,6 @@ class BetaSearchParams(TypedDict, total=False):
 
     This policy governs which sources are allowed/disallowed in results.
     """
+
+    betas: Annotated[List[ParallelBetaParam], PropertyInfo(alias="parallel-beta")]
+    """Optional header to specify the beta version(s) to enable."""
diff --git a/tests/api_resources/test_beta.py b/tests/api_resources/test_beta.py
@@ -97,6 +97,7 @@ def test_method_search_with_all_params(self, client: Parallel) -> None:
                 "exclude_domains": ["reddit.com", "x.com", ".ai"],
                 "include_domains": ["wikipedia.org", "usa.gov", ".edu"],
             },
+            betas=["mcp-server-2025-07-17"],
         )
         assert_matches_type(SearchResult, beta, path=["response"])
 
@@ -202,6 +203,7 @@ async def test_method_search_with_all_params(self, async_client: AsyncParallel)
                 "exclude_domains": ["reddit.com", "x.com", ".ai"],
                 "include_domains": ["wikipedia.org", "usa.gov", ".edu"],
             },
+            betas=["mcp-server-2025-07-17"],
         )
         assert_matches_type(SearchResult, beta, path=["response"])
 

Original file line number	Diff line number	Diff line change
`@@ -97,6 +97,7 @@ def test_method_search_with_all_params(self, client: Parallel) -> None:`
`97`	`97`	`"exclude_domains": ["reddit.com", "x.com", ".ai"],`
`98`	`98`	`"include_domains": ["wikipedia.org", "usa.gov", ".edu"],`
`99`	`99`	`},`
	`100`	`+ betas=["mcp-server-2025-07-17"],`
`100`	`101`	`)`
`101`	`102`	`assert_matches_type(SearchResult, beta, path=["response"])`
`102`	`103`
`@@ -202,6 +203,7 @@ async def test_method_search_with_all_params(self, async_client: AsyncParallel)`
`202`	`203`	`"exclude_domains": ["reddit.com", "x.com", ".ai"],`
`203`	`204`	`"include_domains": ["wikipedia.org", "usa.gov", ".edu"],`
`204`	`205`	`},`
	`206`	`+ betas=["mcp-server-2025-07-17"],`
`205`	`207`	`)`
`206`	`208`	`assert_matches_type(SearchResult, beta, path=["response"])`
`207`	`209`