github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-18 10:43:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Harrison/update regex (#1534)

Browse Source 
Co-authored-by: Luis <57528712+LuisLechugaRuiz@users.noreply.github.com>
This commit is contained in:
Harrison Chase 2023-03-08 17:41:17 -08:00 committed by GitHub
parent 2a7215bc3b
commit 7eba828e1b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
langchain/agents/conversational/base.py
View File

@ -78,7 +78,7 @@ class ConversationalAgent(Agent):
def _extract_tool_and_input(self, llm_output: str) -> Optional[Tuple[str, str]]: def _extract_tool_and_input(self, llm_output: str) -> Optional[Tuple[str, str]]:
if f"{self.ai_prefix}:" in llm_output: if f"{self.ai_prefix}:" in llm_output:
return self.ai_prefix, llm_output.split(f"{self.ai_prefix}:")[-1].strip() return self.ai_prefix, llm_output.split(f"{self.ai_prefix}:")[-1].strip()
regex = r"Action: (.*?)\nAction Input: (.*)" regex = r"Action: (.*?)[\n]*Action Input: (.*)"
match = re.search(regex, llm_output) match = re.search(regex, llm_output)
if not match: if not match:
raise ValueError(f"Could not parse LLM output: `{llm_output}`") raise ValueError(f"Could not parse LLM output: `{llm_output}`")

2
langchain/agents/mrkl/base.py
View File

@ -40,7 +40,7 @@ def get_action_and_input(llm_output: str) -> Tuple[str, str]:
""" """
if FINAL_ANSWER_ACTION in llm_output: if FINAL_ANSWER_ACTION in llm_output:
return "Final Answer", llm_output.split(FINAL_ANSWER_ACTION)[-1].strip() return "Final Answer", llm_output.split(FINAL_ANSWER_ACTION)[-1].strip()
regex = r"Action: (.*?)\nAction Input: (.*)" regex = r"Action: (.*?)[\n]*Action Input: (.*)"
match = re.search(regex, llm_output, re.DOTALL) match = re.search(regex, llm_output, re.DOTALL)
if not match: if not match:
raise ValueError(f"Could not parse LLM output: `{llm_output}`") raise ValueError(f"Could not parse LLM output: `{llm_output}`")