highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-04-02 17:08:22 +02:00
Code Issues Packages Projects Releases Wiki Activity

Use strict=False for Parsing LLM Jsons ()

Browse Source
This commit is contained in:
Yuhong Sun 2023-10-19 12:28:16 -07:00 committed by GitHub
parent 1bd76f528f
commit d9076a6ff6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
backend/danswer
direct_qa
qa_utils.py
utils
text_processing.py

2
backend/danswer/direct_qa/qa_utils.py

@ -85,7 +85,7 @@ def separate_answer_quotes(
answer_raw: str, is_json_prompt: bool = False
) -> Tuple[Optional[str], Optional[list[str]]]:
try:
model_raw_json = json.loads(answer_raw)
model_raw_json = json.loads(answer_raw, strict=False)
return extract_answer_quotes_json(model_raw_json)
except ValueError:
if is_json_prompt:

2
backend/danswer/utils/text_processing.py

@ -22,7 +22,7 @@ def extract_embedded_json(s: str) -> dict:
if first_brace_index == -1 or last_brace_index == -1:
raise ValueError("No valid json found")
return json.loads(s[first_brace_index : last_brace_index + 1])
return json.loads(s[first_brace_index : last_brace_index + 1], strict=False)
def clean_up_code_blocks(model_out_raw: str) -> str: