|
@@ -23,6 +23,8 @@ from pqai_agent.message import MessageType, Message
|
|
|
from pqai_agent.toolkit.lark_alert_for_human_intervention import LarkAlertForHumanIntervention
|
|
|
from pqai_agent.toolkit.lark_sheet_record_for_human_intervention import LarkSheetRecordForHumanIntervention
|
|
|
from pqai_agent.user_manager import UserManager
|
|
|
+from pqai_agent.utils import prompt_utils
|
|
|
+
|
|
|
|
|
|
class DummyVectorMemoryManager:
|
|
|
def __init__(self, user_id):
|
|
@@ -480,6 +482,10 @@ class DialogueManager:
|
|
|
self.user_profile = self.user_manager.get_user_profile(self.user_id)
|
|
|
# 刷新员工画像(不一定需要)
|
|
|
self.staff_profile = self.user_manager.get_staff_profile(self.staff_id)
|
|
|
+ # 员工画像添加前缀,避免冲突,实现Coze Prompt模板的平滑升级
|
|
|
+ legacy_staff_profile = {}
|
|
|
+ for key in self.staff_profile:
|
|
|
+ legacy_staff_profile[f'agent_{key}'] = self.staff_profile[key]
|
|
|
|
|
|
current_datetime = datetime.now()
|
|
|
context = {
|
|
@@ -493,8 +499,9 @@ class DialogueManager:
|
|
|
"last_interaction_interval": self._get_hours_since_last_interaction(2),
|
|
|
"if_first_interaction": True if self.previous_state == DialogueState.INITIALIZED else False,
|
|
|
"if_active_greeting": False if user_message else True,
|
|
|
+ "formatted_staff_profile": prompt_utils.format_agent_profile(self.staff_profile),
|
|
|
**self.user_profile,
|
|
|
- **self.staff_profile
|
|
|
+ **legacy_staff_profile
|
|
|
}
|
|
|
|
|
|
# 获取长期记忆
|