0j&qddlmZddlmZmZmZmZmZmZm Z ddl m Z ddl m Z ddlZddlmZmZmZmZmZmZmZddlmZmZmZmZdd lmZdd lmZm Z dd l!m"Z"m#Z#m$Z$m%Z%dd l&m'Z'm(Z(dd l)m*Z*m+Z+m,Z,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5ddl6m7Z7ddl8m9Z9ddl:m;Z;ddlGddeZ?Gdde Z@GddZAGddZBGddZCGd d!ZDdS)") annotations)AnyDictListUnionIterableOptionalcast)chain)LiteralN)BodyOmitQueryHeadersNotGivenomit not_given)is_givenmaybe_transformstrip_not_givenasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_raw_response_wrapperto_streamed_response_wrapperasync_to_raw_response_wrapper"async_to_streamed_response_wrapper)Stream AsyncStream)task_group_create_paramstask_group_events_paramstask_group_add_runs_paramstask_group_get_runs_params)make_request_options) TaskGroup) TaskSpecParam)ParallelBetaParam)BetaRunInputParam)TaskGroupRunResponse)TaskGroupEventsResponse)TaskGroupGetRunsResponseTaskGroupResourceAsyncTaskGroupResourcec eZdZdZed1dZed2dZedddedd3dZ dddedd4dZ eedddedd5d!Z eeddded"d6d(Z eeeeddded)d7d0Z dS)8r. The Task Group API is currently in beta and enables batch execution of many independent Task runs with group-level monitoring and failure handling. - Submit hundreds or thousands of Tasks as a single group - Observe group progress and receive results as they complete - Real-time updates via Server-Sent Events (SSE) - Add tasks to an existing group while it is running - Group-level retry and error aggregation Status: beta and subject to change. return TaskGroupResourceWithRawResponsec t|Sa 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/parallel-web/parallel-sdk-python#accessing-raw-response-data-eg-headers )r3selfs j/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/parallel/resources/beta/task_group.pywith_raw_responsez#TaskGroupResource.with_raw_response3s0555&TaskGroupResourceWithStreamingResponsec t|Sz An alternative to `.with_raw_response` that doesn't eagerly read the response body. For more information, see https://www.github.com/parallel-web/parallel-sdk-python#with_streaming_response )r;r6s r8with_streaming_responsez)TaskGroupResource.with_streaming_response=s6d;;;r:Nmetadata extra_headers extra_query extra_bodytimeoutr@3Optional[Dict[str, Union[str, float, bool]]] | OmitrAHeaders | NonerB Query | NonerC Body | NonerD'float | httpx.Timeout | None | NotGivenr'c ddi|pi}|dtd|itjt ||||t S) Initiates a TaskGroup to group and track multiple runs. Args: metadata: User-provided metadata stored with the task group. 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 parallel-betasearch-extract-2025-10-10/v1beta/tasks/groupsr@rArBrCrDbodyoptionscast_to)_postrr"TaskGroupCreateParamsr&r'r7r@rArBrCrDs r8createzTaskGroupResource.createFsp2)*E_-J][]_ zz " *h!79Q9ghh(+Q[el    r:rO task_group_idstrc |std|ddi|pi}|d|t||||tS)k Retrieves aggregated status across runs in a TaskGroup. Args: 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 z-TaskGroupResource.add_runs..(4K4KSVV4K4K4K4K4K4Kr:rMr]/runsrgrerOrP) r`rrjoinr rrTrr$TaskGroupAddRunsParamsr&r+ r7rXrgrerfrArBrCrDs r8add_runszTaskGroupResource.add_runssL om\immnn n  #&#SXXe4K4KU4K4K4KNiMj.k.k%l%l%l"  "  )*E_-J][]_ zz 8M 8 8 8 $):+A )+Q[el)   r: last_event_id api_timeoutrArBrCrDr}Optional[str] | Omitr~Optional[float] | OmitStream[TaskGroupEventsResponse]c0|std|ddi|pi}ddi|pi}|d|dt||||t||dtj t ttd tt S)  Streams events from a TaskGroup: status updates and run completions. The connection will remain open for up to an hour as long as at least one run in the group is still active. Args: 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 r\Accepttext/event-streamrLrMr]/eventsr}r~rArBrCrDqueryTrRrSstream stream_cls) r`rar&rr#TaskGroupEventsParamsr rr,r r7rXr}r~rArBrCrDs r8eventszTaskGroupResource.eventss8 om\immnn n!#6P=;NBP (*E_-J][]_ yy :M : : :(+'%%)6'2-B    ,56'   r: include_inputinclude_outputr}statusrArBrCrDr bool | OmitrrrOptional[Literal['queued', 'action_required', 'running', 'completed', 'failed', 'cancelling', 'cancelled']] | Omit Stream[TaskGroupGetRunsResponse]c4|std|ddi|pi}ddi|pi}|d|dt|||| t||||dtj t ttd tt S) f Retrieves task runs in a TaskGroup and optionally their inputs and outputs. All runs within a TaskGroup are returned as a stream. To get the inputs and/or outputs back in the stream, set the corresponding `include_input` and `include_output` parameters to `true`. The stream is resumable using the `event_id` as the cursor. To resume a stream, specify the `last_event_id` parameter with the `event_id` of the last event in the stream. The stream will resume from the next event after the `last_event_id`. Args: 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 r\rrrLrMr]rvrrr}rrTr) r`rar&rr%TaskGroupGetRunsParamsr rr-r r7rXrrr}rrArBrCrDs r8get_runszTaskGroupResource.get_runssN om\immnn n!#6P=;NBP (*E_-J][]_ yy 8M 8 8 8(+'%%)6*8)6"(  /E -67+   r:)r2r3)r2r; r@rErArFrBrGrCrHrDrIr2r' rXrYrArFrBrGrCrHrDrIr2r'rXrYrgrhrerirfrjrArFrBrGrCrHrDrIr2r+)rXrYr}rr~rrArFrBrGrCrHrDrIr2r)rXrYrrrrr}rrrrArFrBrGrCrHrDrIr2r__name__ __module__ __qualname____doc__rr9r>rrrWrcr{rrr:r8r.r.(st666_6<<<_<IM)-$("&;D! ! ! ! ! ! R)-$("&;D       N=A04)-$("&;D@ @ @ @ @ @ L/3.2)-$("&;D4 4 4 4 4 4 t&*&*.2)-$("&;D!A A A A A A A A r:c eZdZdZed1dZed2dZedddedd3dZ dddedd4dZ eedddedd5d!Z eeddded"d6d(Z eeeeddded)d7d0Z dS)8r/r1r2%AsyncTaskGroupResourceWithRawResponsec t|Sr5)rr6s r8r9z(AsyncTaskGroupResource.with_raw_responseRs5T:::r:+AsyncTaskGroupResourceWithStreamingResponsec t|Sr=)rr6s r8r>z.AsyncTaskGroupResource.with_streaming_response\s;4@@@r:Nr?r@rErArFrBrGrCrHrDrIr'c Kddi|pi}|dtd|itjd{Vt ||||t d{VS)rKrLrMrNr@NrOrP)rTrr"rUr&r'rVs r8rWzAsyncTaskGroupResource.createes2)*E_-J][]_ ZZ ",j(-CE]Estttttttt(+Q[el           r:rOrXrYc K|std|ddi|pi}|d|t||||td{VS)r[r\rLrMr]rOr^Nr_rbs r8rczAsyncTaskGroupResource.retrieves. om\immnn n(*E_-J][]_ YY 3M 3 3(+Q[el          r:rdrgrhrerirfrjr+c K|std|itdt|r.dt d|Ddgnt i|pi}ddi|pi}|d|dt||dtj d {Vt|||| t d {VS) rlr\rLrmc34K|]}t|VdSrorprqs r8rtz2AsyncTaskGroupResource.add_runs..rur:rMr]rvrwNrOrP) r`rrrxr rrTrr$ryr&r+rzs r8r{zAsyncTaskGroupResource.add_runss_L om\immnn n  #&#SXXe4K4KU4K4K4KNiMj.k.k%l%l%l"  "  )*E_-J][]_ ZZ 8M 8 8 8,$):+A )+Q[el)         r:r|r}rr~r$AsyncStream[TaskGroupEventsResponse]cLK|std|ddi|pi}ddi|pi}|d|dt||||t||dtjd {V t ttd tt d {VS) rr\rrrLrMr]rrNrTr) r`rar&rr#rr rr,r!rs r8rzAsyncTaskGroupResource.eventss8 om\immnn n!#6P=;NBP (*E_-J][]_ YY :M : : :(+'%1)6'2-B    ,"#:;'         r:rrrrrr%AsyncStream[TaskGroupGetRunsResponse]cPK|std|ddi|pi}ddi|pi}|d|dt|||| t||||dtjd {V t ttd tt d {VS) rr\rrrLrMr]rvrNrTr) r`rar&rr%rr rr-r!rs r8rzAsyncTaskGroupResource.get_runs"s&N om\immnn n!#6P=;NBP (*E_-J][]_ YY 8M 8 8 8(+'%1)6*8)6"(  /E -"#;<+         r:)r2r)r2rrrr)rXrYr}rr~rrArFrBrGrCrHrDrIr2r)rXrYrrrrr}rrrrArFrBrGrCrHrDrIr2rrrr:r8r/r/Gsx;;;_;AAA_AIM)-$("&;D! ! ! ! ! ! R)-$("&;D       N=A04)-$("&;D@ @ @ @ @ @ L/3.2)-$("&;D4 4 4 4 4 4 t&*&*.2)-$("&;D!A A A A A A A A r:ceZdZddZdS)r3 task_groupr.r2Nonec||_t|j|_t|j|_t|j|_t|j|_t|j|_dSro) _task_grouprrWrcr{rrr7rs r8__init__z)TaskGroupResourceWithRawResponse.__init__gs%-     0     0     .     0     r:Nrr.r2rrrrrrr:r8r3r3f(      r:r3ceZdZddZdS)rrr/r2rc||_t|j|_t|j|_t|j|_t|j|_t|j|_dSro)rrrWrcr{rrrs r8rz.AsyncTaskGroupResourceWithRawResponse.__init__|s%3     6     6     4     6     r:Nrr/r2rrrr:r8rr{rr:rceZdZddZdS)r;rr.r2rc||_t|j|_t|j|_t|j|_t|j|_t|j|_dSro)rrrWrcr{rrrs r8rz/TaskGroupResourceWithStreamingResponse.__init__s%2     5     5     3     5     r:Nrrrr:r8r;r;rr:r;ceZdZddZdS)rrr/r2rc||_t|j|_t|j|_t|j|_t|j|_t|j|_dSro)rrrWrcr{rrrs r8rz4AsyncTaskGroupResourceWithStreamingResponse.__init__s%8     ;     ;     9     ;     r:Nrrrr:r8rrrr:r)E __future__rtypingrrrrrr r itertoolsr typing_extensionsr httpx_typesrrrrrrr_utilsrrrr_compatr _resourcerr _responserrrr _streamingr r! types.betar"r#r$r% _base_clientr&types.beta.task_groupr'types.task_spec_paramr(types.beta.parallel_beta_paramr)types.beta.beta_run_input_paramr*"types.beta.task_group_run_responser+%types.beta.task_group_events_responser,'types.beta.task_group_get_runs_responser-__all__r.r/r3rr;rrr:r8rs@#"""""CCCCCCCCCCCCCCCCCC%%%%%% KKKKKKKKKKKKKKKKKKWWWWWWWWWWWW&&&&&&:::::::: .------- 100000......222222??????@@@@@@FFFFFFLLLLLLOOOOOO  8 9\ \ \ \ \ \ \ \ ~\ \ \ \ \ -\ \ \ ~        *        *        *          r: