|
3 | 3 | from __future__ import annotations |
4 | 4 |
|
5 | 5 | from typing import Any, Dict, List, Union, Iterable, Optional, cast |
| 6 | +from itertools import chain |
6 | 7 | from typing_extensions import Literal |
7 | 8 |
|
8 | 9 | import httpx |
@@ -81,6 +82,7 @@ def create( |
81 | 82 |
|
82 | 83 | timeout: Override the client-level default timeout for this request, in seconds |
83 | 84 | """ |
| 85 | + extra_headers = {"parallel-beta": "search-extract-2025-10-10", **(extra_headers or {})} |
84 | 86 | return self._post( |
85 | 87 | "/v1beta/tasks/groups", |
86 | 88 | body=maybe_transform({"metadata": metadata}, task_group_create_params.TaskGroupCreateParams), |
@@ -115,6 +117,7 @@ def retrieve( |
115 | 117 | """ |
116 | 118 | if not task_group_id: |
117 | 119 | raise ValueError(f"Expected a non-empty value for `task_group_id` but received {task_group_id!r}") |
| 120 | + extra_headers = {"parallel-beta": "search-extract-2025-10-10", **(extra_headers or {})} |
118 | 121 | return self._get( |
119 | 122 | f"/v1beta/tasks/groups/{task_group_id}", |
120 | 123 | options=make_request_options( |
@@ -163,9 +166,16 @@ def add_runs( |
163 | 166 | if not task_group_id: |
164 | 167 | raise ValueError(f"Expected a non-empty value for `task_group_id` but received {task_group_id!r}") |
165 | 168 | extra_headers = { |
166 | | - **strip_not_given({"parallel-beta": ",".join(str(e) for e in betas) if is_given(betas) else not_given}), |
| 169 | + **strip_not_given( |
| 170 | + { |
| 171 | + "parallel-beta": ",".join(chain((str(e) for e in betas), ["search-extract-2025-10-10"])) |
| 172 | + if is_given(betas) |
| 173 | + else not_given |
| 174 | + } |
| 175 | + ), |
167 | 176 | **(extra_headers or {}), |
168 | 177 | } |
| 178 | + extra_headers = {"parallel-beta": "search-extract-2025-10-10", **(extra_headers or {})} |
169 | 179 | return self._post( |
170 | 180 | f"/v1beta/tasks/groups/{task_group_id}/runs", |
171 | 181 | body=maybe_transform( |
@@ -212,6 +222,7 @@ def events( |
212 | 222 | if not task_group_id: |
213 | 223 | raise ValueError(f"Expected a non-empty value for `task_group_id` but received {task_group_id!r}") |
214 | 224 | extra_headers = {"Accept": "text/event-stream", **(extra_headers or {})} |
| 225 | + extra_headers = {"parallel-beta": "search-extract-2025-10-10", **(extra_headers or {})} |
215 | 226 | return self._get( |
216 | 227 | f"/v1beta/tasks/groups/{task_group_id}/events", |
217 | 228 | options=make_request_options( |
@@ -276,6 +287,7 @@ def get_runs( |
276 | 287 | if not task_group_id: |
277 | 288 | raise ValueError(f"Expected a non-empty value for `task_group_id` but received {task_group_id!r}") |
278 | 289 | extra_headers = {"Accept": "text/event-stream", **(extra_headers or {})} |
| 290 | + extra_headers = {"parallel-beta": "search-extract-2025-10-10", **(extra_headers or {})} |
279 | 291 | return self._get( |
280 | 292 | f"/v1beta/tasks/groups/{task_group_id}/runs", |
281 | 293 | options=make_request_options( |
@@ -346,6 +358,7 @@ async def create( |
346 | 358 |
|
347 | 359 | timeout: Override the client-level default timeout for this request, in seconds |
348 | 360 | """ |
| 361 | + extra_headers = {"parallel-beta": "search-extract-2025-10-10", **(extra_headers or {})} |
349 | 362 | return await self._post( |
350 | 363 | "/v1beta/tasks/groups", |
351 | 364 | body=await async_maybe_transform({"metadata": metadata}, task_group_create_params.TaskGroupCreateParams), |
@@ -380,6 +393,7 @@ async def retrieve( |
380 | 393 | """ |
381 | 394 | if not task_group_id: |
382 | 395 | raise ValueError(f"Expected a non-empty value for `task_group_id` but received {task_group_id!r}") |
| 396 | + extra_headers = {"parallel-beta": "search-extract-2025-10-10", **(extra_headers or {})} |
383 | 397 | return await self._get( |
384 | 398 | f"/v1beta/tasks/groups/{task_group_id}", |
385 | 399 | options=make_request_options( |
@@ -428,9 +442,16 @@ async def add_runs( |
428 | 442 | if not task_group_id: |
429 | 443 | raise ValueError(f"Expected a non-empty value for `task_group_id` but received {task_group_id!r}") |
430 | 444 | extra_headers = { |
431 | | - **strip_not_given({"parallel-beta": ",".join(str(e) for e in betas) if is_given(betas) else not_given}), |
| 445 | + **strip_not_given( |
| 446 | + { |
| 447 | + "parallel-beta": ",".join(chain((str(e) for e in betas), ["search-extract-2025-10-10"])) |
| 448 | + if is_given(betas) |
| 449 | + else not_given |
| 450 | + } |
| 451 | + ), |
432 | 452 | **(extra_headers or {}), |
433 | 453 | } |
| 454 | + extra_headers = {"parallel-beta": "search-extract-2025-10-10", **(extra_headers or {})} |
434 | 455 | return await self._post( |
435 | 456 | f"/v1beta/tasks/groups/{task_group_id}/runs", |
436 | 457 | body=await async_maybe_transform( |
@@ -477,6 +498,7 @@ async def events( |
477 | 498 | if not task_group_id: |
478 | 499 | raise ValueError(f"Expected a non-empty value for `task_group_id` but received {task_group_id!r}") |
479 | 500 | extra_headers = {"Accept": "text/event-stream", **(extra_headers or {})} |
| 501 | + extra_headers = {"parallel-beta": "search-extract-2025-10-10", **(extra_headers or {})} |
480 | 502 | return await self._get( |
481 | 503 | f"/v1beta/tasks/groups/{task_group_id}/events", |
482 | 504 | options=make_request_options( |
@@ -541,6 +563,7 @@ async def get_runs( |
541 | 563 | if not task_group_id: |
542 | 564 | raise ValueError(f"Expected a non-empty value for `task_group_id` but received {task_group_id!r}") |
543 | 565 | extra_headers = {"Accept": "text/event-stream", **(extra_headers or {})} |
| 566 | + extra_headers = {"parallel-beta": "search-extract-2025-10-10", **(extra_headers or {})} |
544 | 567 | return await self._get( |
545 | 568 | f"/v1beta/tasks/groups/{task_group_id}/runs", |
546 | 569 | options=make_request_options( |
|
0 commit comments