o

���g��@
s^
dZd
dl
mZ
d
dlmZmZmZmZmZ
d
dl	m
Z

d
dlmZ
d
dl
mZ
d
dlmZ
d
dlmZmZ
d
d	lmZ
d
d
lmZ
d
dlmZmZmZ
d
dlmZ
d
d
lmZ
d
dl m!Z!
d
dl"m#Z#
d
dl$m%Z%
errd
dl&m'Z'
e
dedd�Gdd�de��
Z(e
dedd�Gdd�d��
Z)e
dedd�Gdd�de(��
Z*e
dedd�Gdd�de��
Z+dS)zPChain that implements the ReAct paper from https://arxiv.org/pdf/2210.03629.pdf.�)
�annotations)�
TYPE_CHECKING�Any�List�Optional�Sequence)
�
deprecated)
�Document)
�BaseLanguageModel)
�BasePromptTemplate)�BaseTool�Tool)
�Field)
�AGENT_DEPRECATION_WARNING)�Agent�
AgentExecutor�AgentOutputParser)
�	AgentType�
�ReActOutputParser�
�TEXTWORLD_PROMPT�
�WIKI_PROMPT)
�validate_tools_single_input)
�Docstorez0.1.0z1.0)�message�removalc
s�eZ
dZUd
Zeed�
Zded<eddd	��
Z	e
ddd��
Zeddd��
Zed�f
dd��
Z
e
ddd��
Ze
d dd��
Ze
ddd��
Z�ZS)!�ReActDocstoreAgentzAgent for the ReAct chain.)
�default_factoryr�
output_parser�kwargsr�returnc

K
st�S)
Nr)�clsr!�r$�F/usr/local/lib/python3.10/dist-packages/langchain/agents/react/base.py�_get_default_output_parser$�z-ReActDocstoreAgent._get_default_output_parser�strc


C
stj
S)
z#Return Identifier of an agent type.)r�REACT_DOCSTORE�
�selfr$r$r%�_agent_type(szReActDocstoreAgent._agent_type�tools�Sequence[BaseTool]rc
C
�tS�
zReturn default prompt.r�r#r-r$r$r%�
create_prompt-�z ReActDocstoreAgent.create_prompt�Nonec
s^t|j
|
�
t��|
�

t|
�
d
krtd|
���
�
dd�|
D�
}|ddhkr-td|���
�
dS)N�z-Exactly two tools must be specified, but got c
S
�h|]}
|
j�qSr$�
�name��.0�toolr$r$r%�	<setcomp>8�z5ReActDocstoreAgent._validate_tools.<locals>.<setcomp>�Lookup�Searchz,Tool names should be Lookup and Search, got �r�__name__�super�_validate_tools�len�
ValueError�r#r-�
tool_names�
�	__class__r$r%rC2s






��z"ReActDocstoreAgent._validate_toolsc


C
�d
S)z&Prefix to append the observation with.z
Observation: r$r*r$r$r%�observation_prefix>r3z%ReActDocstoreAgent.observation_prefix�	List[str]c


C
sd
g
S)Nz
Observation:r$r*r$r$r%�_stopCr'zReActDocstoreAgent._stopc


C
rJ)z#Prefix to append the LLM call with.zThought:r$r*r$r$r%�
llm_prefixGr3zReActDocstoreAgent.llm_prefix)r!rr"r�r"r(�r-r.r"r�r-r.r"r4�r"rL)rA�
__module__�__qualname__�__doc__rrr �__annotations__�classmethodr&�propertyr,r2rCrKrMrN�
__classcell__r$r$rHr%rs"







rc@
sJeZ
dZd
Zddd�Zdd	d
�Zddd�Zedd
d��
Zeddd��
Z	dS)�DocstoreExplorerz5Class to assist with exploration of a document store.�docstorercC
s|
|_d
|_
d|_d|_d
S)z=Initialize with a docstore, and set initial document to None.N�r
)r[�document�
lookup_str�lookup_index)r+r[r$r$r%�__init__Us



zDocstoreExplorer.__init__�termr(r"cC
s,|j�
|
�
}t|t�r||_|jSd
|_|S)z5Search for a term in the docstore, and if found save.N)r[�search�
isinstancer	r]�_summary)r+ra�resultr$r$r%rb\s




zDocstoreExplorer.searchc
s��jd
ur	t
d�
�
|
���jkr|
���_d�_n�jd7_�f
dd��jD�
}t|�
dkr2dS�jt|�
kr;dSd	�jd�d
t|�
�d�}|�d|�j��S)
z%Lookup a term in document (if saved).Nz/Cannot lookup without a successful search firstr
�
c

sg|]}
�j|
�
�vr|
�qSr$)r^�lower)r:�
pr*r$r%�
<listcomp>osz+DocstoreExplorer.lookup.<locals>.<listcomp>z
No ResultszNo More Resultsz(Result �
/�
)�
 )r]rErgr^r_�_paragraphsrD)r+ra�lookups�
result_prefixr$r*r%�lookupfs











zDocstoreExplorer.lookupc

C
s
|jd
S)Nr
)
rmr*r$r$r%rdxs
zDocstoreExplorer._summaryrLc

C
s |jdur	t
d
�
�
|jj�d�
S)Nz(Cannot get paragraphs without a documentz

)r]rE�page_content�splitr*r$r$r%rm|s


zDocstoreExplorer._paragraphsN)r[r)rar(r"r(rOrR)
rArSrTrUr`rbrprXrdrmr$r$r$r%rZMs





rZc
s4eZ
dZd
Zeddd��
Zed�f
d	d
��
Z�ZS)
�ReActTextWorldAgentz$Agent for the ReAct TextWorld chain.r-r.r"rc
C
r/r0rr1r$r$r%r2�r3z!ReActTextWorldAgent.create_promptr4c
s\t|j
|
�
t��|
�

t|
�
d
krtd|
���
�
dd�|
D�
}|dh
kr,td|���
�
dS)Nrfz,Exactly one tool must be specified, but got c
S
r6r$r7r9r$r$r%r<�r=z6ReActTextWorldAgent._validate_tools.<locals>.<setcomp>�PlayzTool name should be Play, got r@rFrHr$r%rC�s






�z#ReActTextWorldAgent._validate_toolsrPrQ)rArSrTrUrWr2rCrYr$r$rHr%rs�s

rsc
s"eZ
dZd
Zd
�f
dd	�Z�ZS)�
ReActChainz3[Deprecated] Chain that implements the ReAct paper.�llmr
r[rr!rc
sPt|�
}t
d
|jdd�t
d|jdd�g}t�|
|�}t�jd||d�|�
�

dS)	z'Initialize with the LLM and a docstore.r?z"Search for a term in the docstore.)r8�func�descriptionr>zLookup a term in the docstore.)�agentr-Nr$)rZr
rbrpr�from_llm_and_toolsrBr`)r+rvr[r!�docstore_explorerr-ryrHr$r%r`�s


�


��
zReActChain.__init__)rvr
r[rr!r)rArSrTrUr`rYr$r$rHr%ru�sruN),rU�
__future__r�typingrrrrr�langchain_core._apir�langchain_core.documentsr	�langchain_core.language_modelsr
�langchain_core.promptsr�langchain_core.toolsrr
�pydanticr�langchain._api.deprecationr�langchain.agents.agentrrr�langchain.agents.agent_typesr�$langchain.agents.react.output_parserr�'langchain.agents.react.textworld_promptr�"langchain.agents.react.wiki_promptr�langchain.agents.utilsr�!langchain_community.docstore.baserrrZrsrur$r$r$r%�<module>
sT














�.


�1


�


�