Function process_data

WebAgent/WebWalker/src/rag_system.py:230–247

(data)

            temp = {"question": item}
            data_list.append(temp)
    def process_data(data):
        messages = [{"role": "system", "content": "你是 Kimi。"}, {"role": "user", "content": data["question"]}]
        finish_reason = None
        while finish_reason is None or finish_reason == "tool_calls":
            choice = chat(messages)
            finish_reason = choice.finish_reason
            if finish_reason == "tool_calls":
                messages.append(choice.message)
                for tool_call in choice.message.tool_calls:
                    tool_result = search_impl(json.loads(tool_call.function.arguments))
                    messages.append({
                        "role": "tool",
                        "tool_call_id": tool_call.id,
                        "name": tool_call.function.name,
                        "content": json.dumps(tool_result),
                    })
        data["pred"] = choice.message.content
        return data

    with tqdm(total=len(data_list)) as pbar:
        with concurrent.futures.ThreadPoolExecutor(max_workers=MAX_WORKERS) as executor: