a bg@sxddlmZddlmZmZmZmZddlmZddl m Z ddl m Z ddl mZerdddlmZGdd d e Zd S) ) annotations) TYPE_CHECKINGIteratorListOptional)Document) BaseLoader)Blob)!AzureAIDocumentIntelligenceParser)TokenCredentialc@sJeZdZdZddddddddddddd d d d d dZddddZdS)!AzureAIDocumentIntelligenceLoaderz,Load a PDF with Azure Document Intelligence.Nprebuilt-layoutmarkdown)analysis_featuresazure_credentialstrz Optional[str]zOptional[bytes]zOptional[List[str]]zOptional['TokenCredential']None) api_endpointapi_key file_pathurl_path bytes_source api_version api_modelmoderrreturnc  Cs~|dus |dus |dus Jd|dus8| dus8Jd|dusP| dusPJd||_||_||_t|||||| | d|_dS)a Initialize the object for file processing with Azure Document Intelligence (formerly Form Recognizer). This constructor initializes a AzureAIDocumentIntelligenceParser object to be used for parsing files using the Azure Document Intelligence API. The load method generates Documents whose content representations are determined by the mode parameter. Parameters: ----------- api_endpoint: str The API endpoint to use for DocumentIntelligenceClient construction. api_key: str The API key to use for DocumentIntelligenceClient construction. file_path : Optional[str] The path to the file that needs to be loaded. Either file_path, url_path or bytes_source must be specified. url_path : Optional[str] The URL to the file that needs to be loaded. Either file_path, url_path or bytes_source must be specified. bytes_source : Optional[bytes] The bytes array of the file that needs to be loaded. Either file_path, url_path or bytes_source must be specified. api_version: Optional[str] The API version for DocumentIntelligenceClient. Setting None to use the default value from `azure-ai-documentintelligence` package. api_model: str Unique document model name. Default value is "prebuilt-layout". Note that overriding this default value may result in unsupported behavior. mode: Optional[str] The type of content representation of the generated Documents. Use either "single", "page", or "markdown". Default value is "markdown". analysis_features: Optional[List[str]] List of optional analysis features, each feature should be passed as a str that conforms to the enum `DocumentAnalysisFeature` in `azure-ai-documentintelligence` package. Default value is None. azure_credential: Optional[TokenCredential] The credentials to use for DocumentIntelligenceClient construction, when using credentials other than api_key (like AD). Examples: --------- >>> obj = AzureAIDocumentIntelligenceLoader( ... file_path="path/to/file", ... api_endpoint="https://endpoint.azure.com", ... api_key="APIKEY", ... api_version="2023-10-31-preview", ... api_model="prebuilt-layout", ... mode="markdown" ... ) Nz4file_path, url_path or bytes_source must be providedz4Either api_key or azure_credential must be provided.z;Only one of api_key or azure_credential should be provided.)rrrrrrr)rrrr parser) selfrrrrrrrrrrr/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/document_loaders/doc_intelligence.py__init__s@Ez*AzureAIDocumentIntelligenceLoader.__init__zIterator[Document])rccsv|jdur*t|j}|j|EdHnH|jdurJ|j|jEdHn(|jdurj|j|jEdHnt ddS)z Lazy load the document as pages.NzNo data source provided.) rr from_pathrparser parse_urlrZ parse_bytes ValueError)rZblobrrr lazy_loadrs    z+AzureAIDocumentIntelligenceLoader.lazy_load)NNNNNr r)__name__ __module__ __qualname____doc__r r%rrrrr s &^r N) __future__rtypingrrrrZlangchain_core.documentsrZ)langchain_community.document_loaders.baserZ1langchain_community.document_loaders.blob_loadersr Z,langchain_community.document_loaders.parsersr Zazure.core.credentialsr r rrrrs