Просмотр исходного кода

Update api server: management of human intervention status

StrayWarrior 1 день назад
Родитель
Сommit
19171f7aa2
2 измененных файлов с 13 добавлено и 3 удалено
  1. 0 1
      pqai_agent_server/utils/__init__.py
  2. 13 2
      pqai_agent_server/utils/common.py

+ 0 - 1
pqai_agent_server/utils/__init__.py

@@ -1,5 +1,4 @@
 from .common import wrap_response
-from .common import quit_human_intervention_status
 
 from .prompt_util import (
     run_openai_chat,

+ 13 - 2
pqai_agent_server/utils/common.py

@@ -11,7 +11,18 @@ def wrap_response(code, msg=None, data=None):
     return jsonify(resp)
 
 
-def quit_human_intervention_status(user_id, staff_id):
+def quit_human_intervention(user_id, staff_id) -> bool:
     url = f"http://ai-wechat-hook-internal.piaoquantv.com/manage/insertEvent?sender={user_id}&receiver={staff_id}&type=103&content=SYSTEM"
     response = requests.post(url, timeout=20)
-    return response.json()
+    if response.status_code == 200 and response.json().get("code") == 0:
+        return True
+    else:
+        return False
+
+def enter_human_intervention(user_id, staff_id) -> bool:
+    url = f"http://ai-wechat-hook-internal.piaoquantv.com/manage/insertEvent?sender={user_id}&receiver={staff_id}&type=104&content=SYSTEM"
+    response = requests.post(url, timeout=20)
+    if response.status_code == 200 and response.json().get("code") == 0:
+        return True
+    else:
+        return False