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

After agent prompt safe check is parsed as json, load it into a json object

Browse Source
This commit is contained in:
sabaimran
2025-01-05 13:40:20 -08:00
parent 756f4a2a66
commit a5705a5aa6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/khoj/routers/helpers.py
View File

@@ -325,7 +325,7 @@ async def acheck_if_safe_prompt(system_prompt: str, user: KhojUser = None, lax:
response = response.strip() response = response.strip()
try: try:
response = clean_json(response) response = json.loads(clean_json(response))
is_safe = response.get("safe", "True") == "True" is_safe = response.get("safe", "True") == "True"
if not is_safe: if not is_safe:
reason = response.get("reason", "") reason = response.get("reason", "")