NgjddlZddlZddlmZmZmZddlmZddlm Z ddl m Z Gdde Z dS)N)AnyListOptional)Document) BaseLoader)UnstructuredFileLoaderc ^eZdZdZ d dededededeed df d Zd e e fd Z dS) OBSFileLoaderz Load from the `Huawei OBS file`.Nbucketkeyclientendpointconfigreturnc ddlm}n#t$rtdwxYw|s|std|st }|dr||d}nH||d|d |d | }t ||std ||_||_ ||_ d S)a= Initialize the OBSFileLoader with the specified settings. Args: bucket (str): The name of the OBS bucket to be used. key (str): The name of the object in the OBS bucket. client (ObsClient, optional): An instance of the ObsClient to connect to OBS. endpoint (str, optional): The endpoint URL of your OBS bucket. This parameter is mandatory if `client` is not provided. config (dict, optional): The parameters for connecting to OBS, provided as a dictionary. This parameter is ignored if `client` is provided. The dictionary could have the following keys: - "ak" (str, optional): Your OBS access key (required if `get_token_from_ecs` is False and bucket policy is not public read). - "sk" (str, optional): Your OBS secret key (required if `get_token_from_ecs` is False and bucket policy is not public read). - "token" (str, optional): Your security token (required if using temporary credentials). - "get_token_from_ecs" (bool, optional): Whether to retrieve the security token from ECS. Defaults to False if not provided. If set to True, `ak`, `sk`, and `token` will be ignored. Raises: ValueError: If the `esdk-obs-python` package is not installed. TypeError: If the provided `client` is not an instance of ObsClient. ValueError: If `client` is not provided, but `endpoint` is missing. Note: Before using this class, make sure you have registered with OBS and have the necessary credentials. The `ak`, `sk`, and `endpoint` values are mandatory unless `get_token_from_ecs` is True or the bucket policy is public read. `token` is required when using temporary credentials. Example: To create a new OBSFileLoader with a new client: ``` config = { "ak": "your-access-key", "sk": "your-secret-key" } obs_loader = OBSFileLoader("your-bucket-name", "your-object-key", config=config) ``` To create a new OBSFileLoader with an existing client: ``` from obs import ObsClient # Assuming you have an existing ObsClient object 'obs_client' obs_loader = OBSFileLoader("your-bucket-name", "your-object-key", client=obs_client) ``` To create a new OBSFileLoader without an existing client: ``` obs_loader = OBSFileLoader("your-bucket-name", "your-object-key", endpoint="your-endpoint-url") ``` r) ObsClientzfCould not import esdk-obs-python python package. Please install it with `pip install esdk-obs-python`.z.Either OBSClient or endpoint must be provided.get_token_from_ecsECS)serversecurity_provider_policyaksktoken) access_key_idsecret_access_keysecurity_tokenrzClient must be ObsClient typeN) obsr ImportError ValueErrordictget isinstance TypeErrorrr r )selfr r rrrrs i/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/document_loaders/obs_file.py__init__zOBSFileLoader.__init__s0h  % % % % % % %   H     S !QRRR zz.// "(USSS""(**T"2"2&,jj&6&6#)::g#6#6#  &),, =;<< <  s #c~tj5}|d|jd|j}t jtj|d|j |j|j|t|}| cdddS#1swxYwYdS)zLoad documents./T)exist_ok) bucketName objectKey downloadFileN) tempfileTemporaryDirectoryr r osmakedirspathdirnamerr-rload)r%temp_dir file_pathloaders r&r4zOBSFileLoader.load_s  ( * * !h#>>dk>>DH>>I K 22T B B B B K $ $;$( %   ,I66F;;== ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !sBB22B69B6)Nr N) __name__ __module__ __qualname____doc__strrrr!r'rrr4r&r r s** !% MMMM M  M  M MMMM^ !d8n ! ! ! ! ! !r>r ) r0r.typingrrrlangchain_core.documentsr)langchain_community.document_loaders.baser1langchain_community.document_loaders.unstructuredrr r=r>r&rCs &&&&&&&&&&------@@@@@@TTTTTT\!\!\!\!\!J\!\!\!\!\!r>