github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-21 14:18:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Harrison/fix react stateful (#219)

Browse Source 
fix issue with react being stateful
This commit is contained in:
Harrison Chase 2022-11-28 18:03:04 -08:00 committed by GitHub
parent 03c7140228
commit 0568998166
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
langchain/agents/agent.py
View File

@ -87,6 +87,9 @@ class Agent(Chain, BaseModel, ABC):
"""Create a prompt for this class.""" """Create a prompt for this class."""
return cls.prompt return cls.prompt
def _prepare_for_new_call(self) -> None:
pass
@classmethod @classmethod
def from_llm_and_tools(cls, llm: LLM, tools: List[Tool], **kwargs: Any) -> "Agent": def from_llm_and_tools(cls, llm: LLM, tools: List[Tool], **kwargs: Any) -> "Agent":
"""Construct an agent from an LLM and tools.""" """Construct an agent from an LLM and tools."""
@ -119,6 +122,8 @@ class Agent(Chain, BaseModel, ABC):
def _call(self, inputs: Dict[str, str]) -> Dict[str, str]: def _call(self, inputs: Dict[str, str]) -> Dict[str, str]:
"""Run text through and get agent response.""" """Run text through and get agent response."""
text = inputs[self.input_key] text = inputs[self.input_key]
# Do any preparation necessary when receiving a new input.
self._prepare_for_new_call()
# Construct a mapping of tool name to tool for easy lookup # Construct a mapping of tool name to tool for easy lookup
name_to_tool_map = {tool.name: tool.func for tool in self.tools} name_to_tool_map = {tool.name: tool.func for tool in self.tools}
# Construct the initial string to pass into the LLM. This is made up # Construct the initial string to pass into the LLM. This is made up

3
langchain/agents/react/base.py
View File

@ -31,6 +31,9 @@ class ReActDocstoreAgent(Agent, BaseModel):
f"Tool names should be Lookup and Search, got {tool_names}" f"Tool names should be Lookup and Search, got {tool_names}"
) )
def _prepare_for_new_call(self) -> None:
self.i = 1
def _fix_text(self, text: str) -> str: def _fix_text(self, text: str) -> str:
return text + f"\nAction {self.i}:" return text + f"\nAction {self.i}:"