Bug in TokenRAG

[stellaludai]

I was running python main.py -c path/to/FLARE.json on dataset 2wikimultihopqa, and I found that the query input to function retrieve is wrong: [xxx. Can you figure out why?