瀏覽代碼

Update dialogue_manager: add agent behavior config

StrayWarrior 2 周之前
父節點
當前提交
5c6f3ff0f5
共有 3 個文件被更改,包括 6 次插入2 次删除
  1. 2 0
      configs/dev.yaml
  2. 3 1
      configs/prod.yaml
  3. 1 1
      dialogue_manager.py

+ 2 - 0
configs/dev.yaml

@@ -33,6 +33,8 @@ storage:
       database: ai_agent
     table: agent_state
 
+agent_behavior:
+  message_aggregation_sec: 20
 
 chat_api:
   coze:

+ 3 - 1
configs/prod.yaml

@@ -33,7 +33,6 @@ storage:
       database: ai_agent
     table: agent_state
 
-
 chat_api:
   coze:
     oauth_client_id: 1194838245616
@@ -41,6 +40,9 @@ chat_api:
     private_key_path: oauth/coze_privkey.pem
     account_id: 649175100044793
 
+agent_behavior:
+  message_aggregation_sec: 20
+
 debug_flags:
   disable_llm_api_call: False
   use_local_user_storage: False

+ 1 - 1
dialogue_manager.py

@@ -100,7 +100,7 @@ class DialogueManager:
         self.complex_request_counter = 0
         self.human_intervention_triggered = False
         self.vector_memory = DummyVectorMemoryManager(user_id)
-        self.message_aggregation_sec = config.get('message_aggregation_sec', 5)
+        self.message_aggregation_sec = config.get('agent_behavior', {}).get('message_aggregation_sec', 5)
         self.unprocessed_messages = []
         self.history_dialogue_service = HistoryDialogueService(
             config['storage']['history_dialogue']['api_base_url']