&`i\|ddlZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZddlZddlmZddlmZddlmZmZmZmZddlmZddlmZmZm Z m!Z!ddl"m#Z#m$Z$dd l%m&Z&dd l'm(Z(dd l)m*Z*dd l+m,Z,m-Z-m.Z.m/Z/m0Z0dd l1m2Z2m3Z3ddl4m5Z5ddl6m7Z7m8Z8ej9eZ:edZ;edZ<Gdde e;Z=Gdde e<Z>e8dGdde>e<Z?e8dGdde>e<Z@e8dGdde=e;ZAdS)N) Any AsyncIteratorDict GeneratorGenericIteratorOptionalTupleTypeVarUnioncast)serve)ObjectRefGenerator)OBJ_REF_NOT_SUPPORTED_ERRORDeploymentHandleSource DeploymentIDRequestMetadata)SERVE_LOGGER_NAME)CreateRouterCallablecreate_dynamic_handle_optionscreate_init_handle_options create_router)DynamicHandleOptionsBaseInitHandleOptionsBase) ReplicaResult)Router) ServeUsageTag)DEFAULTcalculate_remaining_timeoutget_random_stringinside_ray_client_contextis_running_in_asyncio_loop)RayServeExceptionRequestCancelledError)metrics) DeveloperAPI PublicAPITRceZdZddddddddededeedeedeedeed ee j d eefd Z e deded efd Z ededed efdZdefdZedefdZedefdZedefdZdZdefdZejf d*dZdeedeeefdeejj e!ffdZ"ejejejejejdde#eefde#eefde#eefde#eefd e#eefddf d!Z$d"eddfd#Z%d$Z&d%Z'defd&Z(edeeefdd'fd(Z)d)Z*dS)+_DeploymentHandleBaseN init_optionshandle_options_router_create_router_request_counter _handle_iddeployment_nameapp_namer-r.r/r0r1r2ct|||_||_|p t|_|p t |_|p||||j|_||_ |t|_ dS||_ dS)N)namer4) r deployment_idr-rr.r handle_id_create_request_counterrequest_counterr/rr0) selfr3r4r-r.r/r0r1r2s d/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/serve/handle.py__init__z_DeploymentHandleBase.__init__?s*RRR=I  =;==  $:'8':':/ 43O3O ot~4 4 *1  !"/D   "0D   r8c(|r |d|d|S|d|S)N#)r4r3r8s r<_gen_handle_tagz%_DeploymentHandleBase._gen_handle_tag^s:  4>>>>9>> >%33 33 3r>ctjddd||||||dS)Nserve_handle_request_counterzGThe number of handle.remote() calls that have been made on this handle.)handle deploymentroute application) descriptiontag_keys)r8)rErFrH)r%Counterset_default_tagsrB)clsr4r3r8s r<r9z-_DeploymentHandleBase._create_request_counteresn *'F      --o..'      r>returncF|jdS|jSNF)r/running_replicas_populatedr;s r<rQz0_DeploymentHandleBase.running_replicas_populatedzs# < 5|66888r>c|jjSN)r7r6rRs r<r3z%_DeploymentHandleBase.deployment_names!&&r>c|jjSrT)r7r4rRs r<r4z_DeploymentHandleBase.app_names!**r>c|jduSrT)r/rRs r<is_initializedz$_DeploymentHandleBase.is_initializeds|4''r>c |jtd|jdtd i|}||j|j||_||_td|jd|jdddi |jj tj kr1|j tkr#tjd dSdSdS) a<Initialize this handle with arguments. A handle can only be initialized once. A handle is implicitly initialized when `.options()` or `.remote()` is called. Therefore to initialize a handle with custom init options, you must do it before calling `.options()` or `.remote()`. NaiHandle has already been initialized. Note that a handle is implicitly initialized when you call `.options()` or `.remote()`. You either tried to call `._init()` twice or called `._init()` after calling `.options()` or `.remote()`. If you want to modify the init options, please do so before calling `.options()` or `.remote()`. This handle was initialized with .)r8r7r.zInitialized DeploymentHandle z for log_to_stderrF)extra1rA)r/ RuntimeErrorr-rr0r8r7loggerinfo_sourcerPROXY __class__DeploymentHandlerDEPLOYMENT_HANDLE_API_USEDrecord)r;kwargsr-s r<_initz_DeploymentHandleBase._inits < #= )-(9 === 2;;F;; **n,'+   ) VDN V VAS V V V"E*       %)?)E E E"222  4 ; ;C @ @ @ @ @ F E22r>c,|jdS|jjSrP)r-_run_router_in_separate_looprRs r<#_is_router_running_in_separate_loopz9_DeploymentHandleBase._is_router_running_in_separate_loops   $5 ==r>DeploymentHandle[T]c |ddurtrtd|jjdi|}|t jkr|||js|t|j |j |j ||j |j|j|jS)NstreamTzpStreaming DeploymentHandles are not currently supported when connected to a remote Ray cluster using Ray Client.)_prefer_local_routingr,rA)getr!r]r.copy_and_updaterVALUErgrWrcr3r4r-r/r0r:r8)r;rnrfnew_handle_optionss r<_optionsz_DeploymentHandleBase._optionss ::h  4 ' ',E,G,G 'F  AT0@JJ6JJ !GM 1 1 JJ-BJ C C C"  JJLLL  M*-L.!1~    r>argsrfcB|js|tjj|j|j}|j |j |j d|j td|j j|g|Ri||fS)N)rGrH)tagszRouter is not initialized)rWrgr_private default_implget_request_metadatar-r.r:incrGr4r/r]assign_request)r;rtrfmetadatas r<_remotez_DeploymentHandleBase._remotes "  JJLLL>.CC  t2     !'0 !   < :;; ;*t|*8EdEEEfEExOOr> method_namemultiplexed_model_idrmuse_new_handle_apirnrrrmrrnctrT)NotImplementedErrorr;rrrmrrns r<optionsz_DeploymentHandleBase.optionss "!r>r6c.||S)N)r)r)r;r6s r< __getattr__z!_DeploymentHandleBase.__getattr__s|||---r>c|jr_|j}|r|dStddSdS)NzSynchronously shutting down a router that's running in the same event loop can only be done best effort. Please use `shutdown_async` instead.)r/shutdownrjresultr^warningr;shutdown_futures r<rz_DeploymentHandleBase.shutdownsz < "l3355O7799 &&(((((0  r>cK|jrF|j}|jrtj|d{VdS|d{VdSdSrT)r/rrjasyncio wrap_futurers r<shutdown_asyncz$_DeploymentHandleBase.shutdown_asyncs < & %%'' 7 &)/:::::::::::%%%%%%%%%% & &r>c0|jjd|jdS)Nz (deployment='z'))rb__name__r3rRs r<__repr__z_DeploymentHandleBase.__repr__s!.)TTDz_DeploymentHandleBase[T]c|di|S)zcJ|j|j|jd}|jj|ffS)N)r3r4r.)r3r4r.rbr)r;serialized_constructor_argss r< __reduce__z _DeploymentHandleBase.__reduce__s6#3 "1' ' # ~*-H,JJJr>)rNrk)+r __module__ __qualname__strr rrrrr%rKr= staticmethodrB classmethodr9boolrQpropertyr3r4rWrgrjrrqrsr rr concurrentfuturesFuturerr}r rrrrrrrrAr>r<r+r+>s 9==A$(9=6:$(11111 45 1 !!9: 1&!1!!561#7?31SM1111>4#44444\4   -0 =@   [ (9D9999 ''''X'+#+++X+((((X($A$A$AL>T>>>> %,M     :PCjPS#XP z!(/9 : PPPP4,3=4;M'.}3:=6=m " " "3<( "$CL1 " dGm$ " "$-0 " %T7]3 "  " " " "..(=....   &&&U#UUUU$sCx.5O[KKKKKr>r+ceZdZ ddeejjeejefde de fdZ e de fdZe de fdZ dd eedefd Zdefd Zd Zede fdZd S)_DeploymentResponseBaseTreplica_result_futurerequest_metadatarjcLd|_||_d|_||_||_dSrP) _cancelled_replica_result_future_replica_result_request_metadatarj)r;rrrjs r<r=z _DeploymentResponseBase.__init__!s1 &;#8<2B3V000r>rNc|jjSrT)r request_idrRs r<rz"_DeploymentResponseBase.request_id/s%00r>c|jjSrT)r _by_referencerRs r< by_referencez$_DeploymentResponseBase.by_reference3s%33r>N _timeout_sc|j|jstd ttjjt|j}| ||_nP#tjj $rtddtjj $rt|j dwxYw|jS)ziSynchronously fetch the replica result. The result is cached in `self._replica_result`. NdSync methods should not be called from within an `asyncio` event loop. Use `await response` instead.)timeoutz!Timed out resolving to ObjectRef.)rrjr]r rrrrrr TimeoutErrorCancelledErrorr$r)r;r sync_futures r<_fetch_future_result_syncz1_DeploymentResponseBase._fetch_future_result_sync7s   '; ": G#&-m</ (3'9'9*'9'M'M$$%2 R R R"#FGGTQ%4 G G G+DO<<$F G##s A A**A B7cK|jc|jr%tj|jd{V|_n7t tjt|j}|d{V|_|jS)zgAsynchronously fetch replica result. The result is cached in `self._replica_result`.. N)rrjrrrr rr)r; async_futures r<_fetch_future_result_asyncz2_DeploymentResponseBase._fetch_future_result_asyncUs   '7 :.5-@/..(((((($$ $N=143N   .:'9'9'9'9'9'9$##r>c*|jrdSd|_|j|js"|jr|jdS |n#t $rYdSwxYw|jdS)aLAttempt to cancel the `DeploymentHandle` call. This is best effort. - If the request hasn't been assigned to a replica, the assignment will be cancelled. - If the request has been assigned to a replica, `ray.cancel` will be called on the object ref, attempting to cancel the request and any downstream requests it makes. If the request is successfully cancelled, subsequent operations on the ref will raise an exception: - If the request was cancelled before assignment, they'll raise `asyncio.CancelledError` (or a `concurrent.futures.CancelledError` for synchronous methods like `.result()`.). - If the request was cancelled after assignment, they'll raise `ray.exceptions.TaskCancelledError`. NT)rrcancelrjrrr$rRs r<rz_DeploymentResponseBase.cancells( ?  F #**,,,7 # .$++--- F   * * , , , ,$    FF  ##%%%%%sA)) A76A7c|jS)zWhether or not the request has been cancelled. This is `True` if `.cancel()` is called, but the request may actually have run to completion. )rrRs r< cancelledz!_DeploymentResponseBase.cancelleds r>)TrT)rrrr rrrrrrrr=rrrrr floatrrrr&rrAr>r<rr sG59 W W$   %m 4gn]6S S  W * W .2 W W W W1C111X14d444X4-1$$"5/$ $$$$<$-$$$$.'&'&'&R4\r>rstable) stabilityceZdZdZdeedeffdZdZdddde e d e defd Z e dejfd Ze dd e e d e dejfdZdS)DeploymentResponseaA future-like object wrapping the result of a unary deployment handle call. From inside a deployment, a `DeploymentResponse` can be awaited to retrieve the output of the call without blocking the asyncio event loop. From outside a deployment, `.result()` can be used to retrieve the output in a blocking manner. Example: .. code-block:: python from ray import serve from ray.serve.handle import DeploymentHandle @serve.deployment class Downstream: def say_hi(self, message: str) -> str: return f"Hello {message}!" @serve.deployment class Caller: def __init__(self, handle: DeploymentHandle): self._downstream_handle = handle async def __call__(self, message: str) -> str: # Inside a deployment: `await` the result to enable concurrency. response = self._downstream_handle.say_hi.remote(message) return await response app = Caller.bind(Downstream.bind()) handle: DeploymentHandle = serve.run(app) # Outside a deployment: call `.result()` to get output. response = handle.remote("world") assert response.result() == "Hello world!" A `DeploymentResponse` can be passed directly to another `DeploymentHandle` call without fetching the result to enable composing multiple deployments together. Example: .. code-block:: python from ray import serve from ray.serve.handle import DeploymentHandle @serve.deployment class Adder: def add(self, val: int) -> int: return val + 1 @serve.deployment class Caller: def __init__(self, handle: DeploymentHandle): self._adder_handle = handle async def __call__(self, start: int) -> int: return await self._adder_handle.add.remote( # Pass the response directly to another handle call without awaiting. self._adder_handle.add.remote(start) ) app = Caller.bind(Adder.bind()) handle: DeploymentHandle = serve.run(app) assert handle.remote(0).result() == 2 rNNc#4K |Ed{V}|Ed{V}|S#tj$r*|jrt |jdtjdwxYw)z6Yields the final result of the deployment handle call.N)r __await__ get_asyncrrrr$r)r;replica_resultrs r<rzDeploymentResponse.__await__s 7(,(G(G(I(I(S(S(U(UUUUUUUN . 8 8 : : D D F FFFFFFFFM% 7 7 7 7+DO<<$F,$6  7s AA9Bc td)Na`DeploymentResponse` is not serializable. If you are passing the `DeploymentResponse` in a nested object (e.g. a list or dictionary) to a downstream deployment handle call, that is no longer supported. Please only pass `DeploymentResponse` objects as top level arguments.)r#rRs r<rzDeploymentResponse.__reduce__s M   r>F timeout_s_skip_asyncio_checkrrc|strtdtj}||}t ||tj}||S)aUFetch the result of the handle call synchronously. This should *not* be used from within a deployment as it runs in an asyncio event loop. For model composition, `await` the response instead. If `timeout_s` is provided and the result is not available before the timeout, a `TimeoutError` is raised. rr start_time_s curr_time_s)r"r]timerrro)r;rrrrremaining_timeout_ss r<rzDeploymentResponse.results# 'A'C'C 6  y{{ 77 BB9l    !!"5666r>cKtjd|jjst |d{V}|d{VS)aAdvanced API to convert the response to a Ray `ObjectRef`. This is used to pass the output of a `DeploymentHandle` call to a Ray task or actor method call. This method is `async def` because it will block until the handle call has been assigned to a replica. If there are many requests in flight and all replicas' queues are full, this may be a slow operation. r\N)r(DEPLOYMENT_HANDLE_TO_OBJECT_REF_API_USEDrerrrrto_object_ref_asyncr;rs r<_to_object_refz!DeploymentResponse._to_object_refsz >EEcJJJ%3 .- -#>>@@@@@@@@#77999999999r>r_allow_running_in_asyncio_loopcftjd|jjst |st rtdtj}| |}t||tj}| |S)aAdvanced API to convert the response to a Ray `ObjectRef`. This is used to pass the output of a `DeploymentHandle` call to a Ray task or actor method call. This method is a *blocking* call because it will block until the handle call has been assigned to a replica. If there are many requests in flight and all replicas' queues are full, this may be a slow operation. From inside a deployment, `_to_object_ref` should be used instead to avoid blocking the asyncio event loop. r\uSync methods should not be called from within an `asyncio` event loop. Use `await response._to_object_ref()` instead.r)r) rrrerrrr"r]rrr to_object_ref)r;rrrrrs r<_to_object_ref_syncz&DeploymentResponse._to_object_ref_sync*s& >EEcJJJ%3 .- -- 2L2N2N G  y{{ 77 CC: %     ++6I+JJJr>rP)rrr__doc__rrr)rrr rrrr&ray ObjectRefrrrAr>r<rrs BBH 79S$\2 7 7 7 7   &*$) 777E?7" 7 7777::cm:::\:&'+/4'K'KUO'K)-'K  'K'K'K\'K'K'Kr>rceZdZdZdZdeefdZdefdZde efdZ defdZ dd d d e e d edefd ZedefdZe dde e dedefdZdS)DeploymentResponseGeneratora5A future-like object wrapping the result of a streaming deployment handle call. This is returned when using `handle.options(stream=True)` and calling a generator deployment method. `DeploymentResponseGenerator` is both a synchronous and asynchronous iterator. When iterating over results from inside a deployment, `async for` should be used to avoid blocking the asyncio event loop. When iterating over results from outside a deployment, use a standard `for` loop. Example: .. code-block:: python from typing import AsyncGenerator, Generator from ray import serve from ray.serve.handle import DeploymentHandle @serve.deployment class Streamer: def generate_numbers(self, limit: int) -> Generator[int]: for i in range(limit): yield i @serve.deployment class Caller: def __init__(self, handle: DeploymentHandle): # Set `stream=True` on the handle to enable streaming calls. self._streaming_handle = handle.options(stream=True) async def __call__(self, limit: int) -> AsyncIterator[int]: gen: DeploymentResponseGenerator = ( self._streaming_handle.generate_numbers.remote(limit) ) # Inside a deployment: use `async for` to enable concurrency. async for i in gen: yield i app = Caller.bind(Streamer.bind()) handle: DeploymentHandle = serve.run(app) # Outside a deployment: use a standard `for` loop. gen: DeploymentResponseGenerator = handle.options(stream=True).remote(10) assert [i for i in gen] == list(range(10)) A `DeploymentResponseGenerator` *cannot* currently be passed to another `DeploymentHandle` call. c td)Nzr`DeploymentResponseGenerator` cannot be awaited directly. Use `async for` or `await response.__anext__() instead`. TypeErrorrRs r<rz%DeploymentResponseGenerator.__await__s 7   r>rNc|SrTrArRs r< __aiter__z%DeploymentResponseGenerator.__aiter__ r>cK |d{V}|d{VS#tj$r*|jrt |jdtjdwxYwrT)r __anext__rrrr$rrs r<rz%DeploymentResponseGenerator.__anext__s 7#'#B#B#D#DDDDDDDN'1133333333 3% 7 7 7 7+DO<<$F,$6  7s 389A1c|SrTrArRs r<__iter__z$DeploymentResponseGenerator.__iter__rr>ctrtd|}|S)NzSync methods should not be called from within an `asyncio` event loop. Use `async for` or `await response.__anext__()` instead.)r"r]r__next__rs r<rz$DeploymentResponseGenerator.__next__sM % ' ' Q  7799&&(((r>NFrrrc td)a Not supported on `DeploymentResponseGenerator`. This method exists only for API parity with `DeploymentResponse.result()` to aid static typing. A `DeploymentResponseGenerator` is returned when using streaming handles (e.g., `handle.options(stream=True)`). To consume results, iterate over the generator instead of calling `.result()`: - Outside a deployment: use a standard `for` loop - Inside a deployment: use `async for` Always raises `TypeError`. z`DeploymentResponseGenerator` doesn't support `.result()`. Use iteration instead: `for item in response` (outside deployments) or `async for item in response` (inside deployments).r)r;rrs r<rz"DeploymentResponseGenerator.results$ D   r>cKtjd|jjst |d{V}|S)a3Advanced API to convert the generator to a Ray `ObjectRefGenerator`. This method is `async def` because it will block until the handle call has been assigned to a replica. If there are many requests in flight and all replicas' queues are full, this may be a slow operation. r\N)rrrerrrrto_object_ref_genrs r<_to_object_ref_genz.DeploymentResponseGenerator._to_object_ref_gensh >EEcJJJ%3 .- -#>>@@@@@@@@//111r>rrctjd|jjst |st rtd||}| S)aAdvanced API to convert the generator to a Ray `ObjectRefGenerator`. This method is a *blocking* call because it will block until the handle call has been assigned to a replica. If there are many requests in flight and all replicas' queues are full, this may be a slow operation. From inside a deployment, `_to_object_ref_gen` should be used instead to avoid blocking the asyncio event loop. r\r) rrrerrrr"r]rr)r;rrrs r<_to_object_ref_gen_syncz3DeploymentResponseGenerator._to_object_ref_gen_syncs >EEcJJJ%3 .- -- 2L2N2N G  77 CC//111r>rP)rrrrrrr)rrrrrr rrrrr&rrrrAr>r<rrUs`33j   =+77777(1+)!))))&*$)    E? "      02*<222\2 '+/422UO2)-2  222\222r>rc eZdZdZejejejejejddeeefdeeefdeeefdeeefdeeefdd f d Z dee e e e ffd Z d S) rcalA handle used to make requests to a deployment at runtime. This is primarily used to compose multiple deployments within a single application. It can also be used to make calls to the ingress deployment of an application (e.g., for programmatic testing). Example: .. code-block:: python import ray from ray import serve from ray.serve.handle import DeploymentHandle, DeploymentResponse @serve.deployment class Downstream: def say_hi(self, message: str): return f"Hello {message}!" self._message = message @serve.deployment class Ingress: def __init__(self, handle: DeploymentHandle): self._downstream_handle = handle async def __call__(self, name: str) -> str: response = self._downstream_handle.say_hi.remote(name) return await response app = Ingress.bind(Downstream.bind()) handle: DeploymentHandle = serve.run(app) response = handle.remote("world") assert response.result() == "Hello world!" r~rrrmrrnrNrkc|tjurtjd|tjurtjd|||||S)a(Set options for this handle and return an updated copy of it. Example: .. code-block:: python response: DeploymentResponse = handle.options( method_name="other_method", multiplexed_model_id="model:v1", ).remote() ziSetting `use_new_handle_api` no longer has any effect. This argument will be removed in a future version.z^Modifying `_prefer_local_routing` with `options()` is deprecated. Please use `init()` instead.)rrrmrn)rrqwarningswarnrsrs r<rzDeploymentHandle.optionss|( W] 2 2 ME    ! 5 5 M;    }}#!5"7    r>c|||\}}|jjr$t|||St |||S)aIssue a remote call to a method of the deployment. By default, the result is a `DeploymentResponse` that can be awaited to fetch the result of the call or passed to another `.remote()` call to compose multiple deployments. If `handle.options(stream=True)` is set and a generator method is called, this returns a `DeploymentResponseGenerator` instead. Example: .. code-block:: python # Fetch the result directly. response = handle.remote() result = await response # Pass the result to another handle call. composed_response = handle2.remote(handle1.remote()) composed_result = await composed_response Args: *args: Positional arguments to be serialized and passed to the remote method call. **kwargs: Keyword arguments to be serialized and passed to the remote method call. )rj)r}r.rmrrjr)r;rtrffuturers r<remotezDeploymentHandle.remoteAs>$(<<f#=#=    % . 484\4\4^4^  & 484\4\4^4^ r>N)rrrrrrqr rrrrrrrrAr>r<rcrcs""N,3=4;M'.}3:=6=m% % % 3<(% $CL1 % dGm$ % "$-0 %  %T7]3%  % % % % N+ !#&(CC(HH I++++++r>rc)Brconcurrent.futuresrloggingrrtypingrrrrrrr r r r r rr ray._rayletrray.serve._private.commonrrrrray.serve._private.constantsrray.serve._private.default_implrrrr!ray.serve._private.handle_optionsrr!ray.serve._private.replica_resultrray.serve._private.routerrray.serve._private.usagerray.serve._private.utilsrrr r!r"ray.serve.exceptionsr#r$ray.utilr%ray.util.annotationsr&r' getLoggerr^r(r)r+rrrrcrAr>r<r s                            ****** ;::::: <;;;;;,,,,,,222222JIIIIIII88888888  , - - GCLL GCLL_K_K_K_K_KGAJ_K_K_KD|||||gaj|||~ XrKrKrKrKrK03rKrKrKj X[2[2[2[2[2"9!"<[2[2[2| Xwwwww,Q/wwwwwr>