Ji3ddlmZddlmZddlmZmZmZmZddl Z ddl m Z ddl m Z mZddlmZmZmZmZmZmZmZddlmZmZd d ZGd d Z d dd Zy)) annotations)asynccontextmanager)AnyAsyncGenerator AsyncIteratoroverloadN)Context)Event StartEvent)CancelHandlerResponse HandlerDataHandlersListResponseHealthResponseSendEventResponseStatusWorkflowsListResponse) EventEnvelopeEventEnvelopeWithMetadatac  |jy#tj$r}d|jjcxkrdkrn|jj dd}|j j}|j j}|jj}tj|d|jjd|d|d| |j |j|d}~wwxYw) zw Raise an HTTPStatusError with the first 200 characters of the response body for 400 and 500 level errors. iiXN z for z . Response: )requestresponse) raise_for_statushttpxHTTPStatusErrorr status_codetextrmethodurl reason_phrase)re body_previewrr rs i/var/www/html/BankruptcyAI-uat/bankruptcy-ai/venv/lib/python3.12/site-packages/workflows/client/client.py_raise_for_status_with_bodyr%!s !!#   !**(( .3 . ::??4C0LYY%%F))--C**00K''-q!9!9 :%xq\ZfYgh     sC8C C33C8c$eZdZeddZe ddZddd ddZeddZddZddZ d dd Z d dd Z d dd Z d dd Z dd Z d ddZddZ d d!dZy)"WorkflowClientcyN)self httpx_clients r$__init__zWorkflowClient.__init__7sc4K| t|d}t|tr |j }|xsd|xsid}|r||d<|j 4d{}|jd |d | d{}t|tj|jcdddd{S#t$r}td|d}~wwxYw#t$r}td|d}~wwxYw777G#1d{7swYyxYww) al Run the workflow and wait until completion. Args: start_event (Union[StartEvent, dict[str, Any], None]): start event class or dictionary representation (optional, defaults to None and get passed as an empty dictionary if not provided). context: Context or serialized representation of it (optional, defaults to None if not provided) handler_id (Optional[str]): Workflow handler identifier to continue from a previous completed run. Returns: HandlerData: Data representing the handler running the workflow (including result and metadata) NT)bare4Impossible to serialize the start event because of: 0Impossible to serialize the context because of: r4 start_eventcontext handler_id /workflows/z/runr;) _serialize_event Exceptionr2 isinstancer to_dictr7postr%r r:r; r+ workflow_namerHrFrGr" request_bodyr6rs r$ run_workflowzWorkflowClient.run_workflowosA$  " .{F gw ' Y!//+',"}"  )3L &##% ? ?#[[m_D1 )H ( 1--hmmo> ? ? ?  J1#N  Y #STUSV!WXX Y ? ? ? ? ?sD B?DC(DC=D D;C?<1D- D9D:D? CCCD C:'C55C::D?DDD D DDcRK| t|}t|tr |j }|xstt |xsid}|r||d<|j4d{}|jd|d|d{}t|tj|jcdddd{S#t$r}td|d}~wwxYw#t$r}td|d}~wwxYw777G#1d{7swYyxYww) aE Run the workflow in the background. Args: start_event (Union[StartEvent, dict[str, Any], None]): start event class or dictionary representation (optional, defaults to None and get passed as an empty dictionary if not provided). context: Context or serialized representation of it (optional, defaults to None if not provided) handler_id (Optional[str]): Workflow handler identifier to continue from a previous completed run. Returns: HandlerData: data representing the handler running the workflow. NrCrDrErHrIz /run-nowaitrJ) rKrLr2rMr rNr r7rOr%r r:r;rPs r$run_workflow_nowaitz"WorkflowClient.run_workflow_nowaitsG$  " .{; gw ' Y!//+'H*::<*H}"(  )3L &##% ? ?#[[m_K8|)H ( 1--hmmo> ? ? ?  J1#N  Y #STUSV!WXX Y ? ? ? ? ?sD' CD'C-9D'+D ,D'/D D 1D< D'D D' C*C%%C**D'- D 6DD  D'DD'D$D D$ D'c K|rdnd}d|}|j4d{} |jd|d||dddid 4d{}|jd k(r td |jd k(r" dddd{dddd{yt ||j 23d{}|j stj|} | 5777i7[786dddd{7n#1d{7swYnxYwnI#tj$rtd |tj$r} td| d} ~ wwxYwdddd{7y#1d{7swYyxYww)a  Stream events as they are produced by the workflow. Args: handler_id (str): ID of the handler running the workflow include_internal_events (bool): Include internal workflow events. Defaults to False. lock_timeout (float): Timeout (in seconds) for acquiring the lock to iterate over the events. Returns: AsyncGenerator[EventEnvelopeWithMetadata, None]: Generator for the events that are streamed as instances of `EventEnvelopeWithMetadata`. truefalse/events/NGET)sseinclude_internalacquire_timeout Connectionz keep-alive)paramsheaderstimeoutizHandler not foundz(Timeout waiting for events from handler z#Failed to connect to event stream: )r7streamrr2r% aiter_linesstriprmodel_validate_jsonrTimeoutException TimeoutError RequestErrorConnectionError) r+rHinclude_internal_events lock_timeout incl_interr r6rlineeventr"s r$get_workflow_eventsz"WorkflowClient.get_workflow_eventss" 7VG  %##% Q Q Q!==&,6+7 *<8 ) ((++s2()<==!--4!(( Q Q Q(09&.&:&:&<((d::<$=$Q$QRV$WE"'K3 Q(( Q,(&<)(((((2)) ">zlK%% Q%(KA3&OPP Q? Q Q Q Q Qs  F C(F E6 DC* D +D7 DC,D F C.F D3C27C0 8C2;D D(F *D,D.F 0C22D3 D>D?DD D D DE63E!EE!!E6$ F /E20F 6F<E? =FF cK t|}d|i}|r||d<|j4d{}|j d||d{}t |t j|jcdddd{S#t$r}td|d}~wwxYw77d7(#1d{7swYyxYww)a Send an event to the workflow. Args: handler_id (str): ID of the handler of the running workflow to send the event to event (Event | dict[str, Any] | str): Event to send, represented as an Event object, a dictionary or a serialized string. step (Optional[str]): Step to send the event to (optional, defaults to None) Returns: SendEventResponse: Confirmation of the send operation z,Error while serializing the provided event: NrosteprYrJ) rKrLr2r7rOr%rr:r;) r+rHrorrserialized_eventr"rRr6rs r$ send_eventzWorkflowClient.send_events" Q/?/F )01A'B #'L ##% E E#[[8J<)@|[TTH ' 1$33HMMOD  E E E  QKA3OP P Q  ET E E E EsxC  BC B/C B5 B1 1B5> C  B3 C  B,B''B,,C 1B53C 5C;B> <CC c@K|j|d{S7w)z6 Deprecated. Use get_handler instead. N) get_handler)r+rHs r$ get_resultzWorkflowClient.get_results%%j1111s c,K|j4d{}|jd||dd{}t|tj|j cdddd{S7a7E7 #1d{7swYyxYww)aq Get all the workflow handlers. Args: status (list[Status] | None): List of statuses (e.g. "running", "completed", etc. ) to filter by. Defaults to None. workflow_name (list[str] | None): List of workflow names to filter by. Defaults to None. Returns: HandlersListResponse: List of workflow handlers. Nz /handlers)statusrQr_)r7r9r%rr:r;)r+ryrQr6rs r$ get_handlerszWorkflowClient.get_handlers#s##% H H#ZZ$%2(H ( 1'66x}}G H H H H H H HsTBA9BA?A;1A?' B3A=4B;A?=B?BB B Bc(K|j4d{}|jd|d{}t|tj|j cdddd{S7_7E7 #1d{7swYyxYww)z Get a single workflow handler by identifier. Args: handler_id (str): ID of the handler associated with the workflow run Returns: HandlerData: Handler metadata persisted by the server. N /handlers/)r7r9r%r r:r;)r+rHr6rs r$rvzWorkflowClient.get_handler<s}##% ? ?#ZZ*ZL(ABBH ' 1--hmmo>  ? ? ?B ? ? ? ?sTBA7BA=A91A=% B1A;2B9A=;B=BB B Bc:K|j4d{}|jd|dd|rdndid{}t|tj|j cdddd{S7h7E7 #1d{7swYyxYww)a Stop and cancel a workflow run. Args: handler_id (str): ID of the handler associated with the workflow run purge (bool): Whether or not to delete the run also from the persistent storage. Defaults to false Nr}z/cancelpurgerWrXrz)r7rOr%r r:r;)r+rHrr6rs r$cancel_handlerzWorkflowClient.cancel_handlerLs##% I I#[[ZL05g>)H ( 1(77 H I I I I I I IsTBBB!BB1B. B:B;BBBB B BB)r,zhttpx.AsyncClient)r0str)r,zhttpx.AsyncClient | Noner0 str | None)returnz AsyncIterator[httpx.AsyncClient])rr)rr)NNN) rQrrHrrFz"StartEvent | dict[str, Any] | NonerGzContext | dict[str, Any] | Nonerr )F)rHrrkboolrlfloatrz/AsyncGenerator[EventEnvelopeWithMetadata, None]r))rHrroEvent | dict[str, Any]rrrrr)rHrrr )NN)ryzlist[Status] | NonerQzlist[str] | NonerrF)rHrrrrr )__name__ __module__ __qualname__rr-rr7r=r@rSrUrprtrwr{rvrr*r.r$r'r'6s ??    26# !/ ! ! B I""&:>37 +?+?+?8 +? 1 +?  +?`"&:>37 +?+?+?8 +? 1 +?  +?`). 4Q4Q"&4Q 4Q 9 4Qt EE&E E  E<2'+*.H#H(H  H2?".3II&*I Ir.r'ct|tr|S|r|jStj|jS)N)ro)rMdict model_dumpr from_event)rorBs r$rKrK`sI%    % %E 2 = = ?r.)rzhttpx.ResponserNoner)rorrBrrzdict[str, Any]) __future__r contextlibrtypingrrrrr workflowsr workflows.eventsr r workflows.protocolr r rrrrr&workflows.protocol.serializable_eventsrrr%r'rKr*r.r$rsl#* . *gIgIV 16  ! )-  r.