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

fix regex parser (#446)

Browse Source
This commit is contained in:
Harrison Chase 2022-12-28 09:02:40 -05:00 committed by GitHub
parent 90e8ccc898
commit 5cc6bf1a9c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
langchain/prompts/base.py
View File

@ -69,14 +69,21 @@ class RegexParser(BaseOutputParser, BaseModel):
regex: str regex: str
output_keys: List[str] output_keys: List[str]
default_output_key: Optional[str] = None
def parse(self, text: str) -> Dict[str, str]: def parse(self, text: str) -> Dict[str, str]:
"""Parse the output of an LLM call.""" """Parse the output of an LLM call."""
match = re.search(self.regex, text) match = re.search(self.regex, text)
if match: if match:
return {key: match.group(i) for i, key in enumerate(self.output_keys)} return {key: match.group(i + 1) for i, key in enumerate(self.output_keys)}
else: else:
raise ValueError(f"Could not parse output: {text}") if self.default_output_key is None:
raise ValueError(f"Could not parse output: {text}")
else:
return {
key: text if key == self.default_output_key else ""
for key in self.output_keys
}
class BasePromptTemplate(BaseModel, ABC): class BasePromptTemplate(BaseModel, ABC):