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

This commit is contained in:
sabaimran
2025-01-05 13:40:20 -08:00
parent 756f4a2a66
commit a5705a5aa6

View File

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