File: //usr/local/lib/python3.10/dist-packages/langchain/chains/natbot/__pycache__/base.cpython-310.pyc
o
���g9 � @ s� d Z ddlmZ ddlZddlmZmZmZmZ ddl m
Z
ddl m Z ddl
mZ ddl
mZ dd lmZ dd
lmZ dd lmZ dd lmZmZ dd
lmZ ddlm Z e
dddd�G dd� de��Z e � � dS )z Implement an LLM driven browser.� )� annotationsN)�Any�Dict�List�Optional)�
deprecated)� BaseCache)�CallbackManagerForChainRun)� Callbacks)�BaseLanguageModel)�StrOutputParser)�Runnable)�
ConfigDict�model_validator)�Chain)�PROMPTz0.2.13z�Importing NatBotChain from langchain is deprecated and will be removed in langchain 1.0. Please import from langchain_community instead: from langchain_community.chains.natbot import NatBotChain. You may need to pip install -U langchain-community.z1.0)�since�message�removalc @ s� e Zd ZU dZded< ded< dZded< d Zded
< d Zded < d
Zded< dZ ded< e
ddd�Z e dd�e
d1dd���Ze
d2d d ��Ze
d3d d!��Zed4d#d$��Zed4d%d&��Z d5d6d+d,�Zd7d-d.�Zed8d/d0��ZdS )9� NatBotChaina� Implement an LLM driven browser.
**Security Note**: This toolkit provides code to control a web-browser.
The web-browser can be used to navigate to:
- Any URL (including any internal network URLs)
- And local files
Exercise care if exposing this chain to end-users. Control who is able to
access and use this chain, and isolate the network access of the server
that hosts this chain.
See https://python.langchain.com/docs/security for more information.
Example:
.. code-block:: python
from langchain.chains import NatBotChain
natbot = NatBotChain.from_default("Buy me a new hat.")
r
� llm_chain�str� objectiveNzOptional[BaseLanguageModel]�llm�url�
input_url_key�browser_content�input_browser_content_key� �previous_command�command�
output_keyT�forbid)�arbitrary_types_allowed�extra�before)�mode�valuesr �returnr c C s@ d|v r t �d� d|vr |d d ur t|d B t� B |d< |S )Nr z�Directly instantiating an NatBotChain with an llm is deprecated. Please instantiate with llm_chain argument or using the from_llm class method.r )�warnings�warnr r )�clsr' � r, �G/usr/local/lib/python3.10/dist-packages/langchain/chains/natbot/base.py�raise_deprecationE s �z NatBotChain.raise_deprecation�kwargsc K s t d��)zLoad with default LLMChain.z�This method is no longer implemented. Please use from_llm.llm = OpenAI(temperature=0.5, best_of=10, n=3, max_tokens=50)For example, NatBotChain.from_llm(llm, objective))�NotImplementedError)r+ r r/ r, r, r- � from_defaultR s �zNatBotChain.from_defaultr c K s" t |B t� B }| d||d�|��S )zLoad from LLM.)r r Nr, )r r )r+ r r r/ r r, r, r- �from_llm[ s zNatBotChain.from_llm� List[str]c C s | j | jgS )z@Expect url and browser content.
:meta private:
)r r ��selfr, r, r- �
input_keysc s zNatBotChain.input_keysc C s | j gS )z0Return command.
:meta private:
)r! r4 r, r, r- � output_keysk s zNatBotChain.output_keys�inputs�Dict[str, str]� run_manager�$Optional[CallbackManagerForChainRun]c C sn |pt �� }|| j }|| j }| jj| j|d d� | j|d d� d�d|�� id�}|� � }|| _| j
|iS )N�d i� )r r r r � callbacks)�config) r �get_noop_managerr r r �invoker r � get_child�stripr! )r5 r8 r: � _run_managerr r �llm_cmdr, r, r- �_calls s
�
�
zNatBotChain._callc C s | j || j|i}| |�| j S )a� Figure out next browser command to run.
Args:
url: URL of the site currently on.
browser_content: Content of the page as currently displayed by the browser.
Returns:
Next browser command to run.
Example:
.. code-block:: python
browser_content = "...."
llm_command = natbot.run("www.google.com", browser_content)
)r r r! )r5 r r �_inputsr, r, r- �execute� s �zNatBotChain.executec C s dS )N�
nat_bot_chainr, r4 r, r, r- � _chain_type� s zNatBotChain._chain_type)r' r r( r )r r r/ r r( r )r r r r r/ r r( r )r( r3 )N)r8 r9 r: r; r( r9 )r r r r r( r )r( r )�__name__�
__module__� __qualname__�__doc__�__annotations__r r r r r! r � model_configr � classmethodr. r1 r2 �propertyr6 r7 rE rG rI r, r, r, r- r s>
�
�
r ) rM �
__future__r r) �typingr r r r �langchain_core._apir �langchain_core.cachesr �langchain_core.callbacksr r
� langchain_core.language_modelsr � langchain_core.output_parsersr �langchain_core.runnablesr
�pydanticr r �langchain.chains.baser � langchain.chains.natbot.promptr r �
model_rebuildr, r, r, r- �<module> s, �