sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2025-02-17 16:14:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Only check the prompt safety if the agent is not private

Browse source
This commit is contained in:
sabaimran 2024-10-07 21:42:14 -07:00
parent f700d5bddb
commit e10a0571ff
1 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/khoj/routers/api_agents.py
View file

@ -169,7 +169,11 @@ async def create_agent(
) -> Response: ) -> Response:
user: KhojUser = request.user.object user: KhojUser = request.user.object
is_safe_prompt, reason = await acheck_if_safe_prompt(body.persona) is_safe_prompt, reason = True, ""
if body.privacy_level != Agent.PrivacyLevel.PRIVATE:
is_safe_prompt, reason = await acheck_if_safe_prompt(body.persona)
if not is_safe_prompt: if not is_safe_prompt:
return Response( return Response(
content=json.dumps({"error": f"{reason}"}), content=json.dumps({"error": f"{reason}"}),
@ -217,7 +221,11 @@ async def update_agent(
) -> Response: ) -> Response:
user: KhojUser = request.user.object user: KhojUser = request.user.object
is_safe_prompt, reason = await acheck_if_safe_prompt(body.persona) is_safe_prompt, reason = True, ""
if body.privacy_level != Agent.PrivacyLevel.PRIVATE:
is_safe_prompt, reason = await acheck_if_safe_prompt(body.persona)
if not is_safe_prompt: if not is_safe_prompt:
return Response( return Response(
content=json.dumps({"error": f"{reason}"}), content=json.dumps({"error": f"{reason}"}),