U ZhB@sDddlmZddlZddlmZddlmZddlZddlmZddl m Z m Z m Z m Z mZddlmZmZdd lmZdd lmZmZdd lmZmZdd lmZmZdd lmZmZddl m!Z!m"Z"ddl#m$Z$ddl%m&Z&ddgZ'GdddeZ(GdddeZ)GdddZ*GdddZ+GdddZ,GdddZ-dS)) annotationsN)List)Literal)_legacy_response) NOT_GIVENBodyQueryHeadersNotGiven)maybe_transformasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_streamed_response_wrapper"async_to_streamed_response_wrapper)SyncCursorPageAsyncCursorPage)AsyncPaginatormake_request_options)step_list_paramsstep_retrieve_params)RunStep)RunStepIncludeSteps AsyncStepsc@seZdZeddddZeddddZeded d d ed d d d d d ddddd ddZ edeeeeed d d ed d d ddd ddd ddddd ddZ d S)rStepsWithRawResponsereturncCst|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/openai/openai-python#accessing-raw-response-data-eg-headers )rselfr#L/tmp/pip-unpacked-wheel-693nel7n/openai/resources/beta/threads/runs/steps.pywith_raw_responseszSteps.with_raw_responseStepsWithStreamingResponsecCst|Sz An alternative to `.with_raw_response` that doesn't eagerly read the response body. For more information, see https://www.github.com/openai/openai-python#with_streaming_response )r&r!r#r#r$with_streaming_response%szSteps.with_streaming_response>The Assistants API is deprecated in favor of the Responses APINinclude extra_headers extra_query extra_bodytimeoutstrList[RunStepInclude] | NotGivenHeaders | None Query | None Body | None'float | httpx.Timeout | None | NotGivenr step_id thread_idrun_idr+r,r-r.r/rc Cs|std||s$td||s6td|ddi|pBi}|jd|d|d|t||||td |itjd td S)  Retrieves a run step. Args: include: A list of additional fields to include in the response. Currently the only supported value is `step_details.tool_calls[*].file_search.results[*].content` to fetch the file search result content. See the [file search tool documentation](https://platform.openai.com/docs/assistants/tools/file-search#customizing-file-search-settings) for more information. 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 8Expected a non-empty value for `thread_id` but received 5Expected a non-empty value for `run_id` but received 6Expected a non-empty value for `step_id` but received OpenAI-Beta assistants=v2 /threads//runs//steps/r+r,r-r.r/queryoptionsZcast_to) ValueError_getrr rStepRetrieveParamsr r"r7r8r9r+r,r-r.r/r#r#r$retrieve.s$#zSteps.retrieve afterbeforer+limitorderr,r-r.r/str | NotGivenint | NotGiven#Literal[('asc', 'desc')] | NotGivenzSyncCursorPage[RunStep] r9r8rMrNr+rOrPr,r-r.r/rc Csz|std||s$td|ddi|p0i}|jd|d|dttt|| | | t|||||dtjd td S a Returns a list of run steps belonging to a run. Args: after: A cursor for use in pagination. `after` is an object ID that defines your place in the list. For instance, if you make a list request and receive 100 objects, ending with obj_foo, your subsequent call can include after=obj_foo in order to fetch the next page of the list. before: A cursor for use in pagination. `before` is an object ID that defines your place in the list. For instance, if you make a list request and receive 100 objects, starting with obj_foo, your subsequent call can include before=obj_foo in order to fetch the previous page of the list. include: A list of additional fields to include in the response. Currently the only supported value is `step_details.tool_calls[*].file_search.results[*].content` to fetch the file search result content. See the [file search tool documentation](https://platform.openai.com/docs/assistants/tools/file-search#customizing-file-search-settings) for more information. limit: A limit on the number of objects to be returned. Limit can range between 1 and 100, and the default is 20. order: Sort order by the `created_at` timestamp of the objects. `asc` for ascending order and `desc` for descending order. 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<r>r?r@rAz/steps)rMrNr+rOrPrC)pagerFmodel)rG _get_api_listrrrr rStepListParams r"r9r8rMrNr+rOrPr,r-r.r/r#r#r$listds26z Steps.list __name__ __module__ __qualname__rr%r(typing_extensions deprecatedrrKr[r#r#r#r$rs, $5c@seZdZeddddZeddddZeded d d ed d d d d d ddddd ddZ edeeeeed d d ed d d ddd ddd ddddd ddZ d S)rAsyncStepsWithRawResponsercCst|Sr )rbr!r#r#r$r%szAsyncSteps.with_raw_responseAsyncStepsWithStreamingResponsecCst|Sr')rcr!r#r#r$r(sz"AsyncSteps.with_streaming_responser)Nr*r0r1r2r3r4r5rr6c s|std||s$td||s6td|ddi|pBi}|jd|d|d|t||||td |itjId Hd td Id HS) r:r;r<r=r>r?r@rArBr+NrCrE)rGrHrr rrIrrJr#r#r$rKs$#zAsyncSteps.retrieverLrQrRrSz1AsyncPaginator[RunStep, AsyncCursorPage[RunStep]]rTc Csz|std||s$td|ddi|p0i}|jd|d|dttt|| | | t|||||dtjd td SrU)rGrXrrrr rrYrZr#r#r$r[s26zAsyncSteps.listr\r#r#r#r$rs, $5c@seZdZdddddZdS)rrNonestepsrcCs&||_t|j|_t|j|_dSN)_stepsrZto_raw_response_wrapperrKr[r"rfr#r#r$__init__SszStepsWithRawResponse.__init__Nr]r^r_rjr#r#r#r$rRsrc@seZdZdddddZdS)rbrrdrecCs&||_t|j|_t|j|_dSrg)rhrZasync_to_raw_response_wrapperrKr[rir#r#r$rjcsz"AsyncStepsWithRawResponse.__init__Nrkr#r#r#r$rbbsrbc@seZdZdddddZdS)r&rrdrecCs"||_t|j|_t|j|_dSrg)rhrrKr[rir#r#r$rjssz#StepsWithStreamingResponse.__init__Nrkr#r#r#r$r&rsr&c@seZdZdddddZdS)rcrrdrecCs"||_t|j|_t|j|_dSrg)rhrrKr[rir#r#r$rjsz(AsyncStepsWithStreamingResponse.__init__Nrkr#r#r#r$rcsrc). __future__rr`typingrrZhttpxr_typesrrr r r _utilsr r _compatrZ _resourcerr _responserrZ paginationrrZ _base_clientrrZtypes.beta.threads.runsrrZ types.beta.threads.runs.run_steprZ(types.beta.threads.runs.run_step_includer__all__rrrrbr&rcr#r#r#r$s0