Ji\2ddlZddlZddlZddlmZmZmZddlmZddl m Z ddl m Z m Z mZmZmZmZddlmZddlmZddlmZdd lmZmZmZdd lmZmZdd lm Z Gd d eZ!GddZ"e"Z#y)N)OptionalListUnion) BaseModelAsyncLlamaCloud) ClassifyJobClassifierRuleClassifyJobResultsClassifyParsingConfiguration StatusEnumFile)OMIT) FileClient)POLLING_TIMEOUT_SECONDS)is_terminal_statusaugment_async_errors FileInput)DEFAULT_NUM_WORKERSrun_jobs)ClassifyJobResultsWithFilesc"eZdZUeed<eed<y)ClassificationOutputfile_idclassificationN)__name__ __module__ __qualname__str__annotations__}/var/www/html/BankruptcyAI-uat/bankruptcy-ai/venv/lib/python3.12/site-packages/llama_cloud_services/beta/classifier/client.pyrrs Lr"rceZdZdZddefdedeededefdZ e d%d edeed eed d fd Z d&de e de edeed efdZ d&de e de edeed efdZ d'de e de edeeded ef dZ d'de e de edeeded ef dZddedfde e deeeefdeedededed efdZddedfde e deeeefdeedededed efdZ d'de e dedeeded ef dZ d'de e dedeeded ef dZddedfde e d e edeedededed efd!Z d'de e d e edeeded ef d"Z!d#ed efd$Z"y)( LlamaClassifya Experimental - Client for interacting with the LlamaCloud Classifier API. The Classification API is currently in beta and may change in the future without notice. Args: client: The LlamaCloud client to use. project_id: The project ID to use. polling_interval: The interval to poll for job completion in seconds. polling_timeout: The timeout for the job to complete in seconds. Ng?client project_idpolling_intervalpolling_timeoutc^||_||_||_t|||_||_yN)r&r'r(r file_clientr))selfr&r'r(r)s r#__init__zLlamaClassify.__init__.s1 $ 0%fj9.r"api_keybase_urlreturnClassifyClientc.t||}|||S)z; Create a classify client from an API key. )tokenr0r)clsr/r'r0r&s r# from_api_keyzLlamaClassify.from_api_key;s#!wB    r"rulesfile_idsparsing_configurationcK|jjj|||xst|jd{S7w)a Create a classify job. Meant to expose lower level access to classifier jobs for advanced use cases. Experimental: This endpoint is not yet ready for production use and is subject to change at any time. Args: rules: The rules to use for classification. file_ids: The IDs of the files to classify. parsing_configuration: The parsing configuration to use for classification. Returns: The classify job. )r7r8r9r'N)r& classifiercreate_classify_jobrr'r-r7r8r9s r#acreate_classify_jobz"LlamaClassify.acreate_classify_jobKsJ&[[++??"7"?4 @    s?AAAct5tj|j|||cdddS#1swYyxYwr+)rasynciorunr>r=s r#r<z!LlamaClassify.create_classify_jobes> " # ;;))%;PQ   s &;ATraise_on_errorcK|j|||d{}|j|jd{}|r5|jtj k(rt d|j|jjj|j|jd{}|S777 w)a Classify a list of files by their IDs. Note that even if a job fails, some of the files may have been classified successfully. In this case, you may want to set raise_on_error to False and check the results for successful classifications. Args: rules: The rules to use for classification. file_ids: The IDs of the files to classify. parsing_configuration: The parsing configuration to use for classification. raise_on_error: Whether to raise an error if the classification job fails. Returns: The results of the classification job. )r7r8r9Nz%Error classifying files under job ID r') r>wait_for_job_completionidstatusr ERROR ValueErrorr&r;get_classification_job_resultsr')r-r7r8r9rB classify_jobclassify_job_with_statusresultss r#aclassify_file_idsz LlamaClassify.aclassify_file_idsps*"66"77  *.)E)Eloo)V#V 6==AQAQQ78P8S8S7TU  ..MM $ ' 'N   % $W s2CB>"CCA9C7C8CCCc t5tj|j||||cdddS#1swYyxYwr+)rr@rArN)r-r7r8r9rBs r#classify_file_idszLlamaClassify.classify_file_idssC" # ;;''8%:N   s '<AFfilesworkers show_progressc^Kt|ts|g}|Dcgc]}|jj|}}t |||dd{} |j || D cgc]} | j c} ||d{} tj| | Scc}w7Scc} w7&w)aD Classify one or more files from various input types. Args: rules: The rules to use for classification. files: The file(s) to classify. Can be a single file or list of files. Each can be: - str/Path: File path - SourceText: Text content or file with explicit filename - File: Already uploaded file - BufferedIOBase: File-like object parsing_configuration: The parsing configuration to use for classification. raise_on_error: Whether to raise an error if the classification job fails. workers: Number of parallel workers for uploading files. show_progress: Whether to show progress bars. Returns: The results of the classification job with file metadata. z"Uploading files for classification)rSrRdescN) isinstancelistr,upload_contentrrNrFrfrom_classify_job_results) r-r7rQr9rBrRrS file_input coroutinesuploaded_filesfilerMs r# aclassifyzLlamaClassify.aclassifys8%&GEKP Afile_input_pathc~Ktjdtd|j||||d{S7w)6 Deprecated: Use aclassify() instead. z:aclassify_file_path is deprecated, use aclassify() instead stacklevelNwarningswarnDeprecationWarningr^r-r7rar9rBs r#aclassify_file_pathz!LlamaClassify.aclassify_file_pathsD  H  ^^ ?$9>    s 4=;=cbtjdtd|j||||S)5 Deprecated: Use classify() instead. z8classify_file_path is deprecated, use classify() insteadrdrerhrirjr`rks r#classify_file_pathz LlamaClassify.classify_file_paths6  F  }} ?$9>  r"file_input_pathscKtjdtd|j||||||d{S7w)rcz;aclassify_file_paths is deprecated, use aclassify() insteadrdreNrg)r-r7rqr9rBrRrSs r#aclassify_file_pathsz"LlamaClassify.aclassify_file_paths)sL  I  ^^   !        s 6?=?cbtjdtd|j||||S)rnz9classify_file_paths is deprecated, use classify() insteadrdrero)r-r7rqr9rBs r#classify_file_pathsz!LlamaClassify.classify_file_pathsCs7  G  }} #%:N  r"job_idc:K|jjj||jd{}t j}t |j st j|z }||jkDrtd|d|dtj|jd{|jjj||jd{}t |j s|S77W7 w)ax Wait for a classify job to complete. Meant to expose lower level access to classifier jobs for advanced use cases. Experimental: This endpoint is not yet ready for production use and is subject to change at any time. Args: job_id: The ID of the job to wait for. Returns: The classify job with status. rDNzJob z timed out after z seconds) r&r;get_classify_jobr'timerrGr) TimeoutErrorr@sleepr()r-rvjob start_timepolling_durations r#rEz%LlamaClassify.wait_for_job_completionVsKK**;; t<  YY[ $SZZ0#yy{Z7 $"6"66"6("34D3EXN-- 5 56 6 6 ..??4??@C%SZZ0   7s:5DDBD?D8D8D9DDDD)NNr+)NT)#rrr__doc__rrrrfloatr. classmethodr6rWr r r r>r<boolr rNrPrrrrintrr^r`rlrprsrurEr!r"r#r%r%"s %)"%!8 / /SM / /  /%)"&    SM  3-      &IM  N# s)  ((DE    <IM N# s)  ((DE   IM# 'N#'s)' ((DE '  '  'ZIM# N# s)  ((DE     $IM#*#3 N#3 YY/03  ((DE 3  3  3 3  %3 rIM#*#%N#%YY/0% ((DE %  %  %% %%VIM#  N#   ((DE    %  .IM#  N#   ((DE    %  .IM#*# N# s)  ((DE        % <IM#  N# s)  ((DE    %  &CKr"r%)$r@ryrhtypingrrrpydanticrllama_cloud.clientrllama_cloud.typesr r r r r r'llama_cloud.resources.classifier.clientr!llama_cloud_services.files.clientrllama_cloud_services.constantsrllama_cloud_services.utilsrrrllama_index.core.async_utilsrr*llama_cloud_services.beta.classifier.typesrrr%r2r!r"r#rsd ((.98B G 9 NNb r"