o i@sddlmZddlmZddlmZmZmZmZm Z m Z m Z m Z eddGdddZ Gdd d e ZGd d d e ZGd d d eZdS)) annotations) dataclass)Any AsyncIterable AwaitableMappingOptionalProtocolTypeUnionT)frozenc@s$eZdZUdZded< ded<dS)Contentz A container for a map of headers and a byte array of data. It is used by the SDK's Serializer interface implementations. Mapping[str, str]headersbytesdataN)__name__ __module__ __qualname____doc____annotations__rrZ/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/nexusrpc/_serializer.pyr s r c@s(eZdZdZdddZ ddddZdS) SerializerzX Serializer is used by the framework to serialize/deserialize input and output. valuerreturn"Union[Content, Awaitable[Content]]cCdS)z)Serialize encodes a value into a Content.Nr)selfrrrr serialize(zSerializer.serializeNcontentr as_typeOptional[Type[Any]]Union[Any, Awaitable[Any]]cCr)aDeserialize decodes a Content into a value. Args: content: The content to deserialize. as_type: The type to convert the result of deserialization into. Do not attempt type conversion if this is None. Nr)rr!r"rrr deserialize,s zSerializer.deserialize)rrrrN)r!r r"r#rr$)rrrrrr%rrrrr#s  rc@seZdZ dd ddZdS) LazyValueTNr"r#rr$cCsdSr&rrr"rrrconsume:r zLazyValueT.consumer&)r"r#rr$)rrrr)rrrrr'9sr'c@s*eZdZdZ ddd d ZddddZdS) LazyValuea A container for a value encoded in an underlying stream. It is used to stream inputs and outputs in the various client and server APIs. Example: .. code-block:: python # Creating a LazyValue from raw data lazy_input = LazyValue( serializer=my_serializer, headers={"content-type": "application/json"}, stream=async_data_stream ) # Using LazyValue with Handler.start_operation handler = nexusrpc.handler.Handler([my_service]) result = await handler.start_operation(ctx, lazy_input) # Consuming a LazyValue directly deserialized_data = await lazy_input.consume(as_type=MyInputType) N serializerrrrstreamOptional[AsyncIterable[bytes]]rNonecCs||_||_||_dS)a Args: serializer: The serializer to use for consuming the value. headers: Headers that include information on how to process the stream's content. Headers constructed by the framework always have lower case keys. User provided keys are treated case-insensitively. stream: Iterable that contains request or response data. None means empty data. N)r+rr,)rr+rr,rrr__init__Ws zLazyValue.__init__r"r#rcsx|jdur|jjt|jdd|dIdHSt|jts td|jjt|jddd|j2IdHd|dIdHS)zb Consume the underlying reader stream, deserializing via the embedded serializer. N)rr)r"z3When using consume, stream must be an AsyncIterablecsg|z3dHW}|q6Sr&r).0crrr wsz%LazyValue.consume..) r,r+r%r r isinstancer ValueErrorjoinr(rrrr)is   zLazyValue.consumer&)r+rrrr,r-rr.)r"r#rr)rrrrr/r)rrrrr*?s  r*N) __future__r dataclassesrtypingrrrrrr r r r rr'r*rrrrs (