jX UddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z ddlmZddlmZmZddlmZddlmZddlmZdd lmZd d lmZej<eZ ejBZ"ee#d <Gd dZ$y)N)s3_httpx_client)IteratorListLiteralOptional) TypeAlias) file_uploaderupload_options)client) attachment)core) url_helpers) rest_helpersRESTAttachmentDetailsceZdZdZdej dededejddf dZ d ed ed e d de e fd Z d ed e dd edededeef dZ dd ed e dd ededeedeeddfdZy)AttachmentClienta Client for interacting with attachment-related operations. This client provides methods to retrieve attachment lists, download attachments, and upload attachments for traces and spans. The AttachmentClient supports: - Listing attachments associated with traces or spans - Downloading attachment content as a byte stream - Uploading files as attachments to traces or spans All operations are performed within the context of a specific project and require the project name to be provided. rest_client url_overrideworkspace_namerest_httpx_clientreturnNc<||_||_||_||_y)a Initialize the AttachmentClient. It is typically created via ``Opik.get_attachment_client()`` rather than being instantiated directly. Parameters: rest_client: The REST API client instance for making backend requests. url_override: The base URL for the Opik server. workspace_name: The workspace name used for download operations. rest_httpx_client: The httpx client instance to use for making file uploads. Returns: None N) _rest_client _url_override_workspace_name_rest_httpx_client)selfrrrrs {/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/api_objects/attachment/client.py__init__zAttachmentClient.__init__'s$*()-"3 project_name entity_id entity_type)spantracec.tj|j|}tj|j j djd}|jjj||||}|jxsgS)a Get a list of attachments for a specific entity (trace or span). Parameters: project_name: The name of the project containing the entity. entity_id: The ID of the trace or span to retrieve attachments for. entity_type: The type of entity ("trace" or "span"). Returns: List[RESTAttachmentDetails]: List of attachment detail objects containing metadata about each attachment. utf-8) project_idr$r#path) rresolve_project_id_by_namerbase64 b64encoderencodedecode attachmentsattachment_listcontent)rr"r#r$r)url_override_pathresponses rget_attachment_listz$AttachmentClient.get_attachment_listAs""<<   | #,,T-?-?-F-Fw-OPWW  $$00@@!#" A %2%r!)r&r% file_name mime_typec#K|j|||}d}|D]%}|j|k(s|j|k(s#|}n|td||jtd|t j |jrtjn |j} | jd|j5} d| jcxkrdkr&nn#| jD]} |  dddy| j| j} t#j$| jt'| j(| #tjj $r@t#j$| jt'| j(| j*wxYw#1swYyxYww) a Download an attachment as a stream of bytes. Parameters: project_name: The name of the project containing the entity. entity_type: The type of entity ("trace" or "span"). entity_id: The ID of the trace or span containing the attachment. file_name: The name of the file to download. mime_type: The MIME type of the file. Returns: Iterator[bytes]: Iterator yielding bytes of the attachment content. )r"r$r#NzAttachment not found: z*No download URL available for attachment: GETi,) status_codeheadersbody)r5r6r7 ValueErrorlinkris_aws_presigned_urlr get_cachedrstreamr; iter_bytesreadjsondecoderJSONDecodeError rest_api_coreApiErrordictr<text) rr"r$r#r6r7attachments_detailsattachment_to_downloadattachment_detailshttpx_client_uploadr4chunk response_jsons rdownload_attachmentz$AttachmentClient.download_attachmentbs*#66%#7  CG"5 ",, 9&00I=);& #6 " )5i[AB B ! & & .I)UV V//0F0K0KL  & & (((  ! ' '/E/J/J Kx (..44!)!4!4!6# "7 L K   (   (($00X--."  <<// #,, ( 4 4 !1!12! L KsG+GGB GF: 1E< G E%5F:AF77F::G?G file_pathc  tjj|std||tjj |}|t j |\}}tjj|}tj|jjdjd} tj|||||||| d } tj | |j"|j$y)a Upload an attachment for a specific entity (trace or span). Parameters: project_name: The name of the project containing the entity. entity_type: The type of entity ("trace" or "span"). entity_id: The ID of the trace or span to attach the file to. file_path: The path to the file to upload on the local filesystem. file_name: The name to assign to the uploaded file. If not provided, uses the basename of file_path. mime_type: The MIME type of the file. If not provided, attempts to automatically detect based on the file extension. Returns: None zFile not found: Nr(F) rSr6 file_sizer7r$r#r"encoded_url_overridedelete_after_upload)r rupload_httpx_client)osr*existsFileNotFoundErrorbasename mimetypes guess_typegetsizer,r-rr.r/r FileUploadOptionsr upload_attachmentrr) rr"r$r#rSr6r7_rUrV upload_optss rraz"AttachmentClient.upload_attachments.ww~~i(#&6yk$BC C  ((3I  $// :LIqGGOOI. %//    % %g . &/ %66#%!5 %   ''&)) $ 7 7 r!)NN)__name__ __module__ __qualname____doc__rest_api_clientOpikApistrhttpxClientr rrrr5rbytesrRrrar!rrrs+ 4$,,44 4 !<< 4  44&&&_- & # $ &BBB_-B B  B  B %BT$(#'5 5 _-5  5  5 C= 5 C=5  5 r!r)%r,loggingrYr]rk json.decoderrEopikrtypingrrrrtyping_extensionsropik.file_uploadr r opik.rest_apir rhopik.rest_api.typesr rest_api_attachmentr rHrr getLoggerrdLOGGER Attachmentr__annotations__rrnr!rr}sd    44':3A/   8 $#6#A#AyAD D r!