o TjZ@sTddlmZddlmZmZmZmZmZddlm Z m Z ddl Z ddl m Z ddlmZmZmZmZmZmZdd lmZmZmZmZmZmZdd lmZmZmZm Z m!Z!m"Z"dd l#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)dd l*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2dd l3m4Z4m5Z5m6Z6m7Z7ddl8m9Z9ddl:m;Z;mZ>m?Z?ddl@mAZAmBZBddlCmDZDmEZEddlFmGZGmHZHddlImJZJmKZKmLZLmMZMmNZNmOZOddlPmQZQmRZRmSZSddlTmUZUmVZVmWZWmXZXddlYmZZZddl[m\Z\ddl]m^Z^ddl_m`Z`ddlambZbddgZcGddde;ZdGddde<ZeGd d!d!ZfGd"d#d#ZgGd$d%d%ZhGd&d'd'ZidS)() annotations)DictUnionMappingIterableOptional)LiteraloverloadN)_legacy_response)Items AsyncItemsItemsWithRawResponseAsyncItemsWithRawResponseItemsWithStreamingResponseAsyncItemsWithStreamingResponse)Turns AsyncTurnsTurnsWithRawResponseAsyncTurnsWithRawResponseTurnsWithStreamingResponseAsyncTurnsWithStreamingResponse)Events AsyncEventsEventsWithRawResponseAsyncEventsWithRawResponseEventsWithStreamingResponse AsyncEventsWithStreamingResponse) ArtifactsAsyncArtifactsArtifactsWithRawResponseAsyncArtifactsWithRawResponseArtifactsWithStreamingResponse#AsyncArtifactsWithStreamingResponse)BodyOmitQueryHeadersNotGivenSequenceNotStromit not_given) path_template required_argsmaybe_transformasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_streamed_response_wrapper"async_to_streamed_response_wrapper)Stream AsyncStream)SyncCursorPageAsyncCursorPage)AsyncPaginatormake_request_options) SubagentsAsyncSubagentsSubagentsWithRawResponseAsyncSubagentsWithRawResponseSubagentsWithStreamingResponse#AsyncSubagentsWithStreamingResponse)session_list_paramssession_create_paramssession_update_params) ToolHandlerAsyncToolHandlerAgentSessionStreamAsyncAgentSessionStream) AgentSession)EnvironmentParam)AgentSessionEvent)AgentSessionDeleted)AgentSessionInputMessageParamSessions AsyncSessionsc @eZdZdededdXddZedYddZedZddZed[ddZ ed\ddZ ed]d d!Z ed^d#d$Z ed_d&d'Z eeeeeeeddded( d`d:d;Zeeeeeeddded< dad?d;Zeeeeeeddded< dbdBd;Zed)gd)d1geeeeeeddded( dcdDd;ZdddedEdddFdGZeedddedHdedJdKZeeeedddedLdfdSdTZdddedEdgdVdWZdS)hrNN tool_handlersidempotency_key extra_headerstimeout session_idstrinput-str | Iterable[AgentSessionInputMessageParam]rR Mapping[str, ToolHandler] | NonerS str | OmitrTHeaders | NonerU(float | httpx2.Timeout | None | NotGivenreturnrGc Ct|||||||dS)aMStream one turn of an idle session, subscribing before submitting input. Use as a context manager. Only one caller may submit input to the session while this helper runs. Optional tool handlers receive an arguments dict; their results are submitted automatically. See AgentSessionStream for details. rXrRrSrTrU)rGselfrVrXrRrSrTrUrcq/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/openai/resources/beta/agents/sessions/sessions.pystreamGzSessions.streamr<cC t|jSN)r<_clientrbrcrcrd subagentsa zSessions.subagentsrcCrgrh)rrirjrcrcrd artifactserlzSessions.artifactsr cCrgrh)r rirjrcrcrditemsirlzSessions.itemsrcCrgrh)rrirjrcrcrdeventsmrlzSessions.eventsrcCrgrh)rrirjrcrcrdturnsqrlzSessions.turnsSessionsWithRawResponsecCt|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 )rqrjrcrcrdwith_raw_responseuzSessions.with_raw_responseSessionsWithStreamingResponsecCrrz 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 )rvrjrcrcrdwith_streaming_responsez Sessions.with_streaming_response agentagent_idrXmetadatare vault_idsrT extra_query extra_bodyrU environmentrJr{"session_create_params.Agent | Omitr|@Union[str, Iterable[AgentSessionInputMessageParam], None] | Omitr}Optional[Dict[str, str]] | OmitreLiteral[False] | Omitr~$Optional[SequenceNotStr[str]] | Omitr Query | Noner Body | NonerIc CdSaK Creates a managed agent session, optionally submits initial input, and returns the session or streams its events when stream is true. See [running sessions](https://developers.openai.com/api/docs/guides/agents-api/sessions). Args: environment: An inline execution environment or a reference to an environment template. agent: Agent configuration. With `agent_id`, supplied fields override the saved agent for this session. Without `agent_id`, `model` is required. agent_id: The ID of a saved reusable agent. Omit `agent` to use its configuration unchanged. input: Initial input to submit when the session is created. A string is shorthand for a single user message. Required when `environment.type` is `none`, or when `stream` is `true` for an environment that is not `self_hosted`; optional for self-hosted and non-streaming execution environments. 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. stream: Whether to stream session events as server-sent events. Defaults to `false`. vault_ids: The IDs of vaults made available to the session. 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 Nrc rbrr{r|rXr}rer~rTrrrUrcrcrdcreate4zSessions.create r{r|rXr}r~rTrrrU Literal[True]Stream[AgentSessionEvent]c CraK Creates a managed agent session, optionally submits initial input, and returns the session or streams its events when stream is true. See [running sessions](https://developers.openai.com/api/docs/guides/agents-api/sessions). Args: environment: An inline execution environment or a reference to an environment template. stream: Whether to stream session events as server-sent events. Defaults to `false`. agent: Agent configuration. With `agent_id`, supplied fields override the saved agent for this session. Without `agent_id`, `model` is required. agent_id: The ID of a saved reusable agent. Omit `agent` to use its configuration unchanged. input: Initial input to submit when the session is created. A string is shorthand for a single user message. Required when `environment.type` is `none`, or when `stream` is `true` for an environment that is not `self_hosted`; optional for self-hosted and non-streaming execution environments. 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. vault_ids: The IDs of vaults made available to the session. 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 Nrc rbrrer{r|rXr}r~rTrrrUrcrcrdrrbool(AgentSession | Stream[AgentSessionEvent]c Crrrcrrcrcrdrr%Literal[False] | Literal[True] | Omitc Csbddi|pi}|jdt|||||||d|rtjntjt|| | | ddidt|p+dttd S N OpenAI-Beta agents=v1/agents/sessions)rr{r|rXr}rer~ bearer_authTrTrrrUsecurityF)bodyoptionscast_tore stream_cls) _postr/rCSessionCreateParamsStreamingSessionCreateParamsNonStreamingr;rIr6rKrrcrcrdr*s6 rTrrrUc CL|s td|ddi|pi}|jtd|dt||||ddidtd S) Retrieves the current state of a managed agent session. See [managing sessions](https://developers.openai.com/api/docs/guides/agents-api/sessions/manage). 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 9Expected a non-empty value for `session_id` but received rr/agents/sessions/{session_id}rVrTrrr ValueError_getr-r;rIrbrVrTrrrUrcrcrdretrieveYs zSessions.retriever{r}rTrrrU"session_update_params.Agent | Omitc Cs\|s td|ddi|pi}|jtd|dt||dtjt||||ddid td S) ^Updates session metadata, model, reasoning effort, or service tier. Model settings apply to subsequent turns. Omitted fields are unchanged. See [managing sessions](https://developers.openai.com/api/docs/guides/agents-api/sessions/manage). Args: agent: Model settings for subsequent turns. Omitted fields stay 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. 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 rrrrrr{r}rTrrrr)rrr-r/rDSessionUpdateParamsr;rIrbrVr{r}rTrrrUrcrcrdupdates(" zSessions.updateafterr|limitorderrTrrrUrrOptional[int] | OmitrLiteral['asc', 'desc'] | OmitSyncCursorPage[AgentSession]c CLddi|pi}|jdttt||||t||||dtjddidtdS a" Lists managed agent sessions using ID-based pagination and the requested sort order. See [managing sessions](https://developers.openai.com/api/docs/guides/agents-api/sessions/manage). Args: after: Return resources after this resource ID in the selected order. agent_id: Only return sessions whose root agent has this ID. Omit to return sessions for all agents. limit: The maximum number of resources to return. order: Sort order by the `created_at` timestamp. Use `asc` for ascending order or `desc` for descending order. 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 rrr)rr|rrrT)rTrrrUqueryr)pagermodel) _get_api_listr8rIr;r/rBSessionListParams rbrr|rrrTrrrUrcrcrdlist*) z Sessions.listrLc Cr)  Removes a managed agent session from the public API and returns a deletion confirmation. If backend execution has ended, deletion can cancel a still-open public turn and abandon unpublished outputs. Running execution must be cancelled first. Physical cleanup may continue asynchronously. See [managing sessions](https://developers.openai.com/api/docs/guides/agents-api/sessions/manage). 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 rrrrrrTrrr_deleter-r;rLrrcrcrddeletes zSessions.delete)rVrWrXrYrRrZrSr[rTr\rUr]r^rG)r^r<)r^r)r^r )r^r)r^r)r^rq)r^rvrrJr{rr|r[rXrr}rrerr~rrTr\rrrrrUr]r^rI)rrJrerr{rr|r[rXrr}rr~rrTr\rrrrrUr]r^r)rrJrerr{rr|r[rXrr}rr~rrTr\rrrrrUr]r^r)rrJr{rr|r[rXrr}rrerr~rrTr\rrrrrUr]r^r rVrWrTr\rrrrrUr]r^rIrVrWr{rr}rrTr\rrrrrUr]r^rI)rr[r|r[rrrrrTr\rrrrrUr]r^r rVrWrTr\rrrrrUr]r^rL__name__ __module__ __qualname__r+r,rer1rkrmrnrorprtrxr rr.rrrrrcrcrcrdrNF       5554,;Fc @rP)hrONrQrVrWrXrYrR%Mapping[str, AsyncToolHandler] | NonerSr[rTr\rUr]r^rHc Cr_)aYStream one turn of an idle session, subscribing before submitting input. Use as an async context manager. Only one caller may submit input to the session while this helper runs. Optional tool handlers receive an arguments dict; their results are submitted automatically. See AsyncAgentSessionStream for details. r`)rHrarcrcrdre%rfzAsyncSessions.streamr=cCrgrh)r=rirjrcrcrdrk?rlzAsyncSessions.subagentsr cCrgrh)r rirjrcrcrdrmCrlzAsyncSessions.artifactsrcCrgrh)rrirjrcrcrdrnGrlzAsyncSessions.itemsrcCrgrh)rrirjrcrcrdroKrlzAsyncSessions.eventsrcCrgrh)rrirjrcrcrdrpOrlzAsyncSessions.turnsAsyncSessionsWithRawResponsecCrrrs)rrjrcrcrdrtSruzAsyncSessions.with_raw_response"AsyncSessionsWithStreamingResponsecCrrrw)rrjrcrcrdrx]ryz%AsyncSessions.with_streaming_responserzrrJr{rr|rr}rrerr~rrrrrrIc dSrrcrrcrcrdrf4zAsyncSessions.createrrAsyncStream[AgentSessionEvent]c rrrcrrcrcrdrrr-AgentSession | AsyncStream[AgentSessionEvent]c rrrcrrcrcrdrrrc spddi|pi}|jdt|||||||d|rtjntjIdHt|| | | ddidt|p/dttd IdHSr) rr0rCrrr;rIr7rKrrcrcrdrs8  rc T|s td|ddi|pi}|jtd|dt||||ddidtd Id HS) rrrrrrrTrrNrrrcrcrdr7s  zAsyncSessions.retrieverrc sj|s td|ddi|pi}|jtd|dt||dtjIdHt||||dd id td IdHS) rrrrrrrNrTrr)rrr-r0rDrr;rIrrcrcrdr_s*"  zAsyncSessions.updaterrrrrr;AsyncPaginator[AgentSession, AsyncCursorPage[AgentSession]]c Crr)rr9rIr;r/rBrrrcrcrdrrzAsyncSessions.listrLc r) rrrrrrrTrrNrrrcrcrdrs  zAsyncSessions.delete)rVrWrXrYrRrrSr[rTr\rUr]r^rH)r^r=)r^r )r^r)r^r)r^r)r^r)r^rr)rrJrerr{rr|r[rXrr}rr~rrTr\rrrrrUr]r^r)rrJrerr{rr|r[rXrr}rr~rrTr\rrrrrUr]r^r)rrJr{rr|r[rXrr}rrerr~rrTr\rrrrrUr]r^rrr)rr[r|r[rrrrrTr\rrrrrUr]r^rrrrcrcrcrdrO$rc@\eZdZdddZeddd Zedd d ZedddZedddZedddZ dS)rqsessionsrNr^NonecCP||_t|j|_t|j|_t|j|_t|j|_t|j|_dSrh) _sessionsr to_raw_response_wrapperrrrrrrbrrcrcrd__init__  z SessionsWithRawResponse.__init__r>cC t|jjSrh)r>rrkrjrcrcrdrk z!SessionsWithRawResponse.subagentsr!cCrrh)r!rrmrjrcrcrdrmrz!SessionsWithRawResponse.artifactsrcCrrh)rrrnrjrcrcrdrnrzSessionsWithRawResponse.itemsrcCrrh)rrrorjrcrcrdro"rzSessionsWithRawResponse.eventsrcCrrh)rrrprjrcrcrdrp&rzSessionsWithRawResponse.turnsNrrNr^r)r^r>)r^r!)r^r)r^r)r^r rrrrr1rkrmrnrorprcrcrcrdrq     rqc@r)rrrOr^rcCrrh)rr async_to_raw_response_wrapperrrrrrrrcrcrdr,rz%AsyncSessionsWithRawResponse.__init__r?cCrrh)r?rrkrjrcrcrdrk?rz&AsyncSessionsWithRawResponse.subagentsr"cCrrh)r"rrmrjrcrcrdrmCrz&AsyncSessionsWithRawResponse.artifactsrcCrrh)rrrnrjrcrcrdrnGrz"AsyncSessionsWithRawResponse.itemsrcCrrh)rrrorjrcrcrdroKrz#AsyncSessionsWithRawResponse.eventsrcCrrh)rrrprjrcrcrdrpOrz"AsyncSessionsWithRawResponse.turnsNrrOr^r)r^r?)r^r")r^r)r^r)r^rrrcrcrcrdr+rrc@r)rvrrNr^rcCF||_t|j|_t|j|_t|j|_t|j|_t|j|_dSrh)rr4rrrrrrrcrcrdrU  z&SessionsWithStreamingResponse.__init__r@cCrrh)r@rrkrjrcrcrdrkhrz'SessionsWithStreamingResponse.subagentsr#cCrrh)r#rrmrjrcrcrdrmlrz'SessionsWithStreamingResponse.artifactsrcCrrh)rrrnrjrcrcrdrnprz#SessionsWithStreamingResponse.itemsrcCrrh)rrrorjrcrcrdrotrz$SessionsWithStreamingResponse.eventsrcCrrh)rrrprjrcrcrdrpxrz#SessionsWithStreamingResponse.turnsNr)r^r@)r^r#)r^r)r^r)r^rrrcrcrcrdrvTrrvc@r)rrrOr^rcCrrh)rr5rrrrrrrcrcrdr~rz+AsyncSessionsWithStreamingResponse.__init__rAcCrrh)rArrkrjrcrcrdrkrz,AsyncSessionsWithStreamingResponse.subagentsr$cCrrh)r$rrmrjrcrcrdrmrz,AsyncSessionsWithStreamingResponse.artifactsrcCrrh)rrrnrjrcrcrdrnrz(AsyncSessionsWithStreamingResponse.itemsrcCrrh)rrrorjrcrcrdrorz)AsyncSessionsWithStreamingResponse.eventsrcCrrh)rrrprjrcrcrdrprz(AsyncSessionsWithStreamingResponse.turnsNr)r^rA)r^r$)r^r)r^r)r^rrrcrcrcrdr}rr)j __future__rtypingrrrrrtyping_extensionsrr httpx2r rnr rrrrrrprrrrrrrorrrrrrrmrr r!r"r#r$_typesr%r&r'r(r)r*r+r,_utilsr-r.r/r0_compatr1 _resourcer2r3 _responser4r5 _streamingr6r7 paginationr8r9 _base_clientr:r;subagents.subagentsr<r=r>r?r@rAtypes.beta.agentsrBrCrDlib.streaming.agentsrErFrGrHtypes.beta.agent_sessionrItypes.beta.environment_paramrJtypes.beta.agent_session_eventrK types.beta.agent_session_deletedrL,types.beta.agent_session_input_message_paramrM__all__rNrOrqrrvrrcrcrcrdsL      (       aa)))