o Tj@sddlmZddlmZmZmZddlmZddlZddl m Z ddl m Z m Z mZmZmZmZmZddlmZmZmZdd lmZdd lmZmZdd lmZmZdd lm Z m!Z!dd l"m#Z#m$Z$m%Z%ddl&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,ddl-m.Z.m/Z/ddl0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6ddl7m8Z8ddl9m:Z:m;Z;mZ>m?Z?ddl@mAZAddlBmCZCddlDmEZEddlFmGZGddlHmIZIddgZJGdddeZKGdddeZLGdddZMGdd d ZNGd!d"d"ZOGd#d$d$ZPdS)%) annotations)DictIterableOptional)LiteralN)_legacy_response)BodyOmitQueryHeadersNotGivenomit not_given) path_templatemaybe_transformasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_streamed_response_wrapper"async_to_streamed_response_wrapper)SyncCursorPageAsyncCursorPage)agent_list_paramsagent_create_paramsagent_update_params)Vaults AsyncVaultsVaultsWithRawResponseAsyncVaultsWithRawResponseVaultsWithStreamingResponse AsyncVaultsWithStreamingResponse)AsyncPaginatormake_request_options)Sessions AsyncSessionsSessionsWithRawResponseAsyncSessionsWithRawResponseSessionsWithStreamingResponse"AsyncSessionsWithStreamingResponse)Agent) EnvironmentsAsyncEnvironmentsEnvironmentsWithRawResponse AsyncEnvironmentsWithRawResponse!EnvironmentsWithStreamingResponse&AsyncEnvironmentsWithStreamingResponse) AgentDeleted)AgentTextParam)AgentReasoningParam)MultiAgentConfigParam)PersistedAgentToolParamAgents AsyncAgentsc@eZdZedCddZedDddZedEd d ZedFd d ZedGddZe e e e e e e e ddde d dHd-d.Z ddde d/dId1d2Z e e e e e e e e e ddde d3 dJd5d6Z e e e ddde d7dKd>d?Zddde d/dLdAdBZdS)Mr8returnr-cC t|jSN)r-_clientselfrAf/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/openai/resources/beta/agents/agents.py environments6 zAgents.environmentsrcCr<r=)rr>r?rArArBvaults:rDz Agents.vaultsr&cCr<r=)r&r>r?rArArBsessions>rDzAgents.sessionsAgentsWithRawResponsecCt|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 )rGr?rArArBwith_raw_responseBzAgents.with_raw_responseAgentsWithStreamingResponsecCrHz 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 )rLr?rArArBwith_streaming_responseLzAgents.with_streaming_responseN instructionsmetadata multi_agentname reasoning service_tiertexttools extra_headers extra_query extra_bodytimeoutmodelstrrQOptional[str] | OmitrROptional[Dict[str, str]] | OmitrS&Optional[MultiAgentConfigParam] | OmitrTrU$Optional[AgentReasoningParam] | OmitrVGOptional[Literal['auto', 'default', 'flex', 'priority', 'fast']] | OmitrWOptional[AgentTextParam] | OmitrX2Optional[Iterable[PersistedAgentToolParam]] | OmitrYHeaders | NonerZ Query | Noner[ Body | Noner\(float | httpx2.Timeout | None | NotGivenr,c  CsPddi| pi} |jdt||||||||| d tjt| | | | ddidtdS) Creates a reusable agent without storing credentials. See [agent configuration](https://developers.openai.com/api/docs/guides/agents-api/configuration). Args: model: The model to use for the agent. The requested model name is preserved. instructions: Additional instructions appended to the agent's default base instructions. Omit or set to null to add no custom instructions. metadata: Up to 16 string key-value pairs, with keys up to 64 and values up to 512 characters. Omission or null defaults to an empty map. multi_agent: Configuration for creating and coordinating subagents. Subagent tools are disabled by default. name: A human-readable name for the agent. Omission or null leaves the agent unnamed. reasoning: Configuration for model reasoning. Omission uses the model's default effort. service_tier: The service tier used for model requests. Defaults to `auto`. - `auto` - Selects the service tier automatically. - `default` - Uses the default service tier. - `flex` - Uses the flex service tier. - `priority` - Uses the priority service tier. - `fast` - Uses the fast service tier. text: Configuration for generated text. Defaults to the `text` format and medium verbosity. tools: Tools available to the agent. Defaults to an empty list. 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 OpenAI-Beta agents=v1/agents r]rQrRrSrTrUrVrWrX bearer_authTrYrZr[r\securitybodyoptionscast_to)_postrrAgentCreateParamsr%r,r@r]rQrRrSrTrUrVrWrXrYrZr[r\rArArBcreateUs2= z Agents.createrYrZr[r\agent_idc CL|s td|ddi|pi}|jtd|dt||||ddidtd S) Retrieves a reusable agent by ID. See [agent configuration](https://developers.openai.com/api/docs/guides/agents-api/configuration). 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 7Expected a non-empty value for `agent_id` but received rkrl/agents/{agent_id}r{roTrprtru ValueError_getrr%r,r@r{rYrZr[r\rArArBretrieve zAgents.retrieve rQrRr]rSrTrUrVrWrXrYrZr[r\ str | Omitc  Csj|s td|ddi| pi} |jtd|dt|||||||| | d tjt| | | |ddid td S) Updates a reusable agent. See [agent configuration](https://developers.openai.com/api/docs/guides/agents-api/configuration). Args: instructions: Additional instructions appended to the agent's default base instructions. Omit to leave unchanged. metadata: Replaces all metadata. Omit to leave unchanged, or pass null or {} to clear it. Up to 16 string key-value pairs, with keys up to 64 and values up to 512 characters. model: The model to use for the agent. The requested model name is preserved. multi_agent: Configuration for creating and coordinating subagents. name: A replacement name. Omit to leave unchanged, or pass null to clear it. reasoning: Configuration for model reasoning. Omit to keep the current settings; pass `null` to reset to the model's default effort. service_tier: The service tier used for model requests. - `auto` - Selects the service tier automatically. - `default` - Uses the default service tier. - `flex` - Uses the flex service tier. - `priority` - Uses the priority service tier. - `fast` - Uses the fast service tier. text: Configuration for text generated by the agent. tools: Tools available to the agent. 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~rkrlrr rQrRr]rSrTrUrVrWrXroTrprr)rrvrrrAgentUpdateParamsr%r,r@r{rQrRr]rSrTrUrVrWrXrYrZr[r\rArArBupdates6>  z Agents.updateafterlimitorderrYrZr[r\rrOptional[int] | OmitrLiteral['asc', 'desc'] | OmitSyncCursorPage[Agent]c CJddi|pi}|jdttt||||t|||dtjddidtdS a*Lists reusable agents in the current project. See [agent configuration](https://developers.openai.com/api/docs/guides/agents-api/configuration). Args: after: Return resources after this resource ID in the selected order. limit: The maximum number of resources to return. order: The order in which resources are returned. Defaults to `desc`. - `asc` - Returns resources in ascending order. - `desc` - Returns resources in 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 rkrlrm)rrrroT)rYrZr[r\queryrq)pagertr]) _get_api_listrr,r%rrAgentListParamsr@rrrrYrZr[r\rArArBlist0($z Agents.listr3c Cr|) Deletes a reusable agent. See [agent configuration](https://developers.openai.com/api/docs/guides/agents-api/configuration). 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~rkrlrrroTrprr_deleterr%r3rrArArBdeletejrz Agents.delete)r;r-)r;r)r;r&)r;rG)r;rLr]r^rQr_rRr`rSrarTr_rUrbrVrcrWrdrXrerYrfrZrgr[rhr\rir;r, r{r^rYrfrZrgr[rhr\rir;r,r{r^rQr_rRr`r]rrSrarTr_rUrbrVrcrWrdrXrerYrfrZrgr[rhr\rir;r,)rrrrrrrYrfrZrgr[rhr\rir;r r{r^rYrfrZrgr[rhr\rir;r3__name__ __module__ __qualname__rrCrErFrJrNrrryrrrrrArArArBr85p      ^,^@c@r:)Mr9r;r.cCr<r=)r.r>r?rArArBrCrDzAsyncAgents.environmentsrcCr<r=)rr>r?rArArBrErDzAsyncAgents.vaultsr'cCr<r=)r'r>r?rArArBrFrDzAsyncAgents.sessionsAsyncAgentsWithRawResponsecCrHrI)rr?rArArBrJrKzAsyncAgents.with_raw_response AsyncAgentsWithStreamingResponsecCrHrM)rr?rArArBrNrOz#AsyncAgents.with_streaming_responseNrPr]r^rQr_rRr`rSrarTrUrbrVrcrWrdrXrerYrfrZrgr[rhr\rir,c  s^ddi| pi} |jdt||||||||| d tjIdHt| | | | ddidtd IdHS) rjrkrlrmrnNroTrprr)rvrrrwr%r,rxrArArBrys4=  zAsyncAgents.createrzr{c T|s td|ddi|pi}|jtd|dt||||ddidtd Id HS) r}r~rkrlrrroTrprNrrrArArBr   zAsyncAgents.retrieverrc  sx|s td|ddi| pi} |jtd|dt|||||||| | d tjIdHt| | | |dd id td IdHS) rr~rkrlrrrNroTrprr)rrvrrrrr%r,rrArArBr3s8>   zAsyncAgents.updaterrrrrr-AsyncPaginator[Agent, AsyncCursorPage[Agent]]c Crr)rrr,r%rrrrrArArBrrzAsyncAgents.listr3c r) rr~rkrlrrroTrprNrrrArArBrrzAsyncAgents.delete)r;r.)r;r)r;r')r;r)r;rrrr)rrrrrrrYrfrZrgr[rhr\rir;rrrrArArArBr9rc@@eZdZdddZeddd Zedd d ZedddZdS)rGagentsr8r;NonecCP||_t|j|_t|j|_t|j|_t|j|_t|j|_dSr=)_agentsrto_raw_response_wrapperryrrrrr@rrArArB__init__  zAgentsWithRawResponse.__init__r/cC t|jjSr=)r/rrCr?rArArBrC z"AgentsWithRawResponse.environmentsr cCrr=)r rrEr?rArArBrE rzAgentsWithRawResponse.vaultsr(cCrr=)r(rrFr?rArArBrF rzAgentsWithRawResponse.sessionsNrr8r;r)r;r/)r;r )r;r(rrrrrrCrErFrArArArBrG   rGc@r)rrr9r;rcCrr=)rrasync_to_raw_response_wrapperryrrrrrrArArBrrz#AsyncAgentsWithRawResponse.__init__r0cCrr=)r0rrCr?rArArBrC&rz'AsyncAgentsWithRawResponse.environmentsr!cCrr=)r!rrEr?rArArBrE*rz!AsyncAgentsWithRawResponse.vaultsr)cCrr=)r)rrFr?rArArBrF.rz#AsyncAgentsWithRawResponse.sessionsNrr9r;r)r;r0)r;r!)r;r)rrArArArBrrrc@r)rLrr8r;rcCF||_t|j|_t|j|_t|j|_t|j|_t|j|_dSr=)rrryrrrrrrArArBr4  z$AgentsWithStreamingResponse.__init__r1cCrr=)r1rrCr?rArArBrCGrz(AgentsWithStreamingResponse.environmentsr"cCrr=)r"rrEr?rArArBrEKrz"AgentsWithStreamingResponse.vaultsr*cCrr=)r*rrFr?rArArBrFOrz$AgentsWithStreamingResponse.sessionsNr)r;r1)r;r")r;r*rrArArArBrL3rrLc@r)rrr9r;rcCrr=)rrryrrrrrrArArBrUrz)AsyncAgentsWithStreamingResponse.__init__r2cCrr=)r2rrCr?rArArBrChrz-AsyncAgentsWithStreamingResponse.environmentsr#cCrr=)r#rrEr?rArArBrElrz'AsyncAgentsWithStreamingResponse.vaultsr+cCrr=)r+rrFr?rArArBrFprz)AsyncAgentsWithStreamingResponse.sessionsNr)r;r2)r;r#)r;r+rrArArArBrTrr)Q __future__rtypingrrrtyping_extensionsrhttpx2r_typesr r r r r rr_utilsrrr_compatr _resourcerr _responserr paginationrr types.betarrr vaults.vaultsrrr r!r"r# _base_clientr$r%sessions.sessionsr&r'r(r)r*r+types.beta.agentr,environments.environmentsr-r.r/r0r1r2types.beta.agent_deletedr3types.beta.agent_text_paramr4 types.beta.agent_reasoning_paramr5#types.beta.multi_agent_config_paramr6%types.beta.persisted_agent_tool_paramr7__all__r8r9rGrrLrrArArArBsB   $          ``!!!