Ji|ddlmZddlmZddlZddlmZddlmZddl m Z m Z ddlm Z ddl mZmZGd d Zy) )BytesIO)BinaryION)Path)AsyncLlamaCloud)File FileCreate)Optional) SourceText FileInputc eZdZdZ ddedeedeedefdZded e fd Z ded e fd Z dd ed eed e fdZ de d ed e fdZded eded e fdZ dded eed e fdZy) FileClienta Higher-level client for interacting with the LlamaCloud Files API. Optionally uses presigned URLs for uploads. Args: client: The LlamaCloud client to use. project_id: The project ID to use. organization_id: The organization ID to use. use_presigned_url: Whether to use presigned URLs for uploads (set to False when uploading to BYOC deployments). Nclient project_idorganization_iduse_presigned_urlc<||_||_||_||_yN)rrrr)selfrrrrs s/var/www/html/BankruptcyAI-uat/bankruptcy-ai/venv/lib/python3.12/site-packages/llama_cloud_services/files/client.py__init__zFileClient.__init__s" $.!2file_idreturncK|jjj||j|jd{S7wNrr)rfilesget_filerr)rrs rrzFileClient.get_file#sC[[&&// AUAU0    sAA AA cXK|jjj||j|jd{}|jj j }|j|jd{}|j|jS7c7"wr) rrread_file_contentrr_client_wrapper httpx_clientgeturlraise_for_statuscontent)rr presigned_urlr"responses rr zFileClient.read_file_content(s"kk//AA  00B  {{22?? %))-*;*;<<!!# =s%AB*B&AB*B(!B*(B* file_pathexternal_file_idcK|xs|}tjj|}t|d5}|j |||d{cdddS7 #1swYyxYww)Nrb)ospathgetsizeopen upload_buffer)rr)r* file_sizefiles r upload_filezFileClient.upload_file3si,8yGGOOI. )T " Od++D2BINN O ON O Os.2A'A A A A'AA$ A'byteschK|jt||t|d{S7wr)r1rlen)rr5r*s r upload_byteszFileClient.upload_bytes;s*''8H#e*UUUUs )202bufferr2c |K|jrDt|ddr4tjj t t|d|}n|}|j jj|j|jt|||d{}|j jj}|j|j|j!d{}|j#|j jj%|j&|j|jd{St|dds t)|d||j jj+|||j|jd{S777h7 w)Nname)r;r*r2)rrrequest)datar)r4r*rr)rgetattrr-r.basenamestrrrgenerate_presigned_urlrrrr!r"putr$readr%rrsetattrr4)rr9r*r2r;r'r"upload_responses rr1zFileClient.upload_buffer>s  ! !vvt,ww''GFFDT,U(VW'"&++"3"3"J"J?? $ 4 4"%5'#K#M ;;66CCL$0$4$4!![[]%5%O  , , .**33%%?? $ 4 44 6640(89**66"!1?? $ 4 4 7 3 sKBF<F3AF<1F62AF<F8AF<.F:/F<6F<8F<:F< file_inputcKt|tr|St|tr|j#|j |jd{S|j G|j j d}|j||xs|jxsdd{St|jttfr;|jt|j|xs |jd{St|jtr6|j|j|xs|jxsdd{St|jdrg|jj}t|tr|j d}|j||xs|jxsdd{St!dt#|jt|ttfr#|jt||d{St|drt|dr3t$j&j)t|j*}n|xsd}|j}t|tr|j d}|j||xs|d{St!dt#|d77B777#77-w) a Upload content from various input types or fetch an already-uploaded file. Args: file_input: The content to upload. Can be: - File: Already uploaded file (returned as-is) - str/Path: Path to a file on disk - SourceText: Text content, file, or file_id with explicit filename - BufferedIOBase: File-like binary object external_file_id: Optional external identifier for the file Returns: File: The uploaded (or fetched) file object Raises: ValueError: If the input type is not supported or required info is missing Nzutf-8r3rCzUnsupported file type: r;zUnsupported file input type: zB. Supported types: str, Path, SourceText, BufferedIOBase, or File.) isinstancerr rr text_contentencoder8filenamer3r@rr4r5hasattrrC ValueErrortyper-r.r?r;)rrFr* text_bytesr&rKs rupload_contentzFileClient.upload_contentis* j$ '  j* -!!-!]]:+=+=>>>((4'44;;GD !.. 0 QJ4G4G Q6JOOc4[9!-- (*:*Qj>Q>QJOOU3!..OO%5%V9L9L%VPV&1$//..0gs+%nnW5G!..-N1D1DN!#:4 ;P:Q!RSS S$K 0))#j/;KLL LZ (z6*77++C ,@A+5v!oo'G'3'!..1**74D4PQQ Q/Z0@/ABST _?  MRsAK5K"AK5#K%$AK5>K(?AK5K+Ar^s) ..<llr