标签导航:

如何用javascript高效筛选对话数据并提取特定问题的全部回复?

高效筛选对话数据:JavaScript条件筛选与数据聚合

本文介绍如何使用JavaScript高效处理对话数据,实现基于特定条件的数据筛选和聚合。我们将通过一个示例演示如何从完整的对话记录中提取特定用户问题的全部回复。

假设我们有一个包含对话历史的JavaScript数组 chatHistory:

const chatHistory = [
    { "content": "你好", "role": "user" },
    { "role": "assistant", "content": "你好!很高兴为你服务。有什么问题或需要帮助吗?请随时告诉我。", "flag": "2" },
    { "role": "assistant", "content": "你好,很高兴见到你!我是ai优化助手,有什么可以帮助你的吗?

如果你需要优化你的ai提示词,请告诉我你当前面临的具体问题或需求,我会尽力为你提供帮助。", "flag": "3" },
    { "role": "assistant", "content": "很高兴为您服务!如果您有任何问题或需要帮助,请随时告诉我。。", "flag": "4" },
    { "role": "assistant", "content": "你好!很高兴为你服务。有什么我可以帮助你的吗?", "flag": "1" },
    { "content": "你是谁", "role": "user" },
    { "role": "assistant", "content": "我是", "flag": "2" },
    { "role": "assistant", "content": "你好,我是你的ai优化助手。", "flag": "3" },
    { "role": "assistant", "content": "我是您的人工智能优化助手,专门致力于帮助用户优化与ai大模型的交互体验。", "flag": "4" },
    { "role": "assistant", "content": "作为一个认知智能模型,我的目标是为您提供有关各种主题的有用信息和建议", "flag": "1" }
];

以及一个表示用户问题的对象 question:

const question = { "name": "question", "text": "你好" };

我们的目标是从 chatHistory 中筛选出针对 question 的所有回复。 为此,我们可以编写一个JavaScript函数 lookupChatByQuestion:

function lookupChatByQuestion(question, chatHistory) {
  let replies = [];
  let foundQuestion = false;

  for (const message of chatHistory) {
    if (message.role === "user" && message.content === question.text) {
      foundQuestion = true;
    } else if (foundQuestion && message.role === "assistant") {
      replies.push(message.content);
    } else if (message.role === "user" && foundQuestion) {
      break; // 遇到下一个用户问题,停止收集回复
    }
  }
  return replies;
}

const replies = lookupChatByQuestion(question, chatHistory);
console.log(replies);

这个函数迭代 chatHistory,找到与 question 匹配的用户提问,然后收集其后的所有助手回复,直到遇到下一个用户提问或遍历结束。 最终返回一个包含所有相关回复的数组。