add data-set

push_message.py → evaluate_agent.py

@@ -250,22 +250,23 @@ if __name__ == '__main__':
         push_agent = MessagePushAgent()
         # reply_agent = MessageReplyAgent()
         for message in sub_dialogues['dialogues']:
-            agent_message = push_agent.generate_message(
-                context=user_profile,
-                dialogue_history=message
-            )
-            prompt = evaluate_push_agent_prompt(message, agent_message, user_profile, staff_profile)
+            # agent_message = push_agent.generate_message(
+            #     context=user_profile,
+            #     dialogue_history=message
+            # )
+            # prompt = evaluate_push_agent_prompt(message, agent_message, user_profile, staff_profile)
+            prompt = evaluate_conversation_quality_task(message, user_profile, staff_profile)
             response = fetch_deepseek_completion(prompt, output_type='json')
             obj = {
                 "user_profile": user_profile,
                 "agent_profile": staff_profile,
                 "dialogue_history": message,
-                "push_message": agent_message,
+                # "push_message": agent_message,
                 "push_time": datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
                 "evaluation_result": response
             }
             F.append(obj)
-    with open("scripts/push_message_evaluation_result.json", "w", encoding="utf-8") as f:
+    with open("scripts/reply_message_evaluation_result.json", "w", encoding="utf-8") as f:
         f.write(json.dumps(F, indent=4, ensure_ascii=False))