o Zhw$@sddlmZmZmZmZmZddlmZddlm Z ddl m Z ddl m Z mZer0ddlmZeddd d Gd d d e Zd S)) TYPE_CHECKINGAnyCallableDictOptional) deprecated)Document)get_from_dict_or_env) BaseModelmodel_validatorApifyDatasetLoaderz0.3.18zThis class is deprecated and will be removed in a future version. You can swap to using the `ApifyWrapper` implementation in `langchain_apify` package. See zlangchain_apify.ApifyWrapper)ZsincemessageZalternative_importc@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. rZAPIFY_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`.)r rrrgetattr http_clientheaders ImportError) clsrrrrclientrZ async_clientrr!Z/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/utilities/apify.pyvalidate_environment!s*  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&defaultDatasetIdZ dataset_idr))$langchain_community.document_loadersr ractorcall selfr'r(r)r$r%r&r Z actor_callr!r!r" call_actorCs  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 rr/r0r1r!r!r" acall_actorks   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 r7r$r%r&r,r-)r.r rtaskr0 r2r6r7r)r$r%r&r Z task_callr!r!r"call_actor_tasks  zApifyWrapper.call_actor_taskc r4)r8rr r9Nr,r-)r.r rr:r0r;r!r!r"acall_actor_tasks   zApifyWrapper.acall_actor_task)__name__ __module__ __qualname____doc__r__annotations__rrstrr classmethodrr#rrintr3r5r<r=r!r!r!r"r s  &  .  .  /  rN)typingrrrrrZlangchain_core._apirZlangchain_core.documentsrZlangchain_core.utilsr Zpydanticr r r.r rr!r!r!r"s