瀏覽代碼

Update lark_alert_for_human_intervention: pass url by config

StrayWarrior 1 月之前
父節點
當前提交
006a41cf76

+ 1 - 0
pqai_agent/configs/dev.yaml

@@ -54,6 +54,7 @@ system:
       hostname: 192.168.31.201
       port: 1083
   scheduler_mode: local
+  human_intervention_alert_url: https://open.feishu.cn/open-apis/bot/v2/hook/379fcd1a-0fed-4e58-8cd0-40b6d1895721
 
 debug_flags:
   disable_llm_api_call: True

+ 1 - 0
pqai_agent/configs/prod.yaml

@@ -49,6 +49,7 @@ system:
       hostname: 127.0.0.1
       port: 1083
   scheduler_mode: mq
+  human_intervention_alert_url: https://open.feishu.cn/open-apis/bot/v2/hook/c316b559-1c6a-4c4e-97c9-50b44e4c2a9d
 
 agent_behavior:
   message_aggregation_sec: 20

+ 6 - 3
pqai_agent/toolkit/lark_alert_for_human_intervention.py

@@ -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": {