o TjR;@sddlmZddlZddlmZddlmZddlmZddl m Z m Z m Z m Z mZddlmZddlmZmZddlZd d lmZmZmZmZd d lmZmZd d lmZmZmZm Z m!Z!d dl"m#Z#m$Z$d dl%m&Z&d dl'm(Z(d dl)m*Z*d dl+m,Z,m-Z-d dl.m/Z/e rd dl0m1Z1m2Z2GdddeZ3d-ddZ4d.dd Z5d/d$d%Z6d&Z7Gd'd(d(Z8Gd)d*d*Z9Gd+d,d,Z:dS)0) annotationsN)deepcopy)uuid4) TracebackType) TYPE_CHECKINGMappingIterableIterator AsyncIterator)deque)Self TypedDict) arguments failed_event result_eventis_pending_call_race) ToolHandlerAsyncToolHandler)OmitHeadersNotGivenomit not_given)Stream AsyncStream)BadRequestError)AgentSessionEvent)AgentFunctionCallItem))SessionInputParamAgentSessionInputMessage,SessionInputParamAgentSessionInputToolResult)AgentSessionInputMessageParam)Sessions AsyncSessionsc@seZdZUded<ded<dS)_RequestOptionsHeaders | None extra_headers(float | httpx2.Timeout | None | NotGiventimeoutN)__name__ __module__ __qualname____annotations__r.r.g/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/openai/lib/streaming/agents/_streams.pyr%s  r%optionsreturncCs6|d}i|d|durdd|DiSdiS)Nr'cSs"i|] \}}|dkr||qS)idempotency-key)lower).0keyvaluer.r.r/ 's"z$_request_options..)items)r0headersr.r.r/_request_options#sr:idempotency_key str | Omitr9r&cCsB|piD]\}}|dkr|Sqt|trttS|S)Nr2)r8r3 isinstancerstrr)r;r9r5r6r.r.r/ _input_key-s  r?input-str | Iterable[AgentSessionInputMessageParam]r cCsJt|tr|s tddd|dgdg}n t|}|s tdd|dS)Nzinput must not be emptyuser input_text)typetext)rolecontentzagent.session.input.message)rDr@)r=r> ValueErrorlist)r@messagesr.r.r/ _input_event6s  rKic@s4eZdZdddZddd Zdd d Zdd dZdS) _TurnStater1NonecCs(d|_d|_t|_t|_t|_dS)NF)turn_id turn_endedset event_idsr recent_events handled_callsselfr.r.r/__init__Gs  z_TurnState.__init__eventrboolcCs|j|jvrdSt|jtkr|j|j|j|j|j|j|j dkr:|j j dur:|j dur:|j |_ |j dksI|j dksI|j dkrW|j durW|j |j krWd|_ dS)NFzagent.session.turn.createdzagent.session.turn.completedzagent.session.turn.failedzagent.session.turn.cancelledT)event_idrQlenrR_RECENT_EVENT_LIMITremovepopleftappendaddrDturn subagent_idrNrOrUrWr.r.r/acceptNs     z_TurnState.acceptcCs|jdkp |jdko |jS)Nzagent.session.failedzagent.session.idle)rDrOrbr.r.r/terminalbsz_TurnState.terminalAgentFunctionCallItem | NonecCsJ|jdks |jjdkr dS|j}|j|jf}||jvrdS|j||S)Nzagent.session.turn.item.added function_call)rDitemrNcall_idrSr_)rUrWcallr5r.r.r/ries   z_TurnState.callNr1rM)rWrr1rX)rWrr1re)r*r+r,rVrcrdrir.r.r.r/rLFs    rLc@veZdZdZdededd1ddZd2ddZd3dd Zd2d!d"Z d4d$d%Z d5d&d'Z d5d(d)Z d6d+d,Z d7d/d0ZdS)8AgentSessionStreamaSubmit input to an idle session and stream through the resulting turn's terminal session event. Use as a context manager. Only one caller may submit input to this session while the helper runs: the input endpoint does not return a turn ID for correlating concurrent writers. Initial idle events and subagent turn completions do not end iteration. Failed/cancelled turns remain visible as events; an unexpected end of the connection raises RuntimeError. Closing the stream does not cancel the backend turn. Optional tool handlers run sequentially during iteration, after their call event is yielded. Unregistered tools are left for the caller to handle. A handler exception submits a generic failure result without exception text. Input and tool submissions each use an idempotency key across retries. The optional idempotency_key argument applies only to the input submission. N tool_handlersr;r'r)sessionsr# session_idr>r@rArn Mapping[str, ToolHandler] | Noner;r<r'r&r)r(r1rMcCd||_||_t||_t|pi|_t|||_t||d|_ t |_ d|_ d|_ d|_d|_dSN)r'r)F _sessions _session_idrK_inputdict _handlersr?_idempotency_keyr:_optionsrL_state_stream _iterator_entered_closedrUrorpr@rnr;r'r)r.r.r/rV    zAgentSessionStream.__init__r cCs|js|jr tdd|_|jj|jfi|j}|jdkr"td|jj j |jfi|j|_ z|jj j |jf|j g|jd|jWn tyQ|w||_|S)Nz.An AgentSessionStream can only be entered onceTidle`sessions.stream requires an idle session; use sessions.events.stream to follow an active sessioneventsr;rr RuntimeErrorruretrievervr{statusrHrstreamr}createrwrz BaseExceptionclose_iterater~rUsessionr.r.r/ __enter__s.     zAgentSessionStream.__enter__exc_typetype[BaseException] | NoneexcBaseException | Noneexc_tbTracebackType | NonecCs |dSNrrUrrrr.r.r/__exit__s zAgentSessionStream.__exit__cC|Srr.rTr.r.r/__iter__zAgentSessionStream.__iter__rcCs&|jrt|jdurtdt|jS)Nz#Use sessions.stream in a with block)r StopIterationr~rnextrTr.r.r/__next__s   zAgentSessionStream.__next__cCs|D]}qdSzBConsume the remaining events, including any registered tool calls.Nr.rU_r.r.r/ until_doneszAgentSessionStream.until_donecCs"d|_|jdur|jdSdSz?Close the event connection without cancelling the backend turn.TNrr}rrTr.r.r/rs zAgentSessionStream.closeIterator[AgentSessionEvent]c cs|jdusJzn|jD]f}|j|sq |j|}|r!||j|}|dur2|j|jnd}|durrrurrrvr{rr_sleeprUrr;delayerrorr.r.r/rs$ z!AgentSessionStream._submit_result)ror#rpr>r@rArnrqr;r<r'r&r)r(r1rMr1r rrrrrrr1rMr1rrj)r1rrr!r1rM)r*r+r,__doc__rrrVrrrrrrrrr.r.r.r/rlps       rlc@rk)8AsyncAgentSessionStreamaOAsync counterpart of AgentSessionStream; use with ``async with``. Requires an idle session with a single input writer. Handlers may return a value or an awaitable; synchronous handlers run inline, without a thread pool. Cancellation propagates and closes the event connection. It does not cancel the backend turn. Nrmror$rpr>r@rArn%Mapping[str, AsyncToolHandler] | Noner;r<r'r&r)r(r1rMcCrrrsrtrr.r.r/rVrz AsyncAgentSessionStream.__init__r cs|js|jr tdd|_|jj|jfi|jIdH}|jdkr&td|jj j |jfi|jIdH|_ z|jj j |jf|j g|jd|jIdHWnty^|IdHw||_|S)Nz3An AsyncAgentSessionStream can only be entered onceTrrrrrr.r.r/ __aenter__s0  "   z"AsyncAgentSessionStream.__aenter__rrrrrrcs|IdHdSrrrr.r.r/ __aexit__%sz!AsyncAgentSessionStream.__aexit__cCrrr.rTr.r.r/ __aiter__*rz!AsyncAgentSessionStream.__aiter__rcs.|jrt|jdurtd|jIdHS)Nz*Use sessions.stream in an async with block)rStopAsyncIterationr~r __anext__rTr.r.r/r-s  z!AsyncAgentSessionStream.__anext__cs|2z3dHW}q6dSrr.rr.r.r/r4sz"AsyncAgentSessionStream.until_donecs*d|_|jdur|jIdHdSdSrrrTr.r.r/r9s  zAsyncAgentSessionStream.close AsyncIterator[AgentSessionEvent]c Cs2|jdusJz|j2z3dHW}|j|sq |j|}|r(|IdH|j|}|dur9|j|jnd}|durCt |}|V|sK|j rVW|IdHdS|dur|durz|t |}t |ro|IdH}t||}Wn tyt|}Ynw||IdHq 6td|IdHwr)r}r|rcrdrriryrrrrrinspect isawaitablerrrrr)rUrWrdriroutputrr.r.r/r?s>          z AsyncAgentSessionStream._iteraterr!c stt}dD]A}z|jjj|jf|g|d|jIdHWdStyI}z|dus5t||ds6|j |IdHWYd}~qd}~wwdSrrrr.r.r/r]s&  z&AsyncAgentSessionStream._submit_result)ror$rpr>r@rArnrr;r<r'r&r)r(r1rMrrrrj)r1rr)r*r+r,rrrrVrrrrrrrrr.r.r.r/rs       r)r0r%r1r%)r;r<r9r&r1r<)r@rAr1r ); __future__rrcopyruuidrtypesrtypingrrrr r collectionsr typing_extensionsr r httpx2_toolsrrrr_typesrrrrrrr _streamingrr _exceptionsrtypes.beta.agent_session_eventr#types.beta.agent_function_call_itemr$types.beta.agent_session_input_paramr r!,types.beta.agent_session_input_message_paramr"'resources.beta.agents.sessions.sessionsr#r$r%r:r?rKr[rLrlrr.r.r.r/s:            *