g- ddlmZddlZddlmZmZmZmZddlm Z ddlm Z m Z ddl Z ddl mZdd lmZmZmZmZmZmZdd lmZmZmZmZmZdd lmZmZmZm Z dd l!m"Z"dd l#m$Z$m%Z%ddl&m'Z'm(Z(ddl)m*Z*m+Z+ddl,m-Z-m.Z.ddl/m0Z0m1Z1ddl2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8ddl9m:Z:ddl;mZ>m?Z?ddl@mAZAddlBmCZCddlDmEZEddlFmGZGddlHmIZIddlJmKZKddgZLGdde$ZMGdde%ZNGdd ZOGd!d"ZPGd#d$ZQGd%d&ZRy)') annotationsN)ListUnionIterableOptional)partial)Literaloverload)_legacy_response)Steps AsyncStepsStepsWithRawResponseAsyncStepsWithRawResponseStepsWithStreamingResponseAsyncStepsWithStreamingResponse) NOT_GIVENBodyQueryHeadersNotGiven)is_given required_argsmaybe_transformasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_streamed_response_wrapper"async_to_streamed_response_wrapper)Stream AsyncStream)SyncCursorPageAsyncCursorPage)AsyncPaginatormake_request_options)AssistantEventHandlerAssistantEventHandlerTAssistantStreamManagerAsyncAssistantEventHandlerAsyncAssistantEventHandlerTAsyncAssistantStreamManager) ChatModel)run_list_paramsrun_create_paramsrun_update_paramsrun_submit_tool_outputs_params)Run)AssistantToolParam)AssistantStreamEvent)RunStepInclude)AssistantToolChoiceOptionParam)"AssistantResponseFormatOptionParamRuns AsyncRunsc" eZdZed/dZed0dZed1dZeeeeeeeeeeeeeeeeeddded d2d Z eeeeeeeeeeeeeeeeddded  d3d Z eeeeeeeeeeeeeeeeddded  d4d Z e dgddgeeeeeeeeeeeeeeeeddded d5d Z ddded d6dZ eddded d7dZ eeeeddded d8dZ ddded d6dZeeeeeeeeeeeeeeeeddded d9dZeej"deeeeeeeeeeeeeeddded d:dZeej"deeeeeeeeeeeeeeddded d;dZej"deeeeeeeeeeeeeedddded ddZeeeeeeeeeeeeeeeeddded  d?dZeeeeeeeeeeeeeeedddded  d@d!Zeeddded" dAd$Zeddded dBd%Zeddded dCd&Ze dd#ggd'eddded" dDd(Zeddded) dEd*Zeddded dFd+Zeddded dGd,Zdddded- dHd.Zy)Ir9c,t|jSN)r_clientselfs \/var/www/openai/venv/lib/python3.12/site-packages/openai/resources/beta/threads/runs/runs.pystepsz Runs.steps<sT\\""ct|Sa This property can be used as a prefix for any HTTP method call to return the 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 )RunsWithRawResponser?s rAwith_raw_responsezRuns.with_raw_response@s#4((rCct|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 )RunsWithStreamingResponser?s rAwith_streaming_responsezRuns.with_streaming_responseJs)..rCNincludeadditional_instructionsadditional_messages instructionsmax_completion_tokensmax_prompt_tokensmetadatamodelparallel_tool_callsresponse_formatstream temperature tool_choicetoolstop_ptruncation_strategy extra_headers extra_query extra_bodytimeout thread_id assistant_idrWcya Create a run. Args: assistant_id: The ID of the [assistant](https://platform.openai.com/docs/api-reference/assistants) to use to execute this run. 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. additional_instructions: Appends additional instructions at the end of the instructions for the run. This is useful for modifying the behavior on a per-run basis without overriding other instructions. additional_messages: Adds additional messages to the thread before creating the run. instructions: Overrides the [instructions](https://platform.openai.com/docs/api-reference/assistants/createAssistant) of the assistant. This is useful for modifying the behavior on a per-run basis. max_completion_tokens: The maximum number of completion tokens that may be used over the course of the run. The run will make a best effort to use only the number of completion tokens specified, across multiple turns of the run. If the run exceeds the number of completion tokens specified, the run will end with status `incomplete`. See `incomplete_details` for more info. max_prompt_tokens: The maximum number of prompt tokens that may be used over the course of the run. The run will make a best effort to use only the number of prompt tokens specified, across multiple turns of the run. If the run exceeds the number of prompt tokens specified, the run will end with status `incomplete`. See `incomplete_details` for more info. metadata: Set of 16 key-value pairs that can be attached to an object. This can be useful for storing additional information about the object in a structured format. Keys can be a maximum of 64 characters long and values can be a maximum of 512 characters long. model: The ID of the [Model](https://platform.openai.com/docs/api-reference/models) to be used to execute this run. If a value is provided here, it will override the model associated with the assistant. If not, the model associated with the assistant will be used. parallel_tool_calls: Whether to enable [parallel function calling](https://platform.openai.com/docs/guides/function-calling#configuring-parallel-function-calling) during tool use. response_format: Specifies the format that the model must output. Compatible with [GPT-4o](https://platform.openai.com/docs/models#gpt-4o), [GPT-4 Turbo](https://platform.openai.com/docs/models#gpt-4-turbo-and-gpt-4), and all GPT-3.5 Turbo models since `gpt-3.5-turbo-1106`. Setting to `{ "type": "json_schema", "json_schema": {...} }` enables Structured Outputs which ensures the model will match your supplied JSON schema. Learn more in the [Structured Outputs guide](https://platform.openai.com/docs/guides/structured-outputs). Setting to `{ "type": "json_object" }` enables JSON mode, which ensures the message the model generates is valid JSON. **Important:** when using JSON mode, you **must** also instruct the model to produce JSON yourself via a system or user message. Without this, the model may generate an unending stream of whitespace until the generation reaches the token limit, resulting in a long-running and seemingly "stuck" request. Also note that the message content may be partially cut off if `finish_reason="length"`, which indicates the generation exceeded `max_tokens` or the conversation exceeded the max context length. stream: If `true`, returns a stream of events that happen during the Run as server-sent events, terminating when the Run enters a terminal state with a `data: [DONE]` message. temperature: What sampling temperature to use, between 0 and 2. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. tool_choice: Controls which (if any) tool is called by the model. `none` means the model will not call any tools and instead generates a message. `auto` is the default value and means the model can pick between generating a message or calling one or more tools. `required` means the model must call one or more tools before responding to the user. Specifying a particular tool like `{"type": "file_search"}` or `{"type": "function", "function": {"name": "my_function"}}` forces the model to call that tool. tools: Override the tools the assistant can use for this run. This is useful for modifying the behavior on a per-run basis. top_p: An alternative to sampling with temperature, called nucleus sampling, where the model considers the results of the tokens with top_p probability mass. So 0.1 means only the tokens comprising the top 10% probability mass are considered. We generally recommend altering this or temperature but not both. truncation_strategy: Controls for how a thread will be truncated prior to the run. Use this to control the intial context window of the run. 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 Nr@rarbrMrNrOrPrQrRrSrTrUrVrWrXrYrZr[r\r]r^r_r`s rAcreatez Runs.createST rCrMrNrOrPrQrRrSrTrUrVrXrYrZr[r\r]r^r_r`cya Create a run. Args: assistant_id: The ID of the [assistant](https://platform.openai.com/docs/api-reference/assistants) to use to execute this run. stream: If `true`, returns a stream of events that happen during the Run as server-sent events, terminating when the Run enters a terminal state with a `data: [DONE]` message. 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. additional_instructions: Appends additional instructions at the end of the instructions for the run. This is useful for modifying the behavior on a per-run basis without overriding other instructions. additional_messages: Adds additional messages to the thread before creating the run. instructions: Overrides the [instructions](https://platform.openai.com/docs/api-reference/assistants/createAssistant) of the assistant. This is useful for modifying the behavior on a per-run basis. max_completion_tokens: The maximum number of completion tokens that may be used over the course of the run. The run will make a best effort to use only the number of completion tokens specified, across multiple turns of the run. If the run exceeds the number of completion tokens specified, the run will end with status `incomplete`. See `incomplete_details` for more info. max_prompt_tokens: The maximum number of prompt tokens that may be used over the course of the run. The run will make a best effort to use only the number of prompt tokens specified, across multiple turns of the run. If the run exceeds the number of prompt tokens specified, the run will end with status `incomplete`. See `incomplete_details` for more info. metadata: Set of 16 key-value pairs that can be attached to an object. This can be useful for storing additional information about the object in a structured format. Keys can be a maximum of 64 characters long and values can be a maximum of 512 characters long. model: The ID of the [Model](https://platform.openai.com/docs/api-reference/models) to be used to execute this run. If a value is provided here, it will override the model associated with the assistant. If not, the model associated with the assistant will be used. parallel_tool_calls: Whether to enable [parallel function calling](https://platform.openai.com/docs/guides/function-calling#configuring-parallel-function-calling) during tool use. response_format: Specifies the format that the model must output. Compatible with [GPT-4o](https://platform.openai.com/docs/models#gpt-4o), [GPT-4 Turbo](https://platform.openai.com/docs/models#gpt-4-turbo-and-gpt-4), and all GPT-3.5 Turbo models since `gpt-3.5-turbo-1106`. Setting to `{ "type": "json_schema", "json_schema": {...} }` enables Structured Outputs which ensures the model will match your supplied JSON schema. Learn more in the [Structured Outputs guide](https://platform.openai.com/docs/guides/structured-outputs). Setting to `{ "type": "json_object" }` enables JSON mode, which ensures the message the model generates is valid JSON. **Important:** when using JSON mode, you **must** also instruct the model to produce JSON yourself via a system or user message. Without this, the model may generate an unending stream of whitespace until the generation reaches the token limit, resulting in a long-running and seemingly "stuck" request. Also note that the message content may be partially cut off if `finish_reason="length"`, which indicates the generation exceeded `max_tokens` or the conversation exceeded the max context length. temperature: What sampling temperature to use, between 0 and 2. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. tool_choice: Controls which (if any) tool is called by the model. `none` means the model will not call any tools and instead generates a message. `auto` is the default value and means the model can pick between generating a message or calling one or more tools. `required` means the model must call one or more tools before responding to the user. Specifying a particular tool like `{"type": "file_search"}` or `{"type": "function", "function": {"name": "my_function"}}` forces the model to call that tool. tools: Override the tools the assistant can use for this run. This is useful for modifying the behavior on a per-run basis. top_p: An alternative to sampling with temperature, called nucleus sampling, where the model considers the results of the tokens with top_p probability mass. So 0.1 means only the tokens comprising the top 10% probability mass are considered. We generally recommend altering this or temperature but not both. truncation_strategy: Controls for how a thread will be truncated prior to the run. Use this to control the intial context window of the run. 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 Nrer@rarbrWrMrNrOrPrQrRrSrTrUrVrXrYrZr[r\r]r^r_r`s rArgz Runs.createrhrCcyrkrerls rArgz Runs.createkrhrCcr|std|ddi|xsi}|jd|dtid|d|d|d |d |d |d | d | d| d| d| d|d|d|d|d|tjt ||||td|itjt | xsdttSN8Expected a non-empty value for `thread_id` but received OpenAI-Beta assistants=v2 /threads//runsrbrNrOrPrQrRrSrTrUrVrWrXrYrZr[r\rMr]r^r_r`queryFbodyoptionscast_torW stream_cls) ValueError_postrr0RunCreateParamsr'r3r"r5rfs rArgz Runs.createsn:WXaWdef f&QM#L  ,-B  (): U*+>&f";";UU *+>!$"11'*)+'%%y'&:#L  ,-B  (): U&";";dU*+>*+> U!$"11'*)+Q[el23; >&l-BjShSjkkrCcjddi|xsi}t|rt||d<hd} |jj||||||} | j } | j |vr| St|s+| j jd} | t| }nd}|j|dz ) A helper to poll a run status until it reaches a terminal state. More information on Run lifecycles can be found here: https://platform.openai.com/docs/assistants/how-it-works/runs-and-run-steps X-Stainless-Poll-Helperr X-Stainless-Custom-Poll-Interval>failedexpired cancelled completed incompleterequires_actionrarr]r_r^r`openai-poll-after-ms rstrrGrparsestatusheadersgetint_sleep r@rrar]r^r_r`rterminal_statesresponser from_headers rArz Runs.polls3FT}?RPRT $ %@CDT@UM< =j--66#+%' 7H.."Czz_, ,-&..223IJ *'*;'7$'+$ KK(4/ 0-rCcyrre)r@rbrMrNrOrPrQrRrSrTrUrVrXrYrZr[r\rar]r^r_r`s rArWz Runs.streamrrCcyrrer@rbrMrNrOrPrQrRrSrTrUrVrXrYrZr[r\rarr]r^r_r`s rArWz Runs.stream rrCrMrNrOrPrQrRrSrTrUrVrXrYrZr[r\rr]r^r_r`c|std|dd|rdndd|xsi}t|jd|dtid |d |d |d |d |d|d|d| d| d| d| ddd|d| d|d|tj t ||||td|itj tdtt}t||xs tSrrprrrrrrrsrtrbrNrOrPrQrRrSrTrVrXrYrWTrZrUr\r[rMrurwrr)r@rbrMrNrOrPrQrRrSrTrUrVrXrYrZr[r\rarr]r^r_r`rs rArWz Runs.stream)s:WXaWdef f+)I:GW "   JJ {% ( "L-/F*+>#L  ,-B  (): U&";";dU*+>*+> U!$"11'*)+'%%y'&:dZeeeeeeeeeeeeeeedddded  d?d!Zeeddded" d@d$Zeddded dAd%Zeddded dBd&Ze dd#ggd'eddded" dCd(Zeddded) dDd*Zeddded dEd+Zeddded dFd,Zdddded- dGd.Zy)Hr:c,t|jSr=)rr>r?s rArBzAsyncRuns.stepss$,,''rCct|SrE)AsyncRunsWithRawResponser?s rArGzAsyncRuns.with_raw_responses(--rCct|SrI)AsyncRunsWithStreamingResponser?s rArKz!AsyncRuns.with_streaming_responses.d33rCNrLrarbrWc Kywrdrerfs rArgzAsyncRuns.create T ric Kywrkrerls rArgzAsyncRuns.create4r<r=c Kywrkrerls rArgzAsyncRuns.creater<r=cK|std|ddi|xsi}|jd|dtid|d|d|d |d |d |d | d | d| d| d| d|d|d|d|d|tjd{t ||||td|itjd{t | xsdttd{S7X7.7wro) r|r}rr0r~r'r3r#r5rfs rArgzAsyncRuns.createLs:WXaWdef f&QM#L  ,-B  (): U*+>&f";";UU *+>!$"11'*)+'%19g2FHYHiHijj ?U"#78A ! ! ! 4k9! s6A8C:C ;+C&C''CCCCCrc K|std||std|ddi|xsi}|jd|d|t||||td {S7w) rrprrqrrrsrrrNrrs rArzAsyncRuns.retrieves0WXaWdef fTU[T^_` `&QMB?BBrc|std|ddi|xsi}|jd|dttt |||| t ||||dt jtSr)r|rr%r3r'rr/rrs rArzAsyncRuns.listsVWXaWdef f&QM#L  ,-B  (): U&";";dU*+>U *+>!$"11'*)+Q[el"#789 <+7-BoSmSopprCcKddi|xsi}t|rt||d<hd} |jj||||||d{} | j } | j |vr| St|s+| j jd} | t| }nd}|j|dz d{7w7w) rrrr>rrrrrrrNrrrrs rArzAsyncRuns.poll s3FT}?RPRT $ %@CDT@UM< =j!33<<#+%' =H.."Czz_, ,-&..223IJ *'*;'7$'+$++.56 6 6-* 7s%A C CA1C C C  C cyrrers rArWzAsyncRuns.streamA s8 rCcyrrers rArWzAsyncRuns.stream_ rrCrc|std|dd|rdndd|xsi}|jd|dtid |d |d |d |d |d|d|d| d| d| d| ddd|d| d|d|tjt ||||td|itjt dtt}t||xs tSrrI)r@rbrMrNrOrPrQrRrSrTrUrVrXrYrZr[r\rarr]r^r_r`rs rArWzAsyncRuns.stream s@WXaWdef f+)I:GW "  ** {% ( "L-/F*+>#L  ,-B  (): U&";";dU*+>*+> U!$"11'*)+'%%y'&: sWXaWdef fTU[T^_` `&QM|std||std|dd|rdndd|xsi}|jd|d |d t|d d tjt |||| t d tt} t| |xs tSr) r|r}rr2rr'r3r#r5r-r+rs rArz$AsyncRuns.submit_tool_outputs_stream s,TU[T^_` `WXaWdef f+)R:GW "  ** {&0D E $0"/II )+Q[el"#78  +7-BoSmSopprC)r r)r r8)r r:r ).rarrbrrWrrMr rNr rOr rPr rQrrRrrSrrTrrUrrVrrXrrYrrZrr[rr\rr]rr^rr_rr`rr !AsyncStream[AssistantStreamEvent]).rarrbrrWrrMr rNr rOr rPr rQrrRrrSrrTrrUrrVrrXrrYrrZrr[rr\rr]rr^rr_rr`rr 'Run | AsyncStream[AssistantStreamEvent]).rarrbrrMr rNr rOr rPr rQrrRrrSrrTrrUrrVrrWr rXrrYrrZrr[rr\rr]rr^rr_rr`rr rXr!r")rarrr#rr#rr$rr%r]rr^rr_rr`rr z)AsyncPaginator[Run, AsyncCursorPage[Run]]r&)*rbrrNr rOr rPr rQrrRrrSrrTrrUrrVrrXrrYrrZrr[rr\rrarr]rr^rr_rr`rr 7AsyncAssistantStreamManager[AsyncAssistantEventHandler]),rbrrNr rOr rPr rQrrRrrSrrTrrUrrVrrXrrYrrZrr[rr\rrarrr,r]rr^rr_rr`rr 8AsyncAssistantStreamManager[AsyncAssistantEventHandlerT]),rbrrNr rOr rPr rQrrRrrSrrTrrUrrVrrXrrYrrZrr[rr\rrarr"AsyncAssistantEventHandlerT | Noner]rr^rr_rr`rr rAsyncAssistantStreamManager[AsyncAssistantEventHandler] | AsyncAssistantStreamManager[AsyncAssistantEventHandlerT]r+).rbrrMr rNr rOr rPr rQrrRrrSrrTrrUrrVrrXrrYrrZrr[rr\rrarrr,r]rr^rr_rr`rr rZ).rbrrMr rNr rOr rPr rQrrRrrSrrTrrUrrVrrXrrYrrZrr[rr\rrarrr[r]rr^rr_rr`rr r\r,)rrrarrWrrr-r]rr^rr_rr`rr rW)rrrarrWrrr-r]rr^rr_rr`rr rX)rrrarrr-rWr r]rr^rr_rr`rr rXr.)rr-rrrarr]rr^rr_rr`rr rY)rr-rrrarrr,r]rr^rr_rr`rr rZ)rr-rrrarrr[r]rr^rr_rr`rr r\r/rerCrAr:r:s((..44 4=> KK  )@@ MM  '>> KK  %<< II  '>> KK  $4#K#K  $ $$  rCc@t|jjSr=)rr_rBr?s rArBzRunsWithRawResponse.steps s#DJJ$4$455rCNrbr9r None)r rr0r1r2rcrrBrerCrArFrF s ,66rCrFc&eZdZddZeddZy)r8c||_tj|j|_tj|j|_tj|j |_tj|j |_tj|j|_tj|j|_yr=) r_r async_to_raw_response_wrapperrgrrrrrras rArcz!AsyncRunsWithRawResponse.__init__ s &DD KK  )FF MM  'DD KK  %BB II  'DD KK  $4#Q#Q  $ $$  rCc@t|jjSr=)rr_rBr?s rArBzAsyncRunsWithRawResponse.steps s()9)9::rCNrbr:r rf)r rrgrerCrAr8r8 s ,;;rCr8c&eZdZddZeddZy)rJcJ||_t|j|_t|j|_t|j|_t|j |_t|j |_t|j|_yr=)r_r rgrrrrrras rArcz"RunsWithStreamingResponse.__init__# s 2 KK  5 MM  3 KK  1 II  3 KK  $@  $ $$  rCc@t|jjSr=)rr_rBr?s rArBzRunsWithStreamingResponse.steps9 s)$***:*:;;rCNre)r rrgrerCrArJrJ" s ,< s ,AArCr:)S __future__rr3typingrrrr functoolsrr r httpxr rBrrrrrr_typesrrrrr_utilsrrrr_compatr _resourcerr _responser r! _streamingr"r# paginationr$r% _base_clientr&r' lib.streamingr(r)r*r+r,r-types.chat_modelr.types.beta.threadsr/r0r1r2types.beta.threads.runr3types.beta.assistant_tool_paramr4!types.beta.assistant_stream_eventr5(types.beta.threads.runs.run_step_includer6-types.beta.assistant_tool_choice_option_paramr71types.beta.assistant_response_format_option_paramr8__all__r9r:rFr8rJr:rerCrArs#22/ "BA )<[/;B, ,CGH]e ; Rg?Rgj*Wq Wqt*668;;8<<8AArC