j\ddlZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z d d lmZmZd d lmZd d lmZd d lmZd dlmZej2ej4dZGddZGddZy)N)AsyncClientWrapperSyncClientWrapper)RequestOptions)AttachmentPage)(CompleteMultipartUploadRequestEntityType)MultipartUploadPart)StartMultipartUploadResponse)AsyncRawAttachmentsClientRawAttachmentsClient)AttachmentListRequestEntityType)#DownloadAttachmentRequestEntityType)%StartMultipartUploadRequestEntityType)!UploadAttachmentRequestEntityType.ceZdZdefdZedefdZdddddede d ed ed e je d e je d e je defdZeeedddeded ede dede j$ede jede jede jed e je ddfdZeeedddeded ede dede j$ede jede jede jed e je ddfdZddddeded ededede jed e je de j.efdZeedddede ded ed ede jede jed e je defdZdddddeded ed e j<ee je j>fde jede jed e je ddfd!Z y)"AttachmentsClientclient_wrapperc&t||_yN)r)r _raw_clientselfrs y/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/rest_api/attachments/client.py__init__zAttachmentsClient.__init__s/~Nreturnc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- RawAttachmentsClient rrs rwith_raw_responsez#AttachmentsClient.with_raw_responserNpagesizerequest_options project_id entity_type entity_idpathr$r%r&c ^|jj|||||||}|jS)a Attachments list for entity Parameters ---------- project_id : str entity_type : AttachmentListRequestEntityType entity_id : str path : str page : typing.Optional[int] size : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AttachmentPage Attachment Resource Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.attachments.attachment_list(project_id='project_id', entity_type="trace", entity_id='entity_id', path='path', ) r'r(r)r*r$r%r&rattachment_listdata rr'r(r)r*r$r%r& _responses rr.z!AttachmentsClient.attachment_list$sAT$$44!#+5 ~~r project_name container_id mime_typer& file_name file_size upload_iduploaded_file_partsr3r4r5c d|jj||||||||| |  } | jS)aH Complete multipart attachment upload Parameters ---------- file_name : str entity_type : CompleteMultipartUploadRequestEntityType entity_id : str file_size : int upload_id : str uploaded_file_parts : typing.Sequence[MultipartUploadPart] project_name : typing.Optional[str] If null, the default project is used container_id : typing.Optional[str] mime_type : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import OpikApi from Opik import MultipartUploadPart client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.attachments.complete_multi_part_upload(file_name='file_name', entity_type="trace", entity_id='entity_id', file_size=1000000, upload_id='upload_id', uploaded_file_parts=[MultipartUploadPart(e_tag='e_tag', part_number=1, )], ) r6r(r)r7r8r9r3r4r5r&rcomplete_multi_part_uploadr/ rr6r(r)r7r8r9r3r4r5r&r1s rr=z,AttachmentsClient.complete_multi_part_uploadYsKh$$??# 3%%+@  ~~rc d|jj||||||||| |  } | jS)a. Delete attachments Parameters ---------- file_name : str entity_type : CompleteMultipartUploadRequestEntityType entity_id : str file_size : int upload_id : str uploaded_file_parts : typing.Sequence[MultipartUploadPart] project_name : typing.Optional[str] If null, the default project is used container_id : typing.Optional[str] mime_type : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import OpikApi from Opik import MultipartUploadPart client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.attachments.delete_attachments(file_name='file_name', entity_type="trace", entity_id='entity_id', file_size=1000000, upload_id='upload_id', uploaded_file_parts=[MultipartUploadPart(e_tag='e_tag', part_number=1, )], ) r;rdelete_attachmentsr/r>s rrAz$AttachmentsClient.delete_attachmentssJh$$77# 3%%+8  ~~rworkspace_namer&rCc #K|jj|||||||5}|jEd{dddy7 #1swYyxYww)a Download attachment from MinIO Parameters ---------- container_id : str entity_type : DownloadAttachmentRequestEntityType entity_id : str file_name : str mime_type : str workspace_name : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response. Returns ------- typing.Iterator[bytes] Attachment Resource Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.attachments.download_attachment(container_id='container_id', entity_type="trace", entity_id='entity_id', file_name='x', mime_type='x', ) r4r(r)r6r5rCr&Nrdownload_attachmentr/) rr4r(r)r6r5rCr&rs rrGz%AttachmentsClient.download_attachmentscT   1 1%#)+2 vv       s+#AAAA AAA Ar5r3r&num_of_file_partsc `|jj||||||||} | jS)a Start multipart attachment upload Parameters ---------- file_name : str num_of_file_parts : int entity_type : StartMultipartUploadRequestEntityType entity_id : str path : str mime_type : typing.Optional[str] project_name : typing.Optional[str] If null, the default project is used request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- StartMultipartUploadResponse MultipartUploadResponse Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.attachments.start_multi_part_upload(file_name='file_name', num_of_file_parts=1, entity_type="trace", entity_id='entity_id', path='path', ) r6rJr(r)r*r5r3r&rstart_multi_part_uploadr/ rr6rJr(r)r*r5r3r&r1s rrNz)AttachmentsClient.start_multi_part_uploadsD\$$<</#%+=  ~~rr3r5r&requestc ^|jj|||||||}|jS)a Upload attachment to MinIO Parameters ---------- file_name : str entity_type : UploadAttachmentRequestEntityType entity_id : str request : typing.Dict[str, typing.Optional[typing.Any]] project_name : typing.Optional[str] mime_type : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.attachments.upload_attachment(file_name='file_name', entity_type="trace", entity_id='entity_id', request={'key': 'value' }, ) r6r(r)rQr3r5r&rupload_attachmentr/ rr6r(r)rQr3r5r&r1s rrUz#AttachmentsClient.upload_attachmentLsAT$$66#%+7 ~~r)!__name__ __module__ __qualname__rrpropertyr r!strrtypingOptionalintrrr.OMITrSequencer r=rArIteratorbytesrGrr rNrDictAnyrUrrrrsO*;O #7  "&*%);?335 3  3  3ooc"3ooc"3 83 3|.2-1*.;?@@> @  @  @@$__-@A@ooc*@ooc*@??3'@ 8@ @V.2-1*.;?@@> @  @  @@$__-@A@ooc*@ooc*@??3'@ 8@ @T04;?339 3  3  33,3 83  3z+/-1;?88 8 ; 8  88??3'8ooc*8 88 &8B.2*.;?337 3  3 S&//&**"==> 3ooc*3??3'3 83 3rrceZdZdefdZedefdZdddddede d ed ed e je d e je d e je defdZeeedddeded ede dede j$ede jede jede jed e je ddfdZeeedddeded ede dede j$ede jede jede jed e je ddfdZddddeded ededede jed e je de j.efdZeedddede ded ed ede jede jed e je defdZdddddeded ed e j<ee je j>fde jede jed e je ddfd!Z y)"AsyncAttachmentsClientrc&t||_yr)r rrs rrzAsyncAttachmentsClient.__init__s4NSrrc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- AsyncRawAttachmentsClient rr s rr!z(AsyncAttachmentsClient.with_raw_responser"rNr#r'r(r)r*r$r%r&c zK|jj|||||||d{}|jS7w)a_ Attachments list for entity Parameters ---------- project_id : str entity_type : AttachmentListRequestEntityType entity_id : str path : str page : typing.Optional[int] size : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AttachmentPage Attachment Resource Examples -------- from Opik import AsyncOpikApi import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.attachments.attachment_list(project_id='project_id', entity_type="trace", entity_id='entity_id', path='path', ) asyncio.run(main()) r,Nr-r0s rr.z&AsyncAttachmentsClient.attachment_listsOZ**::!#+;  ~~ &;9;r2r6r7r8r9r3r4r5c K|jj||||||||| |  d{} | jS7w)a Complete multipart attachment upload Parameters ---------- file_name : str entity_type : CompleteMultipartUploadRequestEntityType entity_id : str file_size : int upload_id : str uploaded_file_parts : typing.Sequence[MultipartUploadPart] project_name : typing.Optional[str] If null, the default project is used container_id : typing.Optional[str] mime_type : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import AsyncOpikApi from Opik import MultipartUploadPart import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.attachments.complete_multi_part_upload(file_name='file_name', entity_type="trace", entity_id='entity_id', file_size=1000000, upload_id='upload_id', uploaded_file_parts=[MultipartUploadPart(e_tag='e_tag', part_number=1, )], ) asyncio.run(main()) r;Nr<r>s rr=z1AsyncAttachmentsClient.complete_multi_part_uploadsYn**EE# 3%%+F   ~~  )><>c K|jj||||||||| |  d{} | jS7w)a Delete attachments Parameters ---------- file_name : str entity_type : CompleteMultipartUploadRequestEntityType entity_id : str file_size : int upload_id : str uploaded_file_parts : typing.Sequence[MultipartUploadPart] project_name : typing.Optional[str] If null, the default project is used container_id : typing.Optional[str] mime_type : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import AsyncOpikApi from Opik import MultipartUploadPart import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.attachments.delete_attachments(file_name='file_name', entity_type="trace", entity_id='entity_id', file_size=1000000, upload_id='upload_id', uploaded_file_parts=[MultipartUploadPart(e_tag='e_tag', part_number=1, )], ) asyncio.run(main()) r;Nr@r>s rrAz)AsyncAttachmentsClient.delete_attachmentssXn**==# 3%%+>   ~~  rmrBrCc K|jj|||||||4d{}|j23d{} | 77 6dddd{7y#1d{7swYyxYww)a Download attachment from MinIO Parameters ---------- container_id : str entity_type : DownloadAttachmentRequestEntityType entity_id : str file_name : str mime_type : str workspace_name : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response. Returns ------- typing.AsyncIterator[bytes] Attachment Resource Examples -------- from Opik import AsyncOpikApi import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.attachments.download_attachment(container_id='container_id', entity_type="trace", entity_id='entity_id', file_name='x', mime_type='x', ) asyncio.run(main()) rENrF) rr4r(r)r6r5rCr&rHr/s rrGz*AsyncAttachmentsClient.download_attachmentSs{Z##77%#)+8  ff d   f     se'A5A A5 A A A A A  A5 A  A  A5AA5 A2&A) 'A2.A5rIrJc |K|jj||||||||d{} | jS7w)a Start multipart attachment upload Parameters ---------- file_name : str num_of_file_parts : int entity_type : StartMultipartUploadRequestEntityType entity_id : str path : str mime_type : typing.Optional[str] project_name : typing.Optional[str] If null, the default project is used request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- StartMultipartUploadResponse MultipartUploadResponse Examples -------- from Opik import AsyncOpikApi import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.attachments.start_multi_part_upload(file_name='file_name', num_of_file_parts=1, entity_type="trace", entity_id='entity_id', path='path', ) asyncio.run(main()) rLNrMrOs rrNz.AsyncAttachmentsClient.start_multi_part_uploadsSb**BB/#%+C   ~~  s '<:<rPrQc zK|jj|||||||d{}|jS7w)a Upload attachment to MinIO Parameters ---------- file_name : str entity_type : UploadAttachmentRequestEntityType entity_id : str request : typing.Dict[str, typing.Optional[typing.Any]] project_name : typing.Optional[str] mime_type : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import AsyncOpikApi import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.attachments.upload_attachment(file_name='file_name', entity_type="trace", entity_id='entity_id', request={'key': 'value' }, ) asyncio.run(main()) rSNrTrVs rrUz(AsyncAttachmentsClient.upload_attachmentsOZ**<<#%+=  ~~ rk)!rWrXrYrrrZr r!r[rr\r]r^rrr.r_rr`r r=rAr AsyncIteratorrbrGrr rNrrcrdrUrerrrgrgsT*<T #<  "&*%);?665 6  6  6ooc"6ooc"6 86 6B.2-1*.;?CC> C  C  CC$__-@ACooc*Cooc*C??3'C 8C C\.2-1*.;?CC> C  C  CC$__-@ACooc*Cooc*C??3'C 8C CZ04;?779 7  7  77,7 87   e $7B+/-1;?;; ; ; ;  ;;??3';ooc*; 8; &;H.2*.;?667 6  6 S&//&**"==> 6ooc*6??3'6 86 6rrg)r\core.client_wrapperrrcore.request_optionsrtypes.attachment_pager3types.complete_multipart_upload_request_entity_typertypes.multipart_upload_partr %types.start_multipart_upload_responser raw_clientr r )types.attachment_list_request_entity_typer-types.download_attachment_request_entity_typer0types.start_multipart_upload_request_entity_typer+types.upload_attachment_request_entity_typercastrdr_rrgrerrrs\G12j=PGV^cZv{{6::s#jjZ }}r