klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-07 13:23:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix-up Search Actor GPT's response for decoding it as valid JSON

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2023-03-18 03:30:30 -06:00
parent f63fd0995e
commit 939d7731da
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/khoj/processor/conversation/gpt.py
View File

@@ -167,7 +167,13 @@ Q: {text}
# Extract, Clean Message from GPT's Response # Extract, Clean Message from GPT's Response
response_text = response["choices"][0]["text"] response_text = response["choices"][0]["text"]
try: try:
questions = json.loads(response_text.strip(empty_escape_sequences)) questions = json.loads(
# Clean response to increase likelihood of valid JSON. E.g replace ' with " to enclose strings
response_text.strip(empty_escape_sequences)
.replace("['", '["')
.replace("']", '"]')
.replace("', '", '", "')
)
except json.decoder.JSONDecodeError: except json.decoder.JSONDecodeError:
logger.warn(f"GPT returned invalid JSON. Set question to empty list.\n{response_text}") logger.warn(f"GPT returned invalid JSON. Set question to empty list.\n{response_text}")
questions = [text] questions = [text]