merge master

generate_data_set.py

@@ -228,8 +228,8 @@ def generate_push_dataset():
     samples =random.sample(filter_conversation, 100)
 
     # init message push agent
-    agent = MessagePushAgent()
     for sample in tqdm(samples):
+        agent = MessagePushAgent()
         agent_profile = get_profile_info(sample["staff_id"], "staff")[0]['profile']
         agent_profile = json.loads(agent_profile)
         user_profile = get_profile_info(sample["user_id"], "user")[0]['profile']

pqai_agent/agents/message_push_agent.py

@@ -141,8 +141,8 @@ class MessagePushAgent(SimpleOpenAICompatibleChatAgent):
         self.run(query)
         for tool_call in reversed(self.tool_call_records):
             if tool_call['name'] == MessageNotifier.message_notify_user.__name__:
-                time.sleep(1)
-                print(tool_call['arguments']['message'])
+                # time.sleep(1)
+                print("Function call return", tool_call['arguments']['message'])
                 return tool_call['arguments']['message']
         return ''