Function process_data

WebAgent/AgentFold/infer.py:335–351

(data)

            # thread pool
            lock = threading.Lock()
            def process_data(data):
                try:
                    full_traj = call_llm_with_tool(data, args)
                    final_block = full_traj.split('</think>')[-1]
                    data['final_response'] = final_block.split('<answer>')[-1].split('</answer>')[0].strip() if "<answer>" in final_block else final_block
                    data['full_traj'] = full_traj
                    with lock:
                        with open(save_path, 'a') as f:
                            f.write(json.dumps(data, ensure_ascii=False) + '\n')
                except Exception as e:
                    print(f">> Error in processing the question: {data['question']}")
                    print(traceback.format_exc())
                    error_response = getattr(e, 'response', 'N/A') 
                    with open(log_path, 'a') as f:
                        f.write(f">> Error in processing the question: {data['question']}\n")
                        f.write(traceback.format_exc() + '\n')
                        f.write(f">> Response leading to the error: {error_response}\n")
            
            if args.sequential:
                for data in test_data: