j1ddlZddlZddlmZmZddlmZddlm Z m Z ddl m Z ddl mZdd lmZejejd ZGd d ZGd dZdS)N)AsyncClientWrapperSyncClientWrapper)RequestOptions)AsyncRawMediaClientRawMediaClient)GetMediaResponse)GetMediaUploadUrlResponse)MediaContentType.c<eZdZdefdZedefdZdddede j e de fd Z eedd ded ejd ed e j ede j ede j e ddfdZedddededededede j ede j e defdZdS) MediaClientclient_wrapperc0t||_dSN)r)r _raw_clientselfrs c/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/langfuse/api/media/client.py__init__zMediaClient.__init__s)HHHreturnc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- RawMediaClient rrs rwith_raw_responsezMediaClient.with_raw_response rNrequest_optionsmedia_idrcH|j||}|jS)a< Get a media record Parameters ---------- media_id : str The unique langfuse identifier of a media record request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- GetMediaResponse Examples -------- from langfuse import LangfuseAPI client = LangfuseAPI( x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME", x_langfuse_sdk_version="YOUR_X_LANGFUSE_SDK_VERSION", x_langfuse_public_key="YOUR_X_LANGFUSE_PUBLIC_KEY", username="YOUR_USERNAME", password="YOUR_PASSWORD", base_url="https://yourhost.com/path/to/api", ) client.media.get( media_id="mediaId", ) rrgetdatarr r _responses rr#zMediaClient.get s(D$((?(SS ~rupload_http_errorupload_time_msr uploaded_atupload_http_statusr(r)cP|j||||||}|jS)as Patch a media record Parameters ---------- media_id : str The unique langfuse identifier of a media record uploaded_at : dt.datetime The date and time when the media record was uploaded upload_http_status : int The HTTP status code of the upload upload_http_error : typing.Optional[str] The HTTP error message of the upload upload_time_ms : typing.Optional[int] The time in milliseconds it took to upload the media record request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- import datetime from langfuse import LangfuseAPI client = LangfuseAPI( x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME", x_langfuse_sdk_version="YOUR_X_LANGFUSE_SDK_VERSION", x_langfuse_public_key="YOUR_X_LANGFUSE_PUBLIC_KEY", username="YOUR_USERNAME", password="YOUR_PASSWORD", base_url="https://yourhost.com/path/to/api", ) client.media.patch( media_id="mediaId", uploaded_at=datetime.datetime.fromisoformat( "2024-01-15 09:30:00+00:00", ), upload_http_status=1, ) r*r+r(r)rrpatchr$rr r*r+r(r)rr&s rr/zMediaClient.patchEs>v$** #1/)+ +  ~robservation_idrtrace_id content_typecontent_length sha256hashfieldr2c R|j|||||||}|jS)aD Get a presigned upload URL for a media record Parameters ---------- trace_id : str The trace ID associated with the media record content_type : MediaContentType content_length : int The size of the media record in bytes sha256hash : str The SHA-256 hash of the media record field : str The trace / observation field the media record is associated with. This can be one of `input`, `output`, `metadata` observation_id : typing.Optional[str] The observation ID associated with the media record. If the media record is associated directly with a trace, this will be null. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- GetMediaUploadUrlResponse Examples -------- from langfuse import LangfuseAPI from langfuse.media import MediaContentType client = LangfuseAPI( x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME", x_langfuse_sdk_version="YOUR_X_LANGFUSE_SDK_VERSION", x_langfuse_public_key="YOUR_X_LANGFUSE_PUBLIC_KEY", username="YOUR_USERNAME", password="YOUR_PASSWORD", base_url="https://yourhost.com/path/to/api", ) client.media.get_upload_url( trace_id="traceId", content_type=MediaContentType.IMAGE_PNG, content_length=1, sha256hash="sha256Hash", field="field", ) r3r4r5r6r7r2rrget_upload_urlr$ rr3r4r5r6r7r2rr&s rr;zMediaClient.get_upload_urlsAz$33%)!)+4  ~r)__name__ __module__ __qualname__rrpropertyr rstrtypingOptionalrr r#OMITdtdatetimeintr/r r r;rrrrsI*;IIII >   X TX####171P# ####V37/3;?CCCC[ C  C "?3/ C,C 8C CCCCZ04;?FFFF' F  F  FF,F 8F #FFFFFFrrc<eZdZdefdZedefdZdddede j e de fd Z eedd ded ejd ed e j ede j ede j e ddfdZedddededededede j ede j e defdZdS)AsyncMediaClientrc0t||_dSr)rrrs rrzAsyncMediaClient.__init__s.nMMMrrc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- AsyncRawMediaClient rrs rrz"AsyncMediaClient.with_raw_responserrNrr rcXK|j||d{V}|jS)a Get a media record Parameters ---------- media_id : str The unique langfuse identifier of a media record request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- GetMediaResponse Examples -------- import asyncio from langfuse import AsyncLangfuseAPI client = AsyncLangfuseAPI( x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME", x_langfuse_sdk_version="YOUR_X_LANGFUSE_SDK_VERSION", x_langfuse_public_key="YOUR_X_LANGFUSE_PUBLIC_KEY", username="YOUR_USERNAME", password="YOUR_PASSWORD", base_url="https://yourhost.com/path/to/api", ) async def main() -> None: await client.media.get( media_id="mediaId", ) asyncio.run(main()) rNr"r%s rr#zAsyncMediaClient.getsST*.. o/        ~rr'r*r+r(r)c`K|j||||||d{V}|jS)a Patch a media record Parameters ---------- media_id : str The unique langfuse identifier of a media record uploaded_at : dt.datetime The date and time when the media record was uploaded upload_http_status : int The HTTP status code of the upload upload_http_error : typing.Optional[str] The HTTP error message of the upload upload_time_ms : typing.Optional[int] The time in milliseconds it took to upload the media record request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- import asyncio import datetime from langfuse import AsyncLangfuseAPI client = AsyncLangfuseAPI( x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME", x_langfuse_sdk_version="YOUR_X_LANGFUSE_SDK_VERSION", x_langfuse_public_key="YOUR_X_LANGFUSE_PUBLIC_KEY", username="YOUR_USERNAME", password="YOUR_PASSWORD", base_url="https://yourhost.com/path/to/api", ) async def main() -> None: await client.media.patch( media_id="mediaId", uploaded_at=datetime.datetime.fromisoformat( "2024-01-15 09:30:00+00:00", ), upload_http_status=1, ) asyncio.run(main()) r-Nr.r0s rr/zAsyncMediaClient.patchs`D*00 #1/)+ 1        ~rr1r3r4r5r6r7r2c bK|j|||||||d{V}|jS)a Get a presigned upload URL for a media record Parameters ---------- trace_id : str The trace ID associated with the media record content_type : MediaContentType content_length : int The size of the media record in bytes sha256hash : str The SHA-256 hash of the media record field : str The trace / observation field the media record is associated with. This can be one of `input`, `output`, `metadata` observation_id : typing.Optional[str] The observation ID associated with the media record. If the media record is associated directly with a trace, this will be null. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- GetMediaUploadUrlResponse Examples -------- import asyncio from langfuse import AsyncLangfuseAPI from langfuse.media import MediaContentType client = AsyncLangfuseAPI( x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME", x_langfuse_sdk_version="YOUR_X_LANGFUSE_SDK_VERSION", x_langfuse_public_key="YOUR_X_LANGFUSE_PUBLIC_KEY", username="YOUR_USERNAME", password="YOUR_PASSWORD", base_url="https://yourhost.com/path/to/api", ) async def main() -> None: await client.media.get_upload_url( trace_id="traceId", content_type=MediaContentType.IMAGE_PNG, content_length=1, sha256hash="sha256Hash", field="field", ) asyncio.run(main()) r9Nr:r<s rr;zAsyncMediaClient.get_upload_url]scJ*99%)!)+:        ~r)r=r>r?rrr@rrrArBrCrr r#rDrErFrGr/r r r;rHrrrJrJsN*<NNNN #6   X TX----171P- ----j37/3;?JJJJ[ J  J "?3/ J,J 8J JJJJh04;?NNNN' N  N  NN,N 8N #NNNNNNrrJ)rFrErBcore.client_wrapperrrcore.request_optionsr raw_clientrr types.get_media_responser #types.get_media_upload_url_responser types.media_content_typer castAnyrDrrJrHrrrXs GGGGGGGG111111;;;;;;;;666666JJJJJJ666666v{6:s##DXXXXXXXXXXr