j2ddlZddlZddlZddlZddlZddlZddlmZmZddl m Z ddl m Z ddl mZejeZ ddej$d ed d ed ed edede j*fdZdedefdZdedeefdZdej$deefdZy)N)LiteralOptional) mime_type)messages) attachmentattachment_data entity_type)tracespan entity_id project_name url_overridedelete_after_uploadreturnc |j tdt|}tj|j dj d}t|jtrt|j}d|_ d}nltjj|jr |j}n6 tj|jd} t| }d|_ d}|j"} |} | tjj%|} |jrt'|} | | }d} nd} t)j*|| |||||| S#tj $r} td| d} ~ wwxYw)NzAttachment data cannot be Nonezutf-8FT)validatezWAttachment data must be bytes, an existing file path, or a valid base64-encoded string.) file_path file_namerr rrencoded_url_overrider)data ValueErrorguess_attachment_typebase64 b64encodeencodedecode isinstancebytes_write_file_like_to_temp_filecreate_temp_copyospathexists b64decodebinasciiErrorrbasename_try_create_temp_copyrCreateAttachmentMessage)r r rrrrmimetype base_url_pathr decoded_bytesershould_delete_after_upload tmp_file_paths /Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/api_objects/attachment/converters.pyattachment_to_messager3s}#9::$_5H$$\%8%8%ABII'RM/&&.1/2F2FG +0(" ,, -#((  ",,_-A-ADQM5mDI/4O ,"&   ))I!4GG$$Y/ ''-i8  $%I)- &). &  + +!*6  )~~ i  s5E''F : FF  file_likecltjdd} |j||jtj d|j  |j|j S#t$r}tjd|dd}~wwxYw#|jwxYw) a Writes a bytes-like file object to a temporary file on the filesystem. This function accepts a file-like object in the form of bytes and writes its contents to a temporary file. The temporary file is not deleted automatically on program termination. The path to the created temporary file is returned. Args: file_like: The bytes-like object containing the file content to be written to a temporary file. Returns: The full path to the created temporary file. wbF)modedeletez:Created temporary copy of file-like attachment to file: %sz5Failed to write file-like attachment to temp file: %sTexc_infoN) tempfileNamedTemporaryFilewriteflushLOGGERdebugname Exceptionerrorclose)r4 temp_filer/s r2r!r!Ks++eDI " H NN   >>  C     s$AA77 BBBB!!B3rctjj|\}}tjdd|} t |d5}t j||ddd|j|jtjd||j|jS#1swYVxYw#t$r+|jtjd|d YywxYw) ap Create a temporary copy of a file. This ensures the file remains available for upload even if the user deletes the original file. The temp file is created with delete=False so it persists until the upload manager processes and deletes it. Args: file_path: Path to the original file. Returns: Path to the temporary copy. r6F)r7r8suffixrbNz.Created temporary copy of attachment: %s -> %szZFailed to create temporary copy of attachment: %s. Opik will try to use the original file.Tr9)r#r$splitextr;r<openshutil copyfileobjr>rDr?r@rArBrC)r_ extensionrE original_files r2r*r*os77##I.LAy++eIVI )T "m   }i 8# <  NN ~~# "  h   s* CB4AC4B=9C1C43C4cf|j |jSd}|j tj|j}||St |j t r"tj|j }|St |j trtj}|S)N)file) content_typerrguess_mime_typerrstrr BINARY_MIME_TYPE)r r,s r2rrs##/+++H  ,,,/2K2KL/&&,,,/2F2FG O O((% 0-- O)F)rr'loggingr#rKr;typingrr file_uploadrmessage_processingrr getLogger__name__r? AttachmentrTboolr+r3r r!r*rrVr2ras  $$*   8 $!& 8**8)88 8  8  8%%8v!U!s!H"S"Xc]"J:+@+@Xc]rV