|
@@ -1,15 +1,15 @@
|
|
|
from typing import List
|
|
|
+import requests
|
|
|
|
|
|
from pqai_agent.toolkit.base import BaseToolkit
|
|
|
from pqai_agent.toolkit.function_tool import FunctionTool
|
|
|
-import requests
|
|
|
-
|
|
|
+from pqai_agent import configs
|
|
|
|
|
|
class LarkAlertForHumanIntervention(BaseToolkit):
|
|
|
r"""A toolkit for Lark alert for human intervention."""
|
|
|
|
|
|
def __init__(self):
|
|
|
- self.webhook_url = 'https://open.feishu.cn/open-apis/bot/v2/hook/c316b559-1c6a-4c4e-97c9-50b44e4c2a9d'
|
|
|
+ self.webhook_url = configs.get().get('system', {}).get('human_intervention_alert_url', '')
|
|
|
super().__init__()
|
|
|
|
|
|
def send_lark_alert_for_human_intervention(
|
|
@@ -22,6 +22,9 @@ class LarkAlertForHumanIntervention(BaseToolkit):
|
|
|
Returns:
|
|
|
str: A confirmation message.
|
|
|
"""
|
|
|
+ if not self.webhook_url:
|
|
|
+ return "Webhook URL not configured."
|
|
|
+
|
|
|
req_body = {
|
|
|
"msg_type": "text",
|
|
|
"content": {
|