sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-23 15:38:55 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

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
Show stats 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):