o ȳg"@shddlmZmZmZmZmZddlmZddlm Z ddl m Z m Z er*ddl mZGddde ZdS) ) TYPE_CHECKINGAnyCallableDictOptional)Document)get_from_dict_or_env) BaseModelmodel_validatorApifyDatasetLoaderc@sjeZdZUdZeed<eed<dZeeed<e dde de d efd d Z dddd d ede de e gefdeedeedeed dfddZdddd d ede de e gefdeedeedeed dfddZdddd dede de e gefdeedeedeed dfddZdddd dede de e gefdeedeedeed dfddZdS) ApifyWrapperzWrapper around Apify. To use, you should have the ``apify-client`` python package installed, and the environment variable ``APIFY_API_TOKEN`` set with your API key, or pass `apify_api_token` as a named parameter to the constructor. apify_clientapify_client_asyncNapify_api_tokenbefore)modevaluesreturnc Cst|dd}z=ddlm}m}||}t|jd}r$|jdd7<||}t|jd}r9|jdd7<||d <||d <W|StyNtd w) zValidate environment. Validate that an Apify API token is set and the apify-client Python package exists in the current environment. rAPIFY_API_TOKENr) ApifyClientApifyClientAsync httpx_clientz user-agentz; Origin/langchainhttpx_async_clientrrz`Could not import apify-client Python package. Please install it with `pip install apify-client`.)rrrrgetattr http_clientheaders ImportError) clsrrrrclientr async_clientrr!_/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/langchain_community/utilities/apify.pyvalidate_environments*  z!ApifyWrapper.validate_environment)build memory_mbytes timeout_secsactor_id run_inputdataset_mapping_functionr$r%r&r c C6ddlm}|j|j||||d}||d|dS)aRun an Actor on the Apify platform and wait for results to be ready. Args: actor_id (str): The ID or name of the Actor on the Apify platform. run_input (Dict): The input object of the Actor that you're trying to run. dataset_mapping_function (Callable): A function that takes a single dictionary (an Apify dataset item) and converts it to an instance of the Document class. build (str, optional): Optionally specifies the actor build to run. It can be either a build tag or build number. memory_mbytes (int, optional): Optional memory limit for the run, in megabytes. timeout_secs (int, optional): Optional timeout for the run, in seconds. Returns: ApifyDatasetLoader: A loader that will fetch the records from the Actor run's default dataset. rr r(r$r%r&defaultDatasetId dataset_idr))$langchain_community.document_loadersr ractorcall selfr'r(r)r$r%r&r actor_callr!r!r" call_actor8s  zApifyWrapper.call_actorc >ddlm}|j|j||||dIdH}||d|dS)aRun an Actor on the Apify platform and wait for results to be ready. Args: actor_id (str): The ID or name of the Actor on the Apify platform. run_input (Dict): The input object of the Actor that you're trying to run. dataset_mapping_function (Callable): A function that takes a single dictionary (an Apify dataset item) and converts it to an instance of the Document class. build (str, optional): Optionally specifies the actor build to run. It can be either a build tag or build number. memory_mbytes (int, optional): Optional memory limit for the run, in megabytes. timeout_secs (int, optional): Optional timeout for the run, in seconds. Returns: ApifyDatasetLoader: A loader that will fetch the records from the Actor run's default dataset. rr r+Nr,r-)r/r rr0r1r2r!r!r" acall_actor`s   zApifyWrapper.acall_actortask_id task_inputc Cr*)Run a saved Actor task on Apify and wait for results to be ready. Args: task_id (str): The ID or name of the task on the Apify platform. task_input (Dict): The input object of the task that you're trying to run. Overrides the task's saved input. dataset_mapping_function (Callable): A function that takes a single dictionary (an Apify dataset item) and converts it to an instance of the Document class. build (str, optional): Optionally specifies the actor build to run. It can be either a build tag or build number. memory_mbytes (int, optional): Optional memory limit for the run, in megabytes. timeout_secs (int, optional): Optional timeout for the run, in seconds. Returns: ApifyDatasetLoader: A loader that will fetch the records from the task run's default dataset. rr r9r$r%r&r,r-)r/r rtaskr1 r3r8r9r)r$r%r&r task_callr!r!r"call_actor_tasks  zApifyWrapper.call_actor_taskc r6)r:rr r;Nr,r-)r/r rr<r1r=r!r!r"acall_actor_tasks   zApifyWrapper.acall_actor_task)__name__ __module__ __qualname____doc__r__annotations__rrstrr classmethodrr#rrintr5r7r?r@r!r!r!r"r s &  .  .  /  r N)typingrrrrrlangchain_core.documentsrlangchain_core.utilsrpydanticr r r/r r r!r!r!r"s