0jmddlmZddlZddlmZmZmZmZmZddl Z ddl m Z ddl m Z mZddlmZmZmZmZmZmZmZddlmZmZdd lmZdd lmZmZdd lm Z m!Z!m"Z"m#Z#dd l$m%Z%dd l&m'Z'ddl(m)Z)ddl*m+Z+m,Z,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z4m5Z6m7Z7ddl8m9Z9ddgZ:GddeZ;GddeZ<GddZ=GddZ>GddZ?GddZ@dS) ) annotationsN)DictTypeUnionOptionaloverload)prepare_timeout_float)task_run_create_paramstask_run_result_params)BodyOmitQueryHeadersNotGivenomit not_given)maybe_transformasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_raw_response_wrapperto_streamed_response_wrapperasync_to_raw_response_wrapper"async_to_streamed_response_wrapper)make_request_options)TaskRun) TaskRunResult)OutputT OutputSchema TaskSpecParam)build_task_spec_param)ParsedTaskRunResult)wait_for_resultwait_for_result_asynctask_run_result_parser) SourcePolicyTaskRunResourceAsyncTaskRunResourcec eZdZdZed6dZed7dZeeeedddedd8dZ dddedd9d!Z eddded"d:d&Z edddd'd;d-Z e eeddded.dd5ZdS)?r)The Task API executes web research and extraction tasks. Clients submit a natural-language objective with an optional input schema; the service plans retrieval, fetches relevant URLs, and returns outputs that conform to a provided or inferred JSON schema. Supports deep research style queries and can return rich structured JSON outputs. Processors trade-off between cost, latency, and quality. Each processor supports calibrated confidences. - Output metadata: citations, excerpts, reasoning, and confidence per field returnTaskRunResourceWithRawResponsec 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 )r.selfs c/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/parallel/resources/task_run.pywith_raw_responsez!TaskRunResource.with_raw_response.s.d333$TaskRunResourceWithStreamingResponsec 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 )r6r1s r3with_streaming_responsez'TaskRunResource.with_streaming_response8s4D999r5Nmetadataprevious_interaction_id source_policy task_spec extra_headers extra_query extra_bodytimeoutinputUnion[str, Dict[str, object]] processorstrr;3Optional[Dict[str, Union[str, float, bool]]] | Omitr<Optional[str] | Omitr=Optional[SourcePolicy] | Omitr>Optional[TaskSpecParam] | Omitr?Headers | Noner@ Query | NonerA Body | NonerB'float | httpx.Timeout | None | NotGivenrc |dt||||||dtjt ||| | t S)( Initiates a task run. Returns immediately with a run object in status 'queued'. Beta features can be enabled by setting the 'parallel-beta' header. Args: input: Input to the task, either text or a JSON object. processor: Processor to use for the task. metadata: User-provided metadata stored with the run. Keys and values must be strings with a maximum length of 16 and 512 characters respectively. previous_interaction_id: Interaction ID to use as context for this request. source_policy: Source policy for web search results. This policy governs which sources are allowed/disallowed in results. task_spec: Specification for a task. Auto output schemas can be specified by setting `output_schema={"type":"auto"}`. Not specifying a TaskSpec is the same as setting an auto output schema. For convenience bare strings are also accepted as input or output schemas. 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 /v1/tasks/runsrCrEr;r<r=r>r?r@rArBbodyoptionscast_to)_postrr TaskRunCreateParamsrr r2rCrEr;r<r=r>r?r@rArBs r3createzTaskRunResource.createAsxhzz  "!* (/F%2!* ':  )+Q[el!   r5rSrun_idc |std||d|t||||tS) Retrieves run status by run_id. The run result is available from the `/result` endpoint. 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 5Expected a non-empty value for `run_id` but received /v1/tasks/runs/rSrVrW ValueError_getrrr2r\r?r@rArBs r3retrievezTaskRunResource.retrievesm2 a_U[__`` `yy &f & &(+Q[el    r5 api_timeoutr?r@rArBrh int | Omitrc |std||d|dt||||td|itjt S)z Retrieves a run result by run_id, blocking until the run is completed. 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_r`/resultrhr?r@rArBqueryra)rcrdrrr TaskRunResultParamsrr2r\rhr?r@rArBs r3resultzTaskRunResource.results0 a_U[__`` `yy -f - - -(+'%%}k&BDZDnoo "   r5outputr?r@rAdeadlinefloatrs-Optional[OutputSchema] | Type[OutputT] | Omit,TaskRunResult | ParsedTaskRunResult[OutputT]cDd fd }t|||S) 9Wait for a task run to complete within the given timeout.r\rFrtrur-rwc|tjz }||}t|SNrStime monotonicrqr' r\rtrBtask_run_resultrAr?r@rsr2s r3_fetcherz2TaskRunResource._wait_for_result.._fetchersN!1!11G"kk+'% *O*/6BB Br5r\rtcallabler\rFrtrur-rw)_wait_for_resultr2r\rtrsr?r@rArs` ```` r3rz TaskRunResource._wait_for_resultsZ C C C C C C C C C C v8TTTTr5r;rsr?r@rArBOptional[OutputSchema] | OmitcdSN r2rCrEr;rsr?r@rArBs r3executezTaskRunResource.executes r5r;r?r@rArB Type[OutputT]ParsedTaskRunResult[OutputT]cdSrrrs r3rzTaskRunResource.executes (+sr5c ddi|pi}t|}tj|z} ||||t ||||||} || j| ||||S)? Convenience method to create and execute a task run in a single call. Awaits run completion. If the run is successful, a `ParsedTaskRunResult` is returned when a pydantic was specified in `output`. Otherwise, a `TaskRunResult` is returned. Possible errors: - `TimeoutError`: If the run does not finish within the specified timeout. - `APIStatusError`: If the API returns a non-200-range status code. - `APIConnectionError`: If the connection to the API fails. Args: input: Input to the task, either text or a JSON object. processor: Processor to use for the task. metadata: User-provided metadata stored with the run. Keys and values must be strings with a maximum length of 16 and 512 characters respectively. output: Optional output schema or pydantic type. If pydantic is provided, the response will have a parsed field. 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. If the result is not available within the timeout, a `TimeoutError` is raised. X-Stainless-Poll-HelpertruerCrEr;r>r?r@rArBr\rtrsr?r@rAr r}r~r[r#rr\ r2rCrEr;rsr?r@rArBrttask_runs r3rzTaskRunResource.execute s\3FT}?RPRT '00>##g-;;+FE::'#!   $$?'#! %   r5)r-r.)r-r6rCrDrErFr;rGr<rHr=rIr>rJr?rKr@rLrArMrBrNr-r r\rFr?rKr@rLrArMrBrNr-rr\rFrhrir?rKr@rLrArMrBrNr-rr\rFrtrursrvr?rKr@rLrArMr-rwrCrDrErFr;rGrsrr?rKr@rLrArMrBrNr-rrCrDrErFr;rGrsrr?rKr@rLrArMrBrNr-rrCrDrErFr;rGrsrvr?rKr@rLrArMrBrNr-rw__name__ __module__ __qualname____doc__rr4r9rrr[rfrqrrrrr5r3r)r)'s 444_4:::_:IM8<7;48)-$("&;DE E E E E E Z)-$("&;D! ! ! ! ! ! N#')-$("&;D$ $ $ $ $ $ VAE)-$("&UUUUUU6 IM04)-$("&;D     X  IM)-$("&;D + + + + +X +&IM@D)-$("&;DE E E E E E E E r5c eZdZdZed6dZed7dZeeeedddedd8dZ dddedd9d!Z eddded"d:d&Z edddd'd;d-Z e eeddded.dd5ZdS)?r*r,r-#AsyncTaskRunResourceWithRawResponsec t|Sr0)rr1s r3r4z&AsyncTaskRunResource.with_raw_responseYs34888r5)AsyncTaskRunResourceWithStreamingResponsec t|Sr8)rr1s r3r9z,AsyncTaskRunResource.with_streaming_responsecs9>>>r5Nr:rCrDrErFr;rGr<rHr=rIr>rJr?rKr@rLrArMrBrNrc K|dt||||||dtjd{Vt ||| | t d{VS)rPrQrRNrSrT)rXrr rYrrrZs r3r[zAsyncTaskRunResource.createlshZZ ,"!* (/F%2!* ':        )+Q[el!          r5rSr\c K|std||d|t||||td{VS)r^r_r`rSraNrbres r3rfzAsyncTaskRunResource.retrieves2 a_U[__`` `YY &f & &(+Q[el          r5rgrhrirc K|std||d|dt||||td|itjd{Vt d{VS)rkr_r`rlrhNrmra)rcrdrrr rorrps r3rqzAsyncTaskRunResource.results0 a_U[__`` `YY -f - - -(+'%1"K02H2\ "         r5rrrtrursrvrwcTKd fd }t|||d {VS) ryr\rFrtrur-rwcK|tjz }||d{V}t|Sr{r|rs r3rz7AsyncTaskRunResource._wait_for_result.._fetcher sp!1!11G$(KK+'% %0%%O*/6BB Br5rNr)_wait_for_result_asyncrs` ```` r3rz%AsyncTaskRunResource._wait_for_resultsq C C C C C C C C C C,6HW_``````````r5rrc KdSrrrs r3rzAsyncTaskRunResource.executes r5rrrc KdSrrrs r3rzAsyncTaskRunResource.execute&s (+sr5c Kddi|pi}t|}tj|z} ||||t ||||||d{V} || j| ||||d{VS)rrrrNrrrs r3rzAsyncTaskRunResource.execute3s\3FT}?RPRT '00>##g-+FE::'#!%         **?'#! +         r5)r-r)r-rrrrrrrrrrr5r3r*r*Rs 999_9???_?IM8<7;48)-$("&;DE E E E E E Z)-$("&;D! ! ! ! ! ! N#')-$("&;D& & & & & & ZAE)-$("&aaaaaa6 IM04(,$("&;D     X  IM(,$("&;D + + + + +X +"IM@D)-$("&;DE E E E E E E E r5ceZdZddZdS)r.rr)r-Nonec||_t|j|_t|j|_t|j|_dSr) _task_runrr[rfrqr2rs r3__init__z'TaskRunResourceWithRawResponse.__init__|sT!- O   0     . O   r5Nrr)r-rrrrrrr5r3r.r.{(       r5r.ceZdZddZdS)rrr*r-rc||_t|j|_t|j|_t|j|_dSr)rrr[rfrqrs r3rz,AsyncTaskRunResourceWithRawResponse.__init__sT!3 O   6     4 O   r5Nrr*r-rrrr5r3rrrr5rceZdZddZdS)r6rr)r-rc||_t|j|_t|j|_t|j|_dSr)rrr[rfrqrs r3rz-TaskRunResourceWithStreamingResponse.__init__sT!2 O   5     3 O   r5Nrrrr5r3r6r6rr5r6ceZdZddZdS)rrr*r-rc||_t|j|_t|j|_t|j|_dSr)rrr[rfrqrs r3rz2AsyncTaskRunResourceWithStreamingResponse.__init__sT!8 O   ;     9 O   r5Nrrrr5r3rrrr5r)A __future__rr}typingrrrrrhttpxparallel.lib._timer typesr r _typesr rrrrrr_utilsrr_compatr _resourcerr _responserrrr _base_clientrtypes.task_runrtypes.task_run_resultrtypes.task_spec_paramr r!r"lib._parsing._task_specr#types.parsed_task_run_resultr$lib._parsing._task_run_resultr%rr&rr'!types.shared_params.source_policyr(__all__r)r*r.rr6rrr5r3rs#""""" 88888888888888 444444BBBBBBBBJJJJJJJJJJJJJJJJJJ;;;;;;;;%%%%%%99999999 0/////$$$$$$111111HHHHHHHHHH;;;;;;>>>>>> =<<<<< 4 5h h h h h oh h h V f f f f f +f f f R                                      r5