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

Remove ```python codeblock prefix from raw json before deserialize

Browse Source
This commit is contained in:
Debanjum
2024-11-19 11:25:12 -08:00
parent 001c13ef43
commit 0e7d611a80
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/khoj/processor/conversation/utils.py
View File

@@ -528,7 +528,7 @@ def reciprocal_conversation_to_chatml(message_pair):
def clean_json(response: str): def clean_json(response: str):
"""Remove any markdown json codeblock and newline formatting if present. Useful for non schema enforceable models""" """Remove any markdown json codeblock and newline formatting if present. Useful for non schema enforceable models"""
return response.strip().replace("\n", "").removeprefix("```json").removesuffix("```") return response.strip().replace("\n", "").removeprefix("```json").removeprefix("```python").removesuffix("```")
def clean_code_python(code: str): def clean_code_python(code: str):