jbddlZddlZddlmZmZmZddlmZddlm Z ddl m Z Gdde Z y)N)AnyListOptional)Document) BaseLoader)UnstructuredFileLoaderc VeZdZdZ d dededededeeddf d Zd d ede e fd Z y) OBSFileLoaderz Load from the `Huawei OBS file`.Nbucketkeyclientendpointconfigreturncz ddlm}|sn|s td|s t }|j dr ||d}n9||j d|j d |j d | }t ||s td ||_||_ ||_ y #t$r tdwxYw)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 TypeErrorr r r )selfr r r rrrs w/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langchain_community/document_loaders/obs_file.py__init__zOBSFileLoader.__init__sh  %  !QRRzz./"(US""(**T"2&,jj&6#)::g#6#  &),;< <  / H  s B%%B:modectj5}|d|jd|j}t j tj j|d|jj|j|j|t||}|jcdddS#1swYyxYw)zLoad documents./T)exist_ok) bucketName objectKey downloadFile)r'N) tempfileTemporaryDirectoryr r osmakedirspathdirnamer r-rload)r$r'temp_dir file_pathloaders r%r4zOBSFileLoader.load_s  ( ( *h#*Adkk]!DHH:>I KK 2T B KK $ $;;$(( % ,IDAF;;=+ * *s B!CC )NN)single) __name__ __module__ __qualname____doc__strrrr r&rrr4r%r r sr* !% MMM M  M  M M^ ! !DN !r@r ) r0r.typingrrrlangchain_core.documentsr)langchain_community.document_loaders.baser1langchain_community.document_loaders.unstructuredrr r?r@r%rEs* &&-@T\!J\!r@