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

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

9
langchain/prompts/base.py
View File

@ -69,14 +69,21 @@ class RegexParser(BaseOutputParser, BaseModel):
regex: str
output_keys: List[str]
default_output_key: Optional[str] = None
def parse(self, text: str) -> Dict[str, str]:
"""Parse the output of an LLM call."""
match = re.search(self.regex, text)
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:
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):