blob42/langchain
1
0
Fork 0
forked from Archives/langchain
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Allow recovering from JSONDecoder errors in StructuredOutputParser (#2616)

Browse Source
This commit is contained in:
Nuno Campos 2023-04-09 15:32:49 +01:00 committed by GitHub
parent 6f39e88a2c
commit 5376799a23
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
langchain/output_parsers/structured.py
View File

@ -39,7 +39,10 @@ class StructuredOutputParser(BaseOutputParser):
def parse(self, text: str) -> BaseModel: def parse(self, text: str) -> BaseModel:
json_string = text.split("```json")[1].strip().strip("```").strip() json_string = text.split("```json")[1].strip().strip("```").strip()
try:
json_obj = json.loads(json_string) json_obj = json.loads(json_string)
except json.JSONDecodeError as e:
raise OutputParserException(f"Got invalid JSON object. Error: {e}")
for schema in self.response_schemas: for schema in self.response_schemas:
if schema.name not in json_obj: if schema.name not in json_obj:
raise OutputParserException( raise OutputParserException(